Update app.py

app.py

@@ -1,33 +1,48 @@
 import os
+import openai
 import gradio as gr
 import requests
 import inspect
 import pandas as pd
-
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool
-
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 # (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 ------
+DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
+DEEPSEEK_API_BASE = os.getenv("DEEPSEEK_API_BASE", "https://api.deepseek.com") 
 class BasicAgent:
     def __init__(self):
-        model = OpenAIServerModel(model_id="deepseek-chat",api_key="sk-04da63f756c9468182d278f52e33ad15", api_base="https://api.deepseek.com")
-
+        print("BasicAgent initialized.")
+        # Initialize the model
+        #model = HfApiModel()
+        if not DEEPSEEK_API_KEY: # 添加一个检查，确保密钥已设置
+            raise ValueError("DEEPSEEK_API_KEY environment variable not set. Please set it before running.")
+
+        openai.api_key = DEEPSEEK_API_KEY  # <--- 步骤 2: 全局设置 API 密钥
+        
+        # 为 openai 库设置基础 URL
+        openai.api_base = DEEPSEEK_API_BASE
+        
+        model = OpenAIServerModel(
+            model_id="deepseek-chat",
+            api_key=DEEPSEEK_API_KEY
+        )
+        # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
-        wiki_search = WikipediaSearchTool()
-
+        # Initialize Agent
         self.agent = CodeAgent(
             model = model,
-            tools=[
-                search_tool,
-            ]
+            tools=[search_tool]
         )
-
     def __call__(self, question: str) -> str:
-        return self.agent.run(question) # type: ignore
+        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
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """