Update app.py

app.py

@@ -6,6 +6,10 @@ import cv2
 from PIL import Image
 import spaces
 
+from fastapi import FastAPI, UploadFile, Form
+from fastapi.responses import StreamingResponse
+import io
+
 
 # 🌟 Auto-detect device (CPU/GPU)
 device = "cuda"
@@ -130,5 +134,29 @@ with gr.Blocks() as demo:
         outputs=[edge_output, result_output]
     )
 
+# Define your own FastAPI app for external access
+external_api = FastAPI()
+
+# Add your custom POST route (for Rhino/Grasshopper)
+@external_api.post("/run_external")
+async def run_external(prompt: str = Form(...), image: UploadFile = Form(...)):
+    contents = await image.read()
+    img = Image.open(io.BytesIO(contents)).convert("RGB")
+    # Use your generate_image function (you already have it)
+    _, result = generate_image(prompt, img, 100, 150, 0.7, 7.5, 0.5)
+
+    # Return image as PNG stream
+    buf = io.BytesIO()
+    result.save(buf, format="PNG")
+    buf.seek(0)
+    return StreamingResponse(buf, media_type="image/png")
+
+# 🔁 Mount Gradio under the FastAPI app
+external_api.mount("/", demo)  # Gradio stays at root (/) in the browser
+
+# This is the app Hugging Face expects
+app = external_api
+
+    
 # 🚀 Launch the app
-demo.launch()
+#demo.launch()