Update app.py

app.py

@@ -4,20 +4,19 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-
 import google.generativeai as genai
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 
-# System prompt used by the agent
+# System prompt for Gemini
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
-Report your thoughts, and finish your answer with just the answer — no prefixes like "FINAL ANSWER:".
+Report your thoughts, and finish your answer with just the answer — no prefixes like \"FINAL ANSWER:\".
 Your answer should be a number OR as few words as possible OR a comma-separated list of numbers and/or strings.
 If you're asked for a number, don’t use commas or units like $ or %, unless specified.
 If you're asked for a string, don’t use articles or abbreviations (e.g. for cities), and write digits in plain text unless told otherwise."""
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# Generation result wrapper to match smolagents expectations
+# Generation result wrapper
 class GenerationResult:
     def __init__(self, content, token_usage=None, input_tokens=0, output_tokens=0):
         self.content = content
@@ -47,32 +46,31 @@ class GeminiFlashModel:
 
         try:
             response = self.model.generate_content(prompt)
-            text_output = response.text.strip() if hasattr(response, "text") else str(response)
-
             return GenerationResult(
-                content=text_output,
-                token_usage={"input_tokens": 0, "output_tokens": 0},
+                content=response.text.strip(),
+                token_usage={},
                 input_tokens=0,
                 output_tokens=0
             )
         except Exception as e:
             return GenerationResult(
                 content=f"GENERATION ERROR: {e}",
-                token_usage={"input_tokens": 0, "output_tokens": 0},
+                token_usage={},
                 input_tokens=0,
                 output_tokens=0
             )
 
-# Agent wrapper
+# Agent class
 class MyAgent:
     def __init__(self):
-        self.model = GeminiFlashModel(model_id="gemini-1.5-flash")
+        self.model = GeminiFlashModel()
         self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=self.model)
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str):
         return self.agent.run(question)
 
-# Main evaluation function
+# Evaluation function
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
@@ -106,10 +104,28 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         question_text = item.get("question")
         if not task_id or question_text is None:
             continue
+
         try:
-            submitted_answer = agent(question_text)
-            answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+            result = agent(question_text)
+
+            if hasattr(result, "content"):
+                submitted_answer = result.content
+            elif isinstance(result, dict) and "content" in result:
+                submitted_answer = result["content"]
+            else:
+                submitted_answer = str(result)
+
+            answers_payload.append({
+                "task_id": task_id,
+                "submitted_answer": submitted_answer
+            })
+
+            results_log.append({
+                "Task ID": task_id,
+                "Question": question_text,
+                "Submitted Answer": submitted_answer
+            })
+
         except Exception as e:
             results_log.append({
                 "Task ID": task_id,
@@ -141,7 +157,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission failed: {e}", pd.DataFrame(results_log)
 
-# Gradio UI setup
+# UI setup
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""
@@ -159,7 +175,7 @@ with gr.Blocks() as demo:
     run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
 if __name__ == "__main__":
-    print("🔧 App starting...")
+    print("\ud83d\udd27 App starting...")
     demo.launch(debug=True, share=False)