Update app.py

app.py

@@ -3,21 +3,66 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from google.genai import types
+
+# Import the Google ADK agent components from agent.py
+from agent import runner, session_service, APP_NAME, USER_ID, SESSION_ID
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
+# --- Google ADK Agent Wrapper ---
+# ----- USING THE ACTUAL GOOGLE ADK AGENT FROM AGENT.PY ------
+class GoogleADKAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
+        print("GoogleADKAgent initialized with Google ADK runner and root_agent.")
+        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
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        try:
+            # Create a unique session for each question to avoid state conflicts
+            self.question_counter += 1
+            unique_session_id = f"{SESSION_ID}_q{self.question_counter}"
+
+            session = self.session_service.create_session(
+                app_name=self.app_name,
+                user_id=self.user_id,
+                session_id=unique_session_id
+            )
+
+            # Create the query content
+            query_content = types.Content(
+                role='user',
+                parts=[types.Part(text=question)]
+            )
+
+            # Run the agent synchronously using the runner
+            events = list(self.runner.run(session=session, new_message=query_content))
+
+            # Extract the final answer from the events
+            final_answer = "No response generated."
+            for event in reversed(events):  # Look at events in reverse to find the last meaningful response
+                if event.content and event.content.parts:
+                    for part in event.content.parts:
+                        if part.text and part.text.strip():
+                            final_answer = part.text.strip()
+                            break
+                    if final_answer != "No response generated.":
+                        break
+
+            print(f"Agent returning answer: {final_answer[:100]}...")
+            return final_answer
+
+        except Exception as e:
+            error_msg = f"Error running Google ADK agent: {str(e)}"
+            print(error_msg)
+            return error_msg
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -40,7 +85,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = GoogleADKAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None