Update app.py

app.py

@@ -10,16 +10,25 @@ import io
 from Gradio_UI import GradioUI
 
 @tool
-def generate_image(prompt: str) -> Image.Image:
+def generate_image(prompt: str) -> str:
     """
-    Generates an image based on a provided text prompt using an open-source model.
+    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)
-    return Image.open(io.BytesIO(image_bytes))
+    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
@@ -31,9 +40,9 @@ def cast_spell(mana:int)-> str: #it's import to specify the return type
     """
     if mana > 0:
 
-        if 10 > mana > 5:
+        if 10 > mana >= 5:
             return "FIREBALL"
-        elif 15 > mana > 10:
+        elif 15 > mana >= 10:
             return "THUNDERBALL"
 
     return "AIRBALL"