Update app.py

app.py

@@ -35,7 +35,7 @@ BACKUP_DIR = "generated_images_backup"
 os.makedirs(BACKUP_DIR, exist_ok=True)
 
 # Initialize FastAPI
-app = FastAPI()
+app = FastAPI(title="Image Generator API")
 
 # Global model cache
 model = None
@@ -130,13 +130,13 @@ def generate_image(prompt, project_id=None, scene_num=1):
         raise
 
 # =============================================
-# API ENDPOINTS (Define these BEFORE mounting Gradio)
+# API ENDPOINTS
 # =============================================
 class GenerateRequest(BaseModel):
     prompt: str
     project_id: Optional[str] = None
 
-@app.post("/generate")
+@app.post("/api/generate")  # Changed to /api/generate
 async def generate(request: GenerateRequest):
     """Generate a single image"""
     try:
@@ -149,7 +149,7 @@ async def generate(request: GenerateRequest):
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
-@app.get("/health")
+@app.get("/api/health")  # Changed to /api/health
 async def health():
     """Health check"""
     return {
@@ -158,13 +158,13 @@ async def health():
         "hf_dataset": DATASET_ID if HF_TOKEN else "disabled"
     }
 
-@app.get("/api-test")
+@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"]}
+    """Simple test endpoint"""
+    return {"message": "API is working!"}
 
 # =============================================
-# GRADIO INTERFACE (Mounted at /ui instead of root)
+# GRADIO INTERFACE
 # =============================================
 def gradio_generate(prompt):
     if not prompt:
@@ -172,44 +172,68 @@ def gradio_generate(prompt):
     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")
+# Create Gradio interface
+with gr.Blocks(title="Image Generator", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎨 Image Generator")
+    gr.Markdown("Generate images using Stable Diffusion")
+    
+    with gr.Row():
+        with gr.Column():
+            prompt_input = gr.Textbox(
+                label="Prompt",
+                placeholder="Enter your prompt...",
+                lines=3
+            )
+            generate_btn = gr.Button("Generate", variant="primary")
+        
+        with gr.Column():
+            output_image = gr.Image(label="Generated Image")
+    
+    generate_btn.click(
+        fn=gradio_generate,
+        inputs=[prompt_input],
+        outputs=[output_image]
+    )
+    
+    gr.Markdown("---")
+    gr.Markdown("### API Endpoints")
+    gr.Markdown("""
+    - `GET /api/health` - Health check
+    - `POST /api/generate` - Generate image
+    - `GET /api/test` - Test endpoint
+    """)
 
-# Root endpoint returns API info
+# =============================================
+# ROOT ENDPOINT
+# =============================================
 @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)"
+        "name": "Image Generator",
+        "version": "1.0.0",
+        "api_endpoints": {
+            "health": "GET /api/health",
+            "generate": "POST /api/generate",
+            "test": "GET /api/test"
         },
+        "ui": "Gradio interface available at /ui",
         "status": "running"
     }
 
 # =============================================
-# MAIN
+# MAIN - Hugging Face Spaces Deployment
 # =============================================
 if __name__ == "__main__":
     import uvicorn
+    
     print("\n" + "=" * 60)
-    print("🌐 Server starting on port 7860")
-    print("📊 API endpoints:")
-    print("   - GET  /")
-    print("   - GET  /health")
-    print("   - POST /generate")
-    print("   - GET  /api-test")
-    print("   - UI   /ui (Gradio interface)")
+    print("🌐 Deploying on Hugging Face Spaces")
+    print("📚 API endpoints: /api/*")
+    print("🎨 UI: /ui")
     print("=" * 60)
     
+    # Mount Gradio at /ui path
+    app = gr.mount_gradio_app(app, demo, path="/ui")
+    
+    # Run the combined app
     uvicorn.run(app, host="0.0.0.0", port=7860)