Update app.py

app.py

@@ -112,12 +112,20 @@ def demo():
             doc_source1, doc_source2, doc_source3 = gr.Textbox(label="Referência 1"), gr.Textbox(label="Referência 2"), gr.Textbox(label="Referência 3")
             source1_page, source2_page, source3_page = gr.Number(label="Página 1"), gr.Number(label="Página 2"), gr.Number(label="Página 3")
             
+            # Campo de texto para enviar mensagens
+            user_input = gr.Textbox(label="Sua mensagem")
+
             # Implementação de lógica de interação de conversa
-            qachain_btn.click(initialize_database_and_llm, inputs=[document, slider_chunk_size, slider_chunk_overlap, llm_btn, slider_temperature, slider_maxtokens, slider_topk], outputs=[vector_db, collection_name, qa_chain])
-            chatbot.submit(conversation, inputs=[qa_chain, chatbot.input, chatbot.history], outputs=[qa_chain, gr.update(value=""), chatbot.history, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page])
+            def send_message(message, history, qa_chain):
+                formatted_chat_history = [f"Usuário: {user_message}\nAssistente: {bot_message}" for user_message, bot_message in history]
+                response = qa_chain({"question": message, "chat_history": formatted_chat_history})
+                response_answer = response["answer"].split("Resposta útil:")[-1]
+                response_sources = [doc.page_content.strip() for doc in response["source_documents"]]
+                response_pages = [doc.metadata["page"] + 1 for doc in response["source_documents"]]
+                new_history = history + [(message, response_answer)]
+                return qa_chain, gr.update(value=""), new_history, *response_sources, *response_pages
+
+            user_input.submit(send_message, inputs=[user_input, chatbot.history, qa_chain], outputs=[qa_chain, gr.update(value=""), chatbot.history, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page])
         
     demo.launch()
 
-# Executando o app
-if __name__ == "__main__":
-    demo()