Update app.py

app.py

@@ -7,10 +7,7 @@ import pandas as pd
 # Try to import Google ADK components, fallback to simple agent if not available
 try:
     from google.genai import types
-    from agent import session_service, APP_NAME, USER_ID, SESSION_ID, code_agent, search_agent, image_agent, youtube_agent
-    from google.adk.runners import Runner
-    from google.adk.agents import LlmAgent
-    from google.adk.tools import agent_tool
+    from agent import session_service, APP_NAME, USER_ID, SESSION_ID, runner
     GOOGLE_ADK_AVAILABLE = True
     print("✅ Google ADK components loaded successfully")
 except ImportError as e:
@@ -128,48 +125,14 @@ class GoogleADKAgent:
         print("GoogleADKAgent initialized with Google ADK runner and agents.")
 
         try:
-            # Create a specialized GAIA agent for complex multi-step questions
-            self.question_agent = LlmAgent(
-                name='gaiaAgent',
-                model="gemini-2.5-pro-preview-05-06",
-                description="You are an expert agent specialized in solving complex GAIA benchmark questions that require multi-step reasoning, research, and analysis",
-                instruction=(
-                    "You are an expert agent designed to solve complex GAIA benchmark questions. "
-                    "These questions often require multiple steps, research, and different types of analysis. "
-                    "\n\nFor each question, follow this approach:"
-                    "\n1. ANALYZE the question carefully to identify what information is needed"
-                    "\n2. BREAK DOWN the question into logical steps if it involves multiple parts"
-                    "\n3. USE APPROPRIATE TOOLS:"
-                    "\n   - For images/visual content: use the image agent"
-                    "\n   - For web research/factual information: use the search agent"
-                    "\n   - For data analysis/calculations/file processing: use the code agent"
-                    "\n   - For YouTube videos: use the youtube agent"
-                    "\n4. CHAIN your reasoning across multiple steps when needed"
-                    "\n5. PROVIDE EXACT ANSWERS in the format requested"
-                    "\n\nIMPORTANT: Your final answer must be precise and match exactly what is asked. "
-                    "Pay attention to formatting requirements (lists, ordering, plural/singular forms, etc.). "
-                    "Do not include phrases like 'FINAL ANSWER:' - just provide the direct answer."
-                ),
-                tools=[
-                    agent_tool.AgentTool(agent=code_agent),
-                    agent_tool.AgentTool(agent=search_agent),
-                    agent_tool.AgentTool(agent=image_agent),
-                    agent_tool.AgentTool(agent=youtube_agent)
-                ]
-            )
-
-            # Create runner with the question agent
-            self.runner = Runner(
-                agent=self.question_agent,
-                app_name=APP_NAME,
-                session_service=session_service
-            )
+            # Use the pre-configured runner and root_agent from agent.py
+            self.runner = runner
             self.session_service = session_service
             self.app_name = APP_NAME
             self.user_id = USER_ID
             self.question_counter = 0  # To create unique session IDs for each question
             self.initialized = True
-            print("✅ Google ADK Agent successfully initialized")
+            print("✅ Google ADK Agent successfully initialized using pre-configured runner")
 
         except Exception as e:
             print(f"❌ Failed to initialize Google ADK Agent: {e}")
@@ -187,6 +150,19 @@ class GoogleADKAgent:
             self.question_counter += 1
             unique_session_id = f"{SESSION_ID}_q{self.question_counter}"
 
+            # Create the session before using it
+            try:
+                self.session_service.create_session(
+                    app_name=self.app_name,
+                    user_id=self.user_id,
+                    session_id=unique_session_id
+                )
+                print(f"✅ Created session: {unique_session_id}")
+            except Exception as session_error:
+                print(f"⚠️ Session creation error: {session_error}")
+                # Fallback to the default session
+                unique_session_id = SESSION_ID
+
             # Create the query content
             query_content = types.Content(
                 role='user',
@@ -194,12 +170,23 @@ class GoogleADKAgent:
             )
 
             # Run the agent synchronously using the runner with correct parameters
+            print(f"🚀 Running agent with session: {unique_session_id}")
             events = list(self.runner.run(
                 user_id=self.user_id,
                 session_id=unique_session_id,
                 new_message=query_content
             ))
 
+            print(f"📊 Generated {len(events)} events")
+
+            # Debug: Print event details
+            for i, event in enumerate(events):
+                print(f"Event {i}: author={getattr(event, 'author', 'unknown')}, content_type={type(getattr(event, 'content', None))}")
+                if hasattr(event, 'content') and event.content and hasattr(event.content, 'parts'):
+                    for j, part in enumerate(event.content.parts):
+                        if hasattr(part, 'text') and part.text:
+                            print(f"  Part {j}: {part.text[:100]}...")
+
             # Extract the final answer from the events
             final_answer = "No response generated."