Update app.py

app.py

@@ -4,46 +4,29 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel,WikipediaSearchTool
+from smolagents import LiteLLMModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-
 # --- Basic Agent Definition ---
-
+# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
 
-        model = OpenAIServerModel(
-            model_id="deepseek-chat",  
-            api_key="sk-04da63f756c9468182d278f52e33ad15",
-            api_base="https://api.deepseek.com",  
-            temperature=0.5,
-        )
-        
-        search_tool = DuckDuckGoSearchTool()
-        wiki_search = WikipediaSearchTool()
-       
+        model = LiteLLMModel(
+            model_id="gemini/gemini-2.5-flash-preview-04-17",
+            )
         self.agent = CodeAgent(
-            model = model,      
-            tools=[search_tool,wiki_search] 
+            model=model,
+            tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
+            add_base_tools=True,
         )
+        print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
-        """
-        让 BasicAgent 实例可以像函数一样被调用，处理用户的问题
-        
-        Args:
-            question: 用户提出的问题或请求字符串
-            
-        Returns:
-            str: Agent 生成的回答
-        """
-        print(f"Agent received question (first 50 chars): {question[:50]}...")  
-        fixed_answer = self.agent.run(question)  
-        print(f"Agent returning fixed answer: {fixed_answer}")  
-        return fixed_answer  
+        answer = self.agent.run(question)
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """