Update app.py

app.py

@@ -1,57 +1,81 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+import datetime
+import requests
+import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from tools.visit_webpage import VisitWebpageTool
-from typing import Optional
-from PIL import Image
-from io import BytesIO
 
 from Gradio_UI import GradioUI
 
+# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# @tool
+# def search_ddgs(query:str, max_results:int=5)-> list: #it's import to specify the return type
+#     #Keep this format for the description / args / args description but feel free to modify the tool
+#     """A tool that searches the internet to answer user's question and returns top research results.
+    
+#     Args:
+#         query: The search query
+#         max_results: Number of results to return. Maximum number is 5. 
+
+#     Returns:
+#     List of dictionaries with title, URL, and description.
+#     """
+#     results = []
+#     with DuckDuckGoSearchTool as ddgs:
+#         for result in ddgs.text(query, max_results = max_results):
+#             results.append({
+#                 "title": result["title"],
+#                 "url": result["href"],
+#                 "snippet":result["body"]
+#             })
+    
+#     return results
+
 @tool
-def gimme_a_meme(subreddit: str = 'me_irl') -> Image:
-    """A tool that grabs a random meme from a given subreddit
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
     Args:
-        subreddit: a string representing a valid subreddit (e.g., 'dankmemes', 'memes', 'me_irl') default value: 'me_irl'
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
-        import requests
-        url = f"https://meme-api.com/gimme/{subreddit}"
-        response = requests.get(url)
-        response.raise_for_status()
+        # Create timezone object
+        tz = pytz.timezone(timezone)
+        # Get current time in that 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)}"
 
-        meme_data = response.json()
-        image_url = meme_data.get("url")
-        image_response = requests.get(image_url)
-        image_response.raise_for_status()
 
-        img = Image.open(BytesIO(image_response.content))
-        return img
-    except requests.exceptions.RequestException as e:
-        print(f"Error fetching meme: {e}")
-        return None
+final_answer = FinalAnswerTool()
 
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-final_answer = FinalAnswerTool()
-visit_webpage = VisitWebpageTool()
+# Search tool
+search_tool =  DuckDuckGoSearchTool()
+# search_tool =  DuckDuckGoSearchTool(max_results=3, output_format="json")
+
+# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.1,
-    #model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud",
-    custom_role_conversions=None
+max_tokens=2096,
+temperature=0.5,
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+custom_role_conversions=None,
 )
 
-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)
-
+    
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool(), gimme_a_meme, image_generation_tool, visit_webpage],
-    max_steps=4,
+    tools=[final_answer, get_current_time_in_timezone, search_tool, image_generation_tool], ## add your tools here (don't remove final answer)
+    max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,