Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 import time
-import re 
+import re
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 
@@ -17,18 +17,20 @@ MODELS = {
     "DeepSeek R1": "deepseek-ai/DeepSeek-R1"
 }
 
-history_log = []
+history_log = []    
+recent_history = []   
 show_history_flag = False
 
 def clean_response(text):
     text = re.sub(r"</?think>", "", text)
-    text = re.sub(r"\\boxed\{.*?\}", "", text)
-
+    text = re.sub(r"\\boxed\\{.*?\\}", "", text)
     return text.strip()
 
 def chatbot_response(user_input, model_name):
     model_id = MODELS[model_name]
-    messages = [{"role": "user", "content": entry["Pergunta"]} for entry in history_log[-1:]]
+    
+    # Mantém apenas as 2 últimas interações para o modelo
+    messages = recent_history[-2:] if len(recent_history) >= 2 else []
     messages.append({"role": "user", "content": user_input})
 
     start_time = time.time()
@@ -36,14 +38,17 @@ def chatbot_response(user_input, model_name):
         completion = client.chat.completions.create(
             model=model_id,
             messages=messages,
-            max_tokens = 8192 if any(m in model_id for m in ["Qwen", "DeepSeek"]) else 800
+            max_tokens=8192 if any(m in model_id for m in ["Qwen", "DeepSeek"]) else 800
         )
         response = completion.choices[0].message['content']
-        
     except Exception as e:
         response = f"Erro ao gerar resposta: {str(e)}"
     end_time = time.time()
 
+    # Atualiza os dois históricos
+    recent_history.append({"role": "user", "content": user_input})
+    recent_history.append({"role": "assistant", "content": response})
+
     history_log.append({
         "Modelo": model_name,
         "Pergunta": user_input,
@@ -67,10 +72,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             gr.Markdown("# 🤖 Chatbot - API SambaNova")
             chatbot = gr.Chatbot(height=500)
             msg = gr.Textbox(placeholder="Digite sua mensagem aqui...", show_label=False)
-            
+
             btn = gr.Button("Enviar", variant="primary")
             history_btn = gr.Button("Histórico", variant="secondary")
-                
             history_output = gr.JSON()
 
     def respond(message, chat_history, model_name):
@@ -91,4 +95,3 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
 if __name__ == "__main__":
     demo.launch()
-