Update app.py

app.py

@@ -1,15 +1,12 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool,OpenAIServerModel
+from smolagents import CodeAgent, tool, OpenAIServerModel, FinalAnswerTool
 import datetime
-import requests
 import pytz
-import yaml
-from tools.final_answer import FinalAnswerTool
+import yfinance as yf
 import os
 from Gradio_UI import GradioUI
 import gradio as gr
-import requests
-import yfinance as yf
 
+# Fetch the API key from the environment variable (if needed)
 API_KEY = os.getenv("KEY")
 
 @tool
@@ -27,6 +24,7 @@ def get_currency_rate(currency_pair: str) -> str:
         return f"Current {currency_pair} rate: {rate}"
     except Exception as e:
         return f"Error fetching rate: {str(e)}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -42,25 +40,27 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+# Set up the model
 model = OpenAIServerModel(
     model_id="deepseek-chat",
     api_base="https://api.deepseek.com",
     api_key=API_KEY
 )
 
-# Load your custom tool and others
-currency_tool = get_currency_rate
+# Load tools
 final_answer = FinalAnswerTool()
+currency_tool = get_currency_rate
+time_tool = get_current_time_in_timezone
 
+# Create the agent
 agent = CodeAgent(
     model=model,
-    tools=[currency_tool, final_answer],  # Include your tool here
+    tools=[currency_tool, time_tool, final_answer],  # Include your tools here
     max_steps=3,
     verbosity_level=1
 )
 
-#GradioUI(agent).launch()
-
+# Add a Markdown section for example usage
 examples = """
 ### Example Usage
 1. **Currency Rate**: Ask for the exchange rate of a currency pair.
@@ -77,9 +77,8 @@ def launch_with_examples(agent):
     with gr.Blocks() as demo:
         gr.Markdown("# Currency and Time Agent")
         gr.Markdown(examples)
-        GradioUI(agent).render()
+        GradioUI(agent).launch()  # Use launch() instead of render()
     demo.launch()
 
 # Launch the app
-launch_with_examples(agent)
-#GradioUI(agent).launch()
+launch_with_examples(agent)