Update app.py

app.py

@@ -118,18 +118,12 @@ Responda com base no contexto quando relevante."""
     
     # Adiciona mensagens do histórico, garantindo alternância entre "user" e "assistant"
     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})
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
     
     # Adiciona a nova mensagem do usuário
     messages.append({"role": "user", "content": rag_prompt})
 
-    # Garantia de que a lista de mensagens alterne corretamente
-    assert all(messages[i]["role"] != messages[i + 1]["role"] for i in range(len(messages) - 1)), \
-        "As mensagens devem alternar corretamente entre 'user' e 'assistant'."
-
     response = ""
     try:
         for message_chunk in client.chat_completion(
@@ -148,6 +142,8 @@ Responda com base no contexto quando relevante."""
 
 # Função para carregar PDFs
 def load_pdfs(pdf_files: List[gr.File]):
+    if not pdf_files:
+        return "Nenhum arquivo selecionado."
     status = knowledge_base.load_pdfs(pdf_files)
     return status
 
@@ -168,7 +164,7 @@ with gr.Blocks(title="RAG Avançado com PDFs", theme=gr.themes.Soft()) as demo:
             submit_btn = gr.Button("Enviar")
         
         with gr.Column(scale=1):
-            context_box = gr.Markdown(label="Contexto Recuperado", value="Contexto aparecerá aqui após a pergunta.")
+            context_box = gr.Markdown(label="Contexto Recuperado")
 
     with gr.Accordion("Configurações", open=False):
         with gr.Row():
@@ -192,10 +188,13 @@ with gr.Blocks(title="RAG Avançado com PDFs", theme=gr.themes.Soft()) as demo:
     # Função para atualizar o chat
     def submit_message(message, history, system_message, max_tokens, temperature, top_p, k_initial, k_final):
         history = history or []
+        return_history = history.copy()
+        return_history.append((message, ""))
+        yield return_history, "", ""
+        
         for response, context, _ in respond(message, history, system_message, max_tokens, temperature, top_p, k_initial, k_final):
-            history.append((message, response))
-            yield history, context, ""
-        yield history, context, ""
+            return_history[-1] = (message, response)
+            yield return_history, context, ""
 
     # Conexões de eventos
     submit_btn.click(
@@ -215,4 +214,4 @@ with gr.Blocks(title="RAG Avançado com PDFs", theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch()