Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 
 
 # (Keep Constants as is)
@@ -13,11 +13,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
-coder_model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
+# coder_model = HfApiModel(
+#     max_tokens=2096,
+#     temperature=0.5,
+#     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+#     custom_role_conversions=None,
+# )
+
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise RuntimeError("Please set the GEMINI_API_KEY environment variable")
+
+coder_model = OpenAIServerModel(
+    model_id="gemini-2.5-flash",                                          # or gemini-2.5-flash, gemini-2.5-pro, etc.
+    api_base="https://generativelanguage.googleapis.com/v1beta/openai/",  # Gemini’s OpenAI-compatible URL
+    api_key=GEMINI_API_KEY,
 )
 
 agent = CodeAgent(