Update app.py

app.py

@@ -1,33 +1,18 @@
-from smolagents import CodeAgent, load_tool, tool, HfApiModel
-import datetime
-import pytz
-import yaml
+import gradio as gr
+import pandas as pd
+from smolagents import CodeAgent, load_tool
 from tools.final_answer import FinalAnswerTool
+import yaml
 
-# ===========================
-# 自訂工具示範
-# ===========================
-@tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """
-    A tool that does nothing yet.
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    Returns:
-        str: a placeholder string
-    """
-    return "What magic will you build?"
+# -----------------------------
+# 載入工具
+# -----------------------------
+final_answer = FinalAnswerTool()
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-@tool
+# 你自定義工具
 def get_current_time_in_timezone(timezone: str) -> str:
-    """
-    A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York')
-    Returns:
-        str: current local time in that timezone
-    """
+    import datetime, pytz
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -35,42 +20,59 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# ===========================
-# Final Answer Tool
-# ===========================
-final_answer = FinalAnswerTool()
-
-# ===========================
-# 加載其他工具
-# ===========================
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# ===========================
-# 載入 Prompt Templates
-# ===========================
+# -----------------------------
+# 載入 prompt templates
+# -----------------------------
 with open("prompts.yaml", "r") as f:
     prompt_templates = yaml.safe_load(f)
 
-# ===========================
-# 建立 Agent
-# ===========================
-model = HfApiModel(
+# -----------------------------
+# 初始化 Agent
+# -----------------------------
+agent = CodeAgent(
+    tools=[final_answer, image_generation_tool, get_current_time_in_timezone],
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    max_tokens=2096,
+    max_tokens=2048,
     temperature=0.5,
-    custom_role_conversions=None
 )
 
-agent = CodeAgent(
-    model=model,
-    tools=[my_custom_tool, get_current_time_in_timezone, image_generation_tool, final_answer],
-    prompt_templates=prompt_templates
-)
+# -----------------------------
+# 模擬問題清單（可換成你的 API fetch）
+# -----------------------------
+questions = [
+    {"task_id": "q1", "question": "What is 2+2?"},
+    {"task_id": "q2", "question": "Get current time in New York"},
+    {"task_id": "q3", "question": "Generate an image of a cat riding a skateboard"}
+]
+
+# -----------------------------
+# Run agent
+# -----------------------------
+def run_agent():
+    results = []
+    correct_count = 0
+    for q in questions:
+        answer = agent(q["question"])
+        results.append({"ID": q["task_id"], "Question": q["question"], "Answer": answer})
+        # 這裡假設你有正確答案可以比對
+        # if answer == q["answer"]:
+        #     correct_count += 1
+
+    df = pd.DataFrame(results)
+    score = round((correct_count / len(questions)) * 100, 2)
+    status = f"📊 Score: {score}% ({correct_count}/{len(questions)} correct)"
+    return status, df
+
+# -----------------------------
+# Gradio UI
+# -----------------------------
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎯 Hybrid SmolAgent Demo")
+    run_btn = gr.Button("🚀 Run Hybrid Agent", variant="primary", size="lg")
+    status_box = gr.Textbox(label="📊 Results", lines=2, interactive=False)
+    results_table = gr.DataFrame(label="Questions & Answers", wrap=True)
+
+    run_btn.click(fn=run_agent, outputs=[status_box, results_table])
 
-# ===========================
-# 測試 Agent
-# ===========================
 if __name__ == "__main__":
-    question = "What time is it in Tokyo?"
-    answer = agent(question)
-    print(answer)
+    demo.launch(debug=True)