Update app.py

app.py

@@ -7,30 +7,29 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
 
 @tool
-def generate_huggingface_image(prompt: str, model: str = "stabilityai/stable-diffusion-2") -> str:
-    """A tool that generates images using Hugging Face's HfApiModel.
+def get_nbp_exchange_rate(currency: str) -> float:
+    """A tool that fetches the exchange rate of a given currency from the National Bank of Poland (NBP).
     
     Args:
-        prompt: The text description of the image to generate.
-        model: The model to use for image generation (default: stabilityai/stable-diffusion-2).
+        currency: The currency code (e.g., "USD", "EUR") to fetch the exchange rate for.
     
     Returns:
-        URL of the generated image.
+        The exchange rate of the specified currency in PLN.
     """
-        
-    # Send the prompt to the Hugging Face model
-    response = client.text_to_image(model=model, prompt=prompt)
+    import requests
     
-    # Save the response image
-    image_path = "generated_image.png"
-    with open(image_path, "wb") as f:
-        f.write(response)
+    url = f"https://api.nbp.pl/api/exchangerates/rates/A/{currency}/?format=json"
+    response = requests.get(url)
     
-    return image_path
+    if response.status_code == 200:
+        data = response.json()
+        return data["rates"][0]["mid"]
+    else:
+        raise ValueError(f"Failed to fetch exchange rate for {currency}. HTTP Status: {response.status_code}")
 
+# 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
@@ -78,7 +77,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_nbp_exchange_rate, get_current_time_in_timezone, image_generation_tool, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,