Added Gemini API, improved agent prompt

app.py

@@ -7,13 +7,18 @@ import pandas as pd
 import asyncio
 import aiohttp
 
-from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, CODE_SYSTEM_PROMPT
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+MODEL_ID = "gemini-2.5-pro-preview-06-05" 
+SYSTEM_PROMPT = f"""${CODE_SYSTEM_PROMPT}\n\n
+Give as precise answer as possible. Do not include extra explanation, thoughts or questions. 
+Include just answer to the specific question.
+"""
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized to use smolagents.")
@@ -22,13 +27,16 @@ class BasicAgent:
         try:    
             search_tool = DuckDuckGoSearchTool()
             
-            llm_model = HfApiModel(
-                model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+            llm_model = OpenAIServerModel(
+                model_id=MODEL_ID,
+                api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+                api_key=GOOGLE_API_KEY,
+                temperature=0.7
             )
-            
             self.agent_executor = CodeAgent(
                 model=llm_model,
                 tools=[search_tool],
+                system_prompt=SYSTEM_PROMPT,
             )
             print("smolagents AgentExecutor initialized successfully with WebSearchTool.")
 
@@ -56,7 +64,6 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
     """
-    # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
 
     if profile: