Update app.py

app.py

@@ -130,7 +130,7 @@ def generate_image(prompt, project_id=None, scene_num=1):
         raise
 
 # =============================================
-# API ENDPOINTS
+# API ENDPOINTS (Define these BEFORE mounting Gradio)
 # =============================================
 class GenerateRequest(BaseModel):
     prompt: str
@@ -158,43 +158,58 @@ async def health():
         "hf_dataset": DATASET_ID if HF_TOKEN else "disabled"
     }
 
+@app.get("/api-test")
+async def api_test():
+    """Simple test endpoint to verify API is working"""
+    return {"message": "API is working!", "endpoints": ["/health", "/generate", "/api-test"]}
+
 # =============================================
-# GRADIO INTERFACE (Optional UI)
+# GRADIO INTERFACE (Mounted at /ui instead of root)
 # =============================================
-if 'gradio' in globals():
-    def gradio_generate(prompt):
-        if not prompt:
-            return None
-        result = generate_image(prompt)
-        return result["image"]
-    
-    iface = gr.Interface(
-        fn=gradio_generate,
-        inputs=gr.Textbox(label="Prompt", placeholder="Enter your prompt..."),
-        outputs=gr.Image(label="Generated Image"),
-        title="Image Generator",
-        description="Generate images with Stable Diffusion"
-    )
-    
-    # Mount Gradio
-    gr.mount_gradio_app(app, iface, path="/")
-else:
-    @app.get("/")
-    async def root():
-        return {"message": "API is running. Use /generate endpoint"}
+def gradio_generate(prompt):
+    if not prompt:
+        return None
+    result = generate_image(prompt)
+    return result["image"]
+
+iface = gr.Interface(
+    fn=gradio_generate,
+    inputs=gr.Textbox(label="Prompt", placeholder="Enter your prompt..."),
+    outputs=gr.Image(label="Generated Image"),
+    title="Image Generator",
+    description="Generate images with Stable Diffusion"
+)
+
+# Mount Gradio at /ui path instead of root
+gr.mount_gradio_app(app, iface, path="/ui")
+
+# Root endpoint returns API info
+@app.get("/")
+async def root():
+    return {
+        "message": "Image Generator API",
+        "endpoints": {
+            "health": "GET /health",
+            "generate": "POST /generate",
+            "api_test": "GET /api-test",
+            "ui": "GET /ui (Gradio interface)"
+        },
+        "status": "running"
+    }
 
 # =============================================
-# MAIN - CORRECTED VERSION
+# MAIN
 # =============================================
 if __name__ == "__main__":
     import uvicorn
     print("\n" + "=" * 60)
     print("🌐 Server starting on port 7860")
     print("📊 API endpoints:")
-    print("   - POST /generate")
+    print("   - GET  /")
     print("   - GET  /health")
-    print("   - GET  / (root)")
+    print("   - POST /generate")
+    print("   - GET  /api-test")
+    print("   - UI   /ui (Gradio interface)")
     print("=" * 60)
     
-    # Make sure app is properly configured
     uvicorn.run(app, host="0.0.0.0", port=7860)