Update app.py

app.py

@@ -222,13 +222,19 @@ def demo():
                 db_progress = gr.Textbox(label="Vector database initialization", value="None")
             with gr.Row():
                 db_btn = gr.Button("Generate vector database")
-        
+        # Define o estado para o prompt inicial
+        initial_prompt = gr.State("")
+
+        # Define a aba "Set Initial Prompt"
         with gr.Tab("Step 3 - Set Initial Prompt"):
             with gr.Row():
                 prompt_input = gr.Textbox(label="Initial Prompt", lines=5, value="Você é um advogado sênior, onde seu papel é analisar e trazer as informações sem inventar, dando a sua melhor opinião sempre trazendo contexto e referência. Aprenda o que é jurisprudência.")
             with gr.Row():
                 set_prompt_btn = gr.Button("Set Prompt")
 
+        # Atualiza o estado do prompt inicial ao clicar no botão "Set Prompt"
+        set_prompt_btn.click(fn=lambda prompt: prompt, inputs=prompt_input, outputs=initial_prompt)
+
         with gr.Tab("Step 4 - Initialize QA chain"):
             with gr.Row():
                 llm_btn = gr.Radio(list_llm_simple,