Update app.py

app.py

@@ -7,27 +7,17 @@ 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.
-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-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
+# Gemini model wrapper returning a dictionary instead of custom class
 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"))
@@ -49,17 +39,19 @@ class GeminiFlashModel:
 
         try:
             response = self.model.generate_content(prompt)
-            return GenerationResult(
-                content=response.text.strip(),
-                input_tokens=0,
-                output_tokens=0
-            )
+            return {
+                "content": response.text.strip(),
+                "input_tokens": 0,
+                "output_tokens": 0,
+                "token_usage": {},
+            }
         except Exception as e:
-            return GenerationResult(
-                content=f"GENERATION ERROR: {e}",
-                input_tokens=0,
-                output_tokens=0
-            )
+            return {
+                "content": f"GENERATION ERROR: {e}",
+                "input_tokens": 0,
+                "output_tokens": 0,
+                "token_usage": {},
+            }
 
 # Agent wrapper
 class MyAgent:
@@ -157,7 +149,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("\U0001F527 App starting...")
     demo.launch(debug=True, share=False)