Update app.py

app.py

@@ -9,26 +9,6 @@ from PIL import Image
 import io
 from Gradio_UI import GradioUI
 
-@tool
-def generate_image(prompt: str) -> str:
-    """
-    Generates an image based on a provided text prompt using an open-source model,
-    saves it locally, and returns the path to render it in the chat interface.
-
-    Args:
-        prompt: A detailed text description of the image to create.
-    """
-    # 1. Fetch image from Hugging Face Inference API
-    client = InferenceClient("black-forest-labs/FLUX.1-schnell")
-    image_bytes = client.text_to_image(prompt)
-    image = Image.open(io.BytesIO(image_bytes))
-    
-    # 2. Save it locally inside your execution workspace
-    output_path = "generated_output.png"
-    image.save(output_path)
-    
-    # 3. Return the string file path. Gradio will intercept this path and display the actual image!
-    return output_path
     
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -84,7 +64,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, cast_spell, generate_image], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, cast_spell], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,