Ajustando lógica do botão

app.py

@@ -30,13 +30,13 @@ def process_file(file):
 
     file_size = os.path.getsize(file)
     if file_size > MAX_FILE_SIZE:
-        return f"O arquivo excede o limite. Por favor, realize o upload de um arquivo que contenha no máximo {MAX_FILE_SIZE/1024/1024:.1f}MB.", gr.update(interactive=False)
+        return f"O arquivo excede o limite. Por favor, realize o upload de um arquivo que contenha no máximo {MAX_FILE_SIZE/1024/1024:.1f}MB."
 
     texto_extraido = doc_converter(file)
     if texto_extraido is None:
-        return "Não foi possível extrair o texto.", gr.update(interactive=False)
+        return "Não foi possível extrair o texto."
     
-    return texto_extraido, gr.update(interactive=True)
+    return texto_extraido
 
 @spaces.GPU
 def ask_question(texto_extraido, question):
@@ -49,10 +49,15 @@ def ask_question(texto_extraido, question):
     # resposta = rag_chain.run(question)
     response = rag_chain({"query": question})
     resposta = response["result"]
-    docs_text = "\n\n\n\n\n\n===================================\n\n\n\n\n\n".join([doc.page_content for doc in response["source_documents"]])
+    docs_text = "\n\n\n===================================\n\n\n".join([doc.page_content for doc in response["source_documents"]])
 
     return resposta, docs_text
 
+def update_ask_button(extracted_text, question):
+    if extracted_text and question.strip():
+        return gr.update(interactive=True)
+    return gr.update(interactive=False)
+
 def launch_app():
     with gr.Blocks(title="RAG", theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray", neutral_hue="slate")) as demo:
         gr.Markdown("# 🚀 Retrieval Augmented Generation - RAG")
@@ -72,7 +77,12 @@ def launch_app():
                 docs_sim = gr.Textbox(label="📎 Documentos similares a sua pergunta.", lines=15)
 
         # Conecta funções
-        file_input.change(fn=process_file, inputs=file_input, outputs=[extracted_text, ask_button])
+        file_input.change(fn=process_file, inputs=file_input, outputs=extracted_text)
+        # Sempre que o texto extraído ou a pergunta mudar, atualiza o botão
+        extracted_text.change(fn=update_ask_button, inputs=[extracted_text, question_input], outputs=ask_button)
+        question_input.change(fn=update_ask_button, inputs=[extracted_text, question_input], outputs=ask_button)
+
+        # Chama o ask_question com Enter ou botão
         question_input.submit(fn=ask_question, inputs=[extracted_text, question_input], outputs=[answer_output, docs_sim])
         ask_button.click(fn=ask_question, inputs=[extracted_text, question_input], outputs=[answer_output, docs_sim])