Update app.py

app.py

@@ -1,8 +1,5 @@
-import yaml
+from smolagents import CodeAgent, tool
 import requests
-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:
@@ -17,22 +14,19 @@ def get_btc_price() -> str:
         return '{"error": "Bitcoin price not found"}'
     return f"The current Bitcoin price is ${price:.2f} USD."
 
-final_answer = FinalAnswerTool()
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
+# Initialize the agent with the get_btc_price tool
 agent = CodeAgent(
-    model=model,
-    tools=[get_btc_price, final_answer],
+    model=None,  # Specify your model here
+    tools=[get_btc_price],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=None  # Define your prompt templates if needed
 )
 
+# Launch the agent (ensure Gradio_UI is properly set up)
+from Gradio_UI import GradioUI
 GradioUI(agent).launch()