Update app.py

app.py

@@ -19,8 +19,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # Wrapper class for model output
 class GenerationResult:
-    def __init__(self, content):
+    def __init__(self, content, token_usage=None):
         self.content = content
+        self.token_usage = token_usage or {}
 
 # Gemini model wrapper
 class GeminiFlashModel:
@@ -45,9 +46,9 @@ class GeminiFlashModel:
 
         try:
             response = self.model.generate_content(prompt)
-            return GenerationResult(response.text.strip())
+            return GenerationResult(content=response.text.strip(), token_usage={})  # Gemini doesn't expose usage
         except Exception as e:
-            return GenerationResult(f"GENERATION ERROR: {e}")
+            return GenerationResult(content=f"GENERATION ERROR: {e}", token_usage={})
 
 # Agent using Gemini
 class MyAgent:
@@ -150,3 +151,4 @@ if __name__ == "__main__":
 
 
 
+