Update app.py

app.py

@@ -4,7 +4,8 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from openai import OpenAI
+
+import google.generativeai as genai
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 
 # System prompt used by the agent
@@ -16,35 +17,70 @@ 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 model wrapper
 class GeminiFlashModel:
-    def __init__(self, model_id="gemini-1.5-flash"):
-        self.client = OpenAI(
-            api_key=os.getenv("GEMINI_API_KEY"),
-            base_url="https://generativelanguage.googleapis.com/v1beta/openai/"
-        )
-        self.model_id = model_id
+    def __init__(self, model_id="gemini-1.5-flash", api_key=None):
+        genai.configure(api_key=api_key or os.getenv("GEMINI_API_KEY"))
+        self.model = genai.GenerativeModel(model_id)
         self.system_prompt = SYSTEM_PROMPT
 
-    def generate(self, messages):
-        # Ensure system prompt is present
+    # Accept stop_sequences explicitly to avoid unexpected kwarg errors
+    def generate(self, messages, stop_sequences=None, **kwargs):
+        if not isinstance(messages, list) or not all(isinstance(m, dict) for m in messages):
+            raise TypeError("Expected 'messages' to be a list of dicts")
+
         if not any(m.get("role") == "system" for m in messages):
             messages = [{"role": "system", "content": self.system_prompt}] + messages
 
-        response = self.client.chat.completions.create(
-            model=self.model_id,
-            messages=messages
-        )
-        # Return the generated content string directly
-        return response.choices[0].message.content
+        prompt = ""
+        for m in messages:
+            role = m["role"].capitalize()
+            content = m["content"]
+            prompt += f"{role}: {content}\n"
 
+        try:
+            # Note: genai.GenerativeModel.generate_content may not support stop_sequences
+            response = self.model.generate_content(prompt)
+            return GenerationResult(
+                content=response.text.strip(),
+                token_usage={},  # you can extend if API provides token info
+                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
+            )
+
+# Agent wrapper
 class MyAgent:
     def __init__(self):
-        self.model = GeminiFlashModel()
+        self.model = GeminiFlashModel(model_id="gemini-1.5-flash")
         self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=self.model)
 
     def __call__(self, question: str) -> str:
-        return self.agent.run(question)
-
+        # The agent.run expects a string answer
+        result = self.agent.run(question)
+        # If result is GenerationResult or dict-like, convert to string
+        if hasattr(result, "content"):
+            return result.content
+        elif isinstance(result, dict):
+            return result.get("content", str(result))
+        else:
+            return str(result)
+
+# Main evaluation function
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
@@ -113,6 +149,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
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""
@@ -136,4 +173,3 @@ if __name__ == "__main__":
 
 
 
-