Update app.py

app.py

@@ -76,7 +76,6 @@ class AdvancedPDFKnowledgeBase:
 # Inicializa a base de conhecimento
 knowledge_base = AdvancedPDFKnowledgeBase()
 
-# Função principal de resposta
 def respond(
     message: str,
     history: List[Tuple[str, str]],
@@ -93,32 +92,41 @@ def respond(
     
     context = knowledge_base.get_relevant_context(message, k_initial, k_final)
     
+    # Constrói o prompt RAG
     rag_prompt = f"""Você é Grok 3, criado por xAI. Use o contexto dos documentos para responder:
 {context}
 Pergunta: {message}
 Responda com base no contexto quando relevante."""
     
-    messages = [{"role": "system", "content": system_message},
-                {"role": "user", "content": rag_prompt}]
+    # Inicializa a lista de mensagens
+    messages = [{"role": "system", "content": system_message}]
     
+    # Adiciona mensagens do histórico
     for user_msg, assistant_msg in history:
         if user_msg:
             messages.append({"role": "user", "content": user_msg})
         if assistant_msg:
             messages.append({"role": "assistant", "content": assistant_msg})
+    
+    # Adiciona a nova mensagem do usuário
+    messages.append({"role": "user", "content": rag_prompt})
 
     response = ""
-    for message_chunk in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message_chunk.choices[0].delta.content
-        response += token
-        yield response, context, ""
-
+    try:
+        for message_chunk in client.chat_completion(
+            messages=messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            token = message_chunk.choices[0].delta.content
+            if token:
+                response += token
+                yield response, context, ""
+    except Exception as e:
+        yield f"Erro ao gerar resposta: {str(e)}", context, ""
+        
 # Função para carregar PDFs
 def load_pdfs(pdf_files: List[gr.File]):
     status = knowledge_base.load_pdfs(pdf_files)