Update app.py

app.py

@@ -117,7 +117,7 @@ def run_agent():
     print("Finished submission.")
 
 # ===============================
-# GRADIO UI
+# GRADIO BLOCKS UI
 # ===============================
 
 def run_pipeline():
@@ -127,15 +127,12 @@ def run_pipeline():
     except Exception as e:
         return f"❌ Error occurred: {str(e)}"
 
-# Create a simple text output
-output_text = gr.Textbox(label="Output", lines=4)
-
-# Create a button and link it to the function
-run_button = gr.Button("Run GAIA Agent")
-run_button.click(fn=run_pipeline, inputs=[], outputs=output_text)
-
-# Launch the interface
-demo = gr.Row([run_button, output_text])
+with gr.Blocks() as demo:
+    run_button = gr.Button("Run GAIA Agent")
+    output_text = gr.Textbox(label="Output", lines=4)
+    
+    # Link button click to function
+    run_button.click(fn=run_pipeline, inputs=[], outputs=output_text)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)