Update app.py

app.py

@@ -8,7 +8,18 @@ import pandas as pd
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-from agent import BasicAgent
+from gemini_agent import GEMINI_AGENT
+class BasicAgent:
+    def __init__(self):
+        self.agent = GEMINI_AGENT()
+
+    def __call__(self, task: dict) -> str:
+        try:
+            response = self.agent(task)
+            return response
+        except Exception as e:
+            print(f"Error is raised: {str(e)}")
+            return "Agent could not complete this task"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """