Update app.py

app.py

@@ -10,14 +10,22 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool
 
 # System prompt used by the agent
 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"
 
-# Gemini model wrapper returning a dictionary instead of custom class
+# Generation result wrapper to match smolagents expectations
+class GenerationResult:
+    def __init__(self, content, token_usage=None, input_tokens=0, output_tokens=0):
+        self.content = content
+        self.token_usage = token_usage or {}
+        self.input_tokens = input_tokens
+        self.output_tokens = output_tokens
+
+# Gemini model wrapper
 class GeminiFlashModel:
     def __init__(self, model_id="gemini-1.5-flash", api_key=None):
         genai.configure(api_key=api_key or os.getenv("GEMINI_API_KEY"))
@@ -39,19 +47,21 @@ class GeminiFlashModel:
 
         try:
             response = self.model.generate_content(prompt)
-            return {
-                "content": response.text.strip(),
-                "input_tokens": 0,
-                "output_tokens": 0,
-                "token_usage": {},
-            }
+            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},
+                input_tokens=0,
+                output_tokens=0
+            )
         except Exception as e:
-            return {
-                "content": f"GENERATION ERROR: {e}",
-                "input_tokens": 0,
-                "output_tokens": 0,
-                "token_usage": {},
-            }
+            return GenerationResult(
+                content=f"GENERATION ERROR: {e}",
+                token_usage={"input_tokens": 0, "output_tokens": 0},
+                input_tokens=0,
+                output_tokens=0
+            )
 
 # Agent wrapper
 class MyAgent:
@@ -149,10 +159,11 @@ with gr.Blocks() as demo:
     run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
 if __name__ == "__main__":
-    print("\U0001F527 App starting...")
+    print("🔧 App starting...")
     demo.launch(debug=True, share=False)
 
 
 
 
 
+