Update app.py

app.py

@@ -63,14 +63,31 @@ def generate_prompts_for_object(object_name):
         "future": f"Show a futuristic version of a {object_name}, by predicting advanced features and futuristic design."
     }
 
-
+'''
 image_generation_tool = Tool.from_space(
     "KingNish/Realtime-FLUX",
     api_name="/predict",  # Optional if there's only one endpoint
     name="image_generator",
     description="Generate an image from a prompt"
 )
-
+'''
+import requests
+from smolagents import Tool
+
+def flux_proxy(user_prompt: str):
+    # Call the Hugging Face Space API directly
+    url = "https://black-forest-labs-flux-1-schnell.hf.space/run/infer"
+    headers = {"Authorization": f"Bearer {token}"}
+    response = requests.post(url, headers=headers, json={"data": [user_prompt]})
+    response.raise_for_status()
+    output_url = response.json()["data"][0]  # Usually a URL or base64 image
+    return output_url  # Adjust this based on your agent's expectations
+
+image_generation_tool = Tool.from_function(
+    fn=flux_proxy,
+    name="image_generator",
+    description="Generate an image from a prompt"
+)
 
 # =========================================================
 # Tool and Agent Initialization