gemini 2 flash as first agent

app.py

@@ -4,9 +4,9 @@ import requests
 import inspect
 import pandas as pd
 
-from openai import OpenAI
+from google import genai
 
-client = OpenAI(api_key=os.getenv("API_KEY"), base_url="https://api.deepseek.com")
+client = genai.Client(api_key=os.getenv("API_KEY"))
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -20,18 +20,13 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
 
-        response = client.chat.completions.create(
-    model="deepseek-chat",
-    messages=[
-        {"role": "system", "content": "You are a helpful assistant"},
-        {"role": "user", "content": question},
-    ],
-    stream=False
-)   
-        deepseek_answer = response.choices[0].message.content
+        response = client.models.generate_content(
+    model="gemini-2.0-flash", contents=question
+)
+        gemini_flash_answer = response.text
         #fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {deepseek_answer}")
-        return deepseek_answer
+        print(f"Agent returning fixed answer: {gemini_flash_answer}")
+        return gemini_flash_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

requirements.txt

@@ -1,3 +1,3 @@
 gradio
 requests
-openai
+google-genai