Update app.py

app.py

@@ -9,12 +9,12 @@ import gradio as gr
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Enhanced Agent Definition ---
+
 class GAIAAgent:
     def __init__(self):
         print("GAIAAgent initialized.")
         self.model = gr.load("models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", provider="novita")
-        search_tool = DuckDuckGoSearchTool()
-        self.agent = CodeAgent(model=model, tools=[search_tool])
+        self.search_tool = DuckDuckGoSearchTool()  # optional if you still want to use search manually
 
     def format_prompt(self, question: str, file_content: str = None) -> str:
         prompt = (
@@ -36,10 +36,40 @@ class GAIAAgent:
     def __call__(self, question: str, file_name: str = None) -> str:
         file_content = self.read_file(file_name) if file_name else None
         prompt = self.format_prompt(question, file_content)
-        result = self.agent.run(prompt)
+        result = self.model(prompt)  # directly call the Gradio-loaded model
         return result.strip()
-
-
+        
+# class GAIAAgent:
+#     def __init__(self):
+#         print("GAIAAgent initialized.")
+#         self.model = gr.load("models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", provider="novita")
+#         search_tool = DuckDuckGoSearchTool()
+#         self.agent = CodeAgent(model=model, tools=[search_tool])
+
+#     def format_prompt(self, question: str, file_content: str = None) -> str:
+#         prompt = (
+#             "You are a helpful AI agent solving a question from the GAIA benchmark. "
+#             "Respond only with the final answer."
+#         )
+#         if file_content:
+#             prompt += f"\nAttached File Content:\n{file_content}\n"
+#         prompt += f"\nQuestion: {question}\nAnswer:"
+#         return prompt
+
+#     def read_file(self, filename: str) -> str:
+#         filepath = os.path.join("./", filename)
+#         if filename.endswith(".txt") and os.path.exists(filepath):
+#             with open(filepath, "r") as file:
+#                 return file.read()[:1000]  # limit to 1000 chars
+#         return ""
+
+#     def __call__(self, question: str, file_name: str = None) -> str:
+#         file_content = self.read_file(file_name) if file_name else None
+#         prompt = self.format_prompt(question, file_content)
+#         result = self.agent.run(prompt)
+#         return result.strip()
+
+        
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if profile: