up3

app.py

@@ -17,15 +17,22 @@ from Gradio_UI import GradioUI
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def inference_image_tool(prompt: str, model: str = "black-forest-labs/FLUX.1-dev") -> "Image.Image":
-    """Generate an image from a text prompt using Hugging Face Inference Providers." 
+    """Generate an image from a text prompt using Hugging Face Inference Providers. Return a local file path to a PNG." 
     Args: 
     prompt: Text prompt for the image. 
     model: HF model id (e.g. black-forest-labs/FLUX.1-dev)."""
     token = os.getenv("HF_TOKEN")
     if not token:
         raise RuntimeError("HF_TOKEN is missing. Add it in Space Settings → Secrets.")
+
     client = InferenceClient(provider="auto", api_key=token)
-    return client.text_to_image(prompt, model=model)
+    img = client.text_to_image(prompt, model=model)  # PIL.Image
+
+    out_dir = "/tmp/agent_images"
+    os.makedirs(out_dir, exist_ok=True)
+    out_path = os.path.join(out_dir, f"{uuid.uuid4().hex}.png")
+    img.save(out_path)
+    return out_path
 
 
 @tool