Update app.py

app.py

@@ -7,6 +7,7 @@ import pandas as pd
 
 import google.generativeai as genai
 from smolagents import CodeAgent, DuckDuckGoSearchTool
+from types import SimpleNamespace
 
 # System prompt used by the agent
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
@@ -17,13 +18,14 @@ If you're asked for a string, don’t use articles or abbreviations (e.g. for ci
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# 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-compatible result wrapper that mimics smolagents' expected structure
+def GenerationResult(content, token_usage=None, input_tokens=0, output_tokens=0):
+    return SimpleNamespace(
+        content=content,
+        token_usage=token_usage or {},
+        input_tokens=input_tokens,
+        output_tokens=output_tokens
+    )
 
 # Gemini model wrapper
 class GeminiFlashModel:
@@ -49,14 +51,12 @@ class GeminiFlashModel:
             response = self.model.generate_content(prompt)
             return GenerationResult(
                 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
             )