fix: modify prompt templates and add web tools

app.py

@@ -10,7 +10,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
+# 💡 導入 smolagents 內建的系統提示詞結構
+from smolagents.agents import SYSTEM_PROMPT_WITH_CODE 
 import os
 import spaces 
 
@@ -21,19 +23,39 @@ def fake_gpu_function():
 class BasicAgent:
     def __init__(self):
         self.openrouter_key = os.getenv("OPENROUTER_API_KEY")
-        
-        # 💡 將預設模型改為 OpenRouter 提供的免費 Llama 3.1 模型
-        self.model_id = os.getenv("HF_MODEL_ID", "openrouter/meta-llama/llama-3.1-8b-instruct:free")
+        self.model_id = os.getenv("HF_MODEL_ID", "google/gemini-2.5-flash")
 
     def __call__(self, question: str) -> str:
         try:
-            model = LiteLLMModel(
+            model = OpenAIServerModel(
                 model_id=self.model_id,
                 api_key=self.openrouter_key,
                 api_base="https://openrouter.ai/api/v1"
             )
+            
             search_tool = DuckDuckGoSearchTool()
-            agent = CodeAgent(tools=[search_tool], model=model)
+            visit_tool = VisitWebpageTool()
+            tools = [search_tool, visit_tool]
+            
+            # 💡 精準修正：灌輸 Agent 解題邏輯，並提醒它留意當前目錄下的附加檔案（如 .xlsx, .mp3, .py 等）
+            custom_prompt = (
+                "You are an expert AI Assistant capable of running Python code and web searches to solve complex tasks.\n"
+                "When a question requires factual data, always double check the facts using search tools.\n"
+                "If the first search yields no good results, try different keywords.\n"
+                "If the question mentions an attached file (e.g., .mp3, .xlsx, .py), check the current directory, "
+                "and write Python code to read/analyze the file content directly.\n"
+                "Be concise and output only the final correct answer without unnecessary prose."
+            )
+            
+            # 💡 正確傳遞 Prompt 的方式：替換預設的 system_prompt 範本
+            prompt_templates = {"system_prompt": SYSTEM_PROMPT_WITH_CODE + "\n\nSpecific Rules:\n" + custom_prompt}
+            
+            agent = CodeAgent(
+                tools=tools, 
+                model=model, 
+                max_steps=8,
+                prompt_templates=prompt_templates  # 💡 修正參數名稱
+            )
             
             return str(agent.run(question))
         except Exception as e: