Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -6,17 +6,42 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+import os
+
+from PIL import Image
+import requests
+from io import BytesIO
+
+
+print(os.getcwd())
+current_directory = os.path.dirname(os.path.abspath(__file__))
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #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 does nothing yet 
+# it's import to specify the return type
+def get_random_image(width: int, height: int) -> Image.Image:
+    """Retrieve a random image with the desired dimensions.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        width: Desired width of the image.
+        height: Desired height of the image.
     """
-    return "What magic will you build ?"
+    if width <= 0 or height <= 0:
+        raise ValueError("Width and height must be positive integers.")
+
+    # Generate the URL for a random image
+    url = f"https://picsum.photos/{width}/{height}"
+
+    # Download the image
+    response = requests.get(url)
+    if response.status_code == 200:
+        # Open the image
+        image = Image.open(BytesIO(response.content))
+        return image
+    else:
+        raise Exception("Failed to download the image.")
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -37,25 +62,32 @@ def get_current_time_in_timezone(timezone: str) -> str:
 final_answer = FinalAnswerTool()
 
 # 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_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
+# model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 
 model = HfApiModel(
-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,
+    max_tokens=2096,
+    temperature=0.5,
+    # it is possible that this model may be overloaded
+    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+    custom_role_conversions=None,
 )
 
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+image_generation_tool = load_tool(
+    "agents-course/text-to-image", trust_remote_code=True)
 
-with open("prompts.yaml", 'r') as stream:
+prompts_file = f"prompts.yaml"
+prompts_path = f"{current_directory}/{prompts_file}"
+
+with open(prompts_path, 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    # add your tools here (don't remove final answer)
+    tools=[get_random_image, get_current_time_in_timezone, final_answer],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -66,4 +98,4 @@ agent = CodeAgent(
 )
 
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()