Update app.py

app.py

@@ -1,12 +1,11 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import datetime
-import json
 import requests
-import pytz
-import yaml
+import json
+from typing import Dict, Any
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+
 @tool
 def get_btc_price() -> str:
     """Fetches the current Bitcoin price in USD using the CoinGecko API."""
@@ -19,7 +18,6 @@ def get_btc_price() -> str:
     if price is None:
         return '{"error": "Bitcoin price not found"}'
     return f"The current Bitcoin price is ${price:.2f} USD."
-
 @tool
 def get_btc_news() -> Dict[str, Any]:
     """Fetches the latest Bitcoin news using DuckDuckGo Instant Answer API."""
@@ -35,24 +33,18 @@ def get_btc_news() -> Dict[str, Any]:
         return {"error": "Failed to fetch BTC news"}
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetches the current local time in a specified timezone."""
-    try:
-        tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+def analyze_btc_news(news: str) -> str:
+    """Analyzes the provided Bitcoin news content using the Qwen AI model."""
+    model = HfApiModel(
+        max_tokens=1048,
+        temperature=0.5,
+        model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
+        custom_role_conversions=None,
+    )
+    analysis = model.generate(f"Analyze the following Bitcoin news: {news}")
+    return analysis
 
 final_answer = FinalAnswerTool()
-model = HfApiModel(
-    max_tokens=1048,
-    temperature=0.5,
-    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
-    custom_role_conversions=None,
-)
-
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
@@ -60,7 +52,7 @@ with open("prompts.yaml", 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[get_btc_price, get_btc_news, DuckDuckGoSearchTool(), image_generation_tool, final_answer],
+    tools=[get_btc_news, analyze_btc_news, image_generation_tool, final_answer],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -70,4 +62,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()