updated app.py

app.py

@@ -4,12 +4,14 @@ import requests
 import inspect
 import pandas as pd
 # from agents import LlamaIndexAgent
-from new_langraph_agent import run_agent as _sync_run_agent  # Updated: use the new multi-agent runner
+from langgraph_agent_system import run_agent_system  # Updated: use the latest multi-agent system
+from observability import flush_traces, shutdown_observability  # Add cleanup functions
 import asyncio
 import aiohttp
 from langfuse.langchain import CallbackHandler
 from langchain_core.messages import HumanMessage
 import tempfile
+import uuid
 
 # Initialize Langfuse CallbackHandler for LangGraph/Langchain (tracing)
 langfuse_handler = CallbackHandler()
@@ -21,18 +23,25 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------         
 class BasicAgent:
-    """Wrapper that executes the new multi-agent LangGraph system in a background thread."""
+    """Wrapper that executes the latest multi-agent LangGraph system."""
 
     def __init__(self):
-        print("BasicAgent (multi-agent) initialized.")
+        print("BasicAgent (latest multi-agent system) initialized.")
 
     async def aquery(self, question: str) -> str:
-        """Run the synchronous `run_agent` helper inside the event-loop executor."""
+        """Run the latest async multi-agent system directly."""
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        loop = asyncio.get_event_loop()
         try:
-            # Off-load the blocking call to a thread so we don't block the Gradio event loop
-            answer = await loop.run_in_executor(None, _sync_run_agent, question)
+            # Generate unique session ID for this query
+            session_id = f"app_session_{uuid.uuid4().hex[:8]}"
+            
+            # Call the latest async agent system directly
+            answer = await run_agent_system(
+                query=question,
+                user_id="gradio_app_user",
+                session_id=session_id,
+                max_iterations=3
+            )
             print(f"Agent returning answer: {answer}")
             return answer
         except Exception as e:
@@ -234,6 +243,15 @@ with gr.Blocks() as demo:
         api_name="submit_answers",
     )
 
+def cleanup_agent_system():
+    """Cleanup function for the agent system."""
+    try:
+        flush_traces(background=False)
+        shutdown_observability()
+        print("✅ Agent system cleanup completed")
+    except Exception as e:
+        print(f"⚠️ Cleanup warning: {e}")
+
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)
     # Check for SPACE_HOST and SPACE_ID at startup for information
@@ -255,5 +273,16 @@ if __name__ == "__main__":
 
     print("-"*(60 + len(" App Starting ")) + "\n")
 
-    print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    print("Launching Gradio Interface with Latest Multi-Agent System...")
+    print("🤖 Using: LangGraph Multi-Agent System (Lead → Research → Code → Formatter)")
+    print("📊 Features: Langfuse v3 observability, iterative workflows, GAIA compliance")
+    
+    try:
+        demo.launch(debug=True, share=False)
+    except KeyboardInterrupt:
+        print("\n🛑 Shutting down gracefully...")
+        cleanup_agent_system()
+    except Exception as e:
+        print(f"❌ Error during app execution: {e}")
+        cleanup_agent_system()
+        raise