Update app.py

app.py

@@ -83,33 +83,41 @@ def flux_krea_generate(
                     seed=seed if seed != -1 else random.randint(1, 1000000000)
                 )
                 
-                # Save the image to a temporary file that Gradio can serve
-                # This creates a local file that Gradio can serve with a public URL
+                # Save the image and return a Gradio-accessible URL
                 if image:
-                    # Generate a unique filename
-                    unique_id = str(uuid.uuid4())[:8]
-                    temp_filename = f"flux_krea_{generation_id}_{unique_id}.png"
-                    
-                    # Save to temp directory
+                    # Save to a temporary file that Gradio can serve
                     temp_file = tempfile.NamedTemporaryFile(
                         delete=False, 
                         suffix=".png", 
-                        prefix=f"flux_krea_{generation_id}_"
+                        prefix=f"flux_krea_mcp_{generation_id}_"
                     )
                     image.save(temp_file.name)
                     temp_file.close()
                     
-                    # For MCP clients, we need to provide a URL that they can access
-                    # Gradio will serve temp files via the /file/ endpoint
-                    file_url = temp_file.name
+                    # Create the Gradio file URL that will be accessible to MCP clients
+                    # This matches the format you saw: /gradio_api/file=<file_path>
+                    gradio_file_url = f"/gradio_api/file={temp_file.name}"
+                    
+                    # For hosted spaces, we also need the full URL
+                    # Try to detect if we're running on Spaces
+                    space_name = os.getenv("SPACE_ID")
+                    if space_name:
+                        # Running on HF Spaces
+                        full_url = f"https://{space_name}.hf.space{gradio_file_url}"
+                    else:
+                        # Running locally
+                        server_port = os.getenv("GRADIO_SERVER_PORT", "7860")
+                        full_url = f"http://localhost:{server_port}{gradio_file_url}"
                     
                     print(f'\033[1mMCP Generation {generation_id} completed with {provider}!\033[0m')
+                    print(f'🌐 Image accessible at: {full_url}')
                     
-                    # Return JSON with both file path and success metadata
+                    # Return JSON with accessible URLs and metadata
                     result = {
                         "success": True,
-                        "image_url": file_url,
-                        "image_path": file_url,
+                        "image_url": full_url,
+                        "gradio_file_url": gradio_file_url,
+                        "local_path": temp_file.name,
                         "generation_id": generation_id,
                         "provider": provider,
                         "model": "black-forest-labs/FLUX.1-Krea-dev",
@@ -125,7 +133,8 @@ def flux_krea_generate(
                         "metadata": {
                             "tool": "flux_krea_generate",
                             "timestamp": str(generation_id),
-                            "mcp_compatible": True
+                            "mcp_compatible": True,
+                            "accessible_url": full_url
                         }
                     }
                     
@@ -152,7 +161,14 @@ def flux_krea_generate(
             "success": False,
             "error": error_message,
             "image_url": None,
-            "generation_id": generation_id
+            "gradio_file_url": None,
+            "local_path": None,
+            "generation_id": generation_id,
+            "metadata": {
+                "tool": "flux_krea_generate",
+                "mcp_compatible": True,
+                "error": True
+            }
         })
 
 # For UI compatibility - this function returns a PIL Image for the Gradio interface
@@ -161,9 +177,11 @@ def flux_krea_generate_ui(*args) -> Optional[Image.Image]:
     result_json = flux_krea_generate(*args)
     try:
         result = json.loads(result_json)
-        if result.get("success") and result.get("image_path"):
-            return Image.open(result["image_path"])
-    except:
+        if result.get("success") and result.get("local_path"):
+            # Return the PIL Image for the UI
+            return Image.open(result["local_path"])
+    except Exception as e:
+        print(f"UI wrapper error: {e}")
         pass
     return None
 
@@ -298,7 +316,9 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css, title="FLUX.1-Krea MCP Server
                 <h3>📡 MCP Server Information</h3>
                 <p><strong>Server Endpoint:</strong> <code>/gradio_api/mcp/sse</code></p>
                 <p><strong>Tool Name:</strong> <code>flux_krea_generate</code></p>
-                <p>This server exposes the image generation function as an MCP tool that returns JSON with image URLs for LLM integration.</p>
+                <p><strong>Image URLs:</strong> Returns accessible Gradio file URLs like <code>/gradio_api/file=&lt;path&gt;</code></p>
+                <p>This server exposes the image generation function as an MCP tool that returns JSON with accessible image URLs for LLM integration.</p>
+                <p><em>✅ Fixed: LLMs can now access generated images via proper Gradio file URLs</em></p>
             </div>
             """)