Update app.py

app.py

@@ -418,6 +418,27 @@ with gr.Blocks(title="GAIA Agent") as app:
                 outputs=[questions_display]
             )
 
+# Import Agent class from agent.py for template compatibility
+try:
+    from agent import Agent
+    print("✅ Agent class imported from agent.py")
+except ImportError:
+    # Fallback: create Agent class here
+    class Agent:
+        """Fallback Agent class."""
+        def __init__(self):
+            print("Agent initialized (fallback)")
+        
+        def __call__(self, question: str) -> str:
+            try:
+                if not question or not question.strip():
+                    return "Error: Empty question"
+                answer = run_agent(question)
+                return answer if answer else "No answer found"
+            except Exception as e:
+                return f"Error: {str(e)}"
+
 if __name__ == "__main__":
-    app.launch(share=False)
+    # Launch main app
+    app.launch(share=False, server_name="0.0.0.0", server_port=7860)