Update app.py

app.py

@@ -11,16 +11,9 @@ from torchvision import transforms
 from huggingface_hub import hf_hub_download, snapshot_download
 import subprocess
 import shutil
-from fastapi import FastAPI, HTTPException, Depends, File, UploadFile
-from fastapi.security import APIKeyHeader
-from fastapi.staticfiles import StaticFiles
-from pydantic import BaseModel
-import uvicorn
-import asyncio
 
 # Install additional dependencies
 subprocess.run("pip install spandrel==0.4.1 --no-deps", shell=True, check=True)
-subprocess.run("pip install fastapi==0.116.1 uvicorn==0.31.0", shell=True, check=True)
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16
@@ -52,143 +45,6 @@ sys.path.append(os.path.join(TRIPOSG_CODE_DIR, "scripts"))
 sys.path.append(MV_ADAPTER_CODE_DIR)
 sys.path.append(os.path.join(MV_ADAPTER_CODE_DIR, "scripts"))
 
-# Initialize FastAPI app
-app = FastAPI()
-
-# Mount static files for serving generated models
-app.mount("/files", StaticFiles(directory=TMP_DIR), name="files")
-
-# API key authentication
-api_key_header = APIKeyHeader(name="X-API-Key")
-VALID_API_KEY = os.getenv("POLYGENIX_API_KEY", "your-secret-api-key")
-
-async def verify_api_key(api_key: str = Depends(api_key_header)):
-    if api_key != VALID_API_KEY:
-        raise HTTPException(status_code=401, detail="Invalid API key")
-    return api_key
-
-# API request model
-class GenerateRequest(BaseModel):
-    seed: int = 0
-    num_inference_steps: int = 50
-    guidance_scale: float = 7.5
-    simplify: bool = True
-    target_face_num: int = DEFAULT_FACE_NUMBER
-
-# Test endpoint
-@app.get("/api/test")
-async def test_endpoint():
-    return {"message": "FastAPI is running"}
-
-# FastAPI endpoint for generating 3D models
-@app.post("/api/generate")
-async def generate_3d_model(request: GenerateRequest, image: UploadFile = File(...), api_key: str = Depends(verify_api_key)):
-    try:
-        # Save uploaded image to temporary directory
-        session_hash = get_random_hex()
-        save_dir = os.path.join(TMP_DIR, session_hash)
-        os.makedirs(save_dir, exist_ok=True)
-        image_path = os.path.join(save_dir, f"input_{get_random_hex()}.png")
-        with open(image_path, "wb") as f:
-            f.write(await image.read())
-
-        # Run the full pipeline
-        image_seg, mesh_path, textured_glb_path = run_full(image_path, req=None)
-
-        # Return the file URL for the textured GLB
-        file_url = f"/files/{session_hash}/{os.path.basename(textured_glb_path)}"
-        return {"file_url": file_url}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-    finally:
-        # Clean up temporary directory
-        if os.path.exists(save_dir):
-            shutil.rmtree(save_dir)
-
-# Gradio-related functions
-HEADER = """
-# 🌌 PolyGenixAI: Craft 3D Worlds with Cosmic Precision
-## Unleash Infinite Creativity with AI-Powered 3D Generation by AnvilInteractive Solutions
-<p style="font-size: 1.1em; color: #A78BFA;">By <a href="https://www.anvilinteractive.com/" style="color: #A78BFA; text-decoration: none; font-weight: bold;">AnvilInteractive Solutions</a></p>
-## 🚀 Launch Your Creation:
-1. **Upload an Image** (clear, single-object images shine brightest)
-2. **Choose a Style Filter** to infuse your unique vision
-3. Click **Generate 3D Model** to sculpt your mesh
-4. Click **Apply Texture** to bring your model to life
-5. **Download GLB** to share your masterpiece
-<p style="font-size: 0.9em; margin-top: 10px; color: #D1D5DB;">Powered by cutting-edge AI and multi-view technology from AnvilInteractive Solutions. Join our <a href="https://www.anvilinteractive.com/community" style="color: #A78BFA; text-decoration: none;">PolyGenixAI Community</a> to connect with creators and spark inspiration.</p>
-<style>
-  @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;700&display=swap');
-  body {
-    background-color: #1A1A1A !important;
-    font-family: 'Inter', sans-serif !important;
-    color: #D1D5DB !important;
-  }
-  .gr-panel {
-    background-color: #2D2D2D !important;
-    border: 1px solid #7C3AED !important;
-    border-radius: 12px !important;
-    padding: 20px !important;
-    box-shadow: 0 4px 10px rgba(124, 58, 237, 0.2) !important;
-  }
-  .gr-button-primary {
-    background: linear-gradient(45deg, #7C3AED, #A78BFA) !important;
-    color: white !important;
-    border: none !important;
-    border-radius: 8px !important;
-    padding: 12px 24px !important;
-    font-weight: 600 !important;
-    transition: transform 0.2s, box-shadow 0.2s !important;
-  }
-  .gr-button-primary:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 4px 12px rgba(124, 58, 237, 0.5) !important;
-  }
-  .gr-button-secondary {
-    background-color: #4B4B4B !important;
-    color: #D1D5DB !important;
-    border: 1px solid #A78BFA !important;
-    border-radius: 8px !important;
-    padding: 10px 20px !important;
-    transition: transform 0.2s !important;
-  }
-  .gr-button-secondary:hover {
-    transform: translateY(-1px) !important;
-    background-color: #6B6B6B !important;
-  }
-  .gr-accordion {
-    background-color: #2D2D2D !important;
-    border-radius: 8px !important;
-    border: 1px solid #7C3AED !important;
-  }
-  .gr-tab {
-    background-color: #2D2D2D !important;
-    color: #A78BFA !important;
-    border: 1px solid #7C3AED !important;
-    border-radius: 8px !important;
-    margin: 5px !important;
-  }
-  .gr-tab:hover, .gr-tab-selected {
-    background: linear-gradient(45deg, #7C3AED, #A78BFA) !important;
-    color: white !important;
-  }
-  .gr-slider input[type=range]::-webkit-slider-thumb {
-    background-color: #7C3AED !important;
-    border: 2px solid #A78BFA !important;
-  }
-  .gr-dropdown {
-    background-color: #2D2D2D !important;
-    color: #D1D5DB !important;
-    border: 1px solid #A78BFA !important;
-    border-radius: 8px !important;
-  }
-  h1, h3 {
-    color: #A78BFA !important;
-    text-shadow: 0 0 10px rgba(124, 58, 237, 0.5) !important;
-  }
-</style>
-"""
-
 # triposg
 from image_process import prepare_image
 from briarmbg import BriaRMBG
@@ -230,7 +86,7 @@ remove_bg_fn = lambda x: remove_bg(x, birefnet, transform_image, DEVICE)
 if not os.path.exists("checkpoints/RealESRGAN_x2plus.pth"):
     hf_hub_download("dtarnow/UPscaler", filename="RealESRGAN_x2plus.pth", local_dir="checkpoints")
 if not os.path.exists("checkpoints/big-lama.pt"):
-    hf_hub_download("Sanster/models", filename="big-lama.pt", local_dir="checkpoints")
+    subprocess.run("wget -P checkpoints/ https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt", shell=True, check=True)
 
 def start_session(req: gr.Request):
     save_dir = os.path.join(TMP_DIR, str(req.session_hash))
@@ -495,6 +351,89 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
 
     return textured_glb_path
 
+HEADER = """
+# 🌌 PolyGenixAI: Craft 3D Worlds with Cosmic Precision
+## Unleash Infinite Creativity with AI-Powered 3D Generation by AnvilInteractive Solutions
+<p style="font-size: 1.1em; color: #A78BFA;">By <a href="https://www.anvilinteractive.com/" style="color: #A78BFA; text-decoration: none; font-weight: bold;">AnvilInteractive Solutions</a></p>
+## 🚀 Launch Your Creation:
+1. **Upload an Image** (clear, single-object images shine brightest)
+2. **Choose a Style Filter** to infuse your unique vision
+3. Click **Generate 3D Model** to sculpt your mesh
+4. Click **Apply Texture** to bring your model to life
+5. **Download GLB** to share your masterpiece
+<p style="font-size: 0.9em; margin-top: 10px; color: #D1D5DB;">Powered by cutting-edge AI and multi-view technology from AnvilInteractive Solutions. Join our <a href="https://www.anvilinteractive.com/community" style="color: #A78BFA; text-decoration: none;">PolyGenixAI Community</a> to connect with creators and spark inspiration.</p>
+<style>
+  @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;700&display=swap');
+  body {
+    background-color: #1A1A1A !important;
+    font-family: 'Inter', sans-serif !important;
+    color: #D1D5DB !important;
+  }
+  .gr-panel {
+    background-color: #2D2D2D !important;
+    border: 1px solid #7C3AED !important;
+    border-radius: 12px !important;
+    padding: 20px !important;
+    box-shadow: 0 4px 10px rgba(124, 58, 237, 0.2) !important;
+  }
+  .gr-button-primary {
+    background: linear-gradient(45deg, #7C3AED, #A78BFA) !important;
+    color: white !important;
+    border: none !important;
+    border-radius: 8px !important;
+    padding: 12px 24px !important;
+    font-weight: 600 !important;
+    transition: transform 0.2s, box-shadow 0.2s !important;
+  }
+  .gr-button-primary:hover {
+    transform: translateY(-2px) !important;
+    box-shadow: 0 4px 12px rgba(124, 58, 237, 0.5) !important;
+  }
+  .gr-button-secondary {
+    background-color: #4B4B4B !important;
+    color: #D1D5DB !important;
+    border: 1px solid #A78BFA !important;
+    border-radius: 8px !important;
+    padding: 10px 20px !important;
+    transition: transform 0.2s !important;
+  }
+  .gr-button-secondary:hover {
+    transform: translateY(-1px) !important;
+    background-color: #6B6B6B !important;
+  }
+  .gr-accordion {
+    background-color: #2D2D2D !important;
+    border-radius: 8px !important;
+    border: 1px solid #7C3AED !important;
+  }
+  .gr-tab {
+    background-color: #2D2D2D !important;
+    color: #A78BFA !important;
+    border: 1px solid #7C3AED !important;
+    border-radius: 8px !important;
+    margin: 5px !important;
+  }
+  .gr-tab:hover, .gr-tab-selected {
+    background: linear-gradient(45deg, #7C3AED, #A78BFA) !important;
+    color: white !important;
+  }
+  .gr-slider input[type=range]::-webkit-slider-thumb {
+    background-color: #7C3AED !important;
+    border: 2px solid #A78BFA !important;
+  }
+  .gr-dropdown {
+    background-color: #2D2D2D !important;
+    color: #D1D5DB !important;
+    border: 1px solid #A78BFA !important;
+    border-radius: 8px !important;
+  }
+  h1, h3 {
+    color: #A78BFA !important;
+    text-shadow: 0 0 10px rgba(124, 58, 237, 0.5) !important;
+  }
+</style>
+"""
+
 # Gradio interface
 with gr.Blocks(title="PolyGenixAI", css="body { background-color: #1A1A1A; } .gr-panel { background-color: #2D2D2D; }") as demo:
     gr.Markdown(HEADER)
@@ -595,13 +534,5 @@ with gr.Blocks(title="PolyGenixAI", css="body { background-color: #1A1A1A; } .gr
     demo.load(start_session)
     demo.unload(end_session)
 
-# Run FastAPI and Gradio concurrently
-async def run_servers():
-    config = uvicorn.Config(app=app, host="0.0.0.0", port=8000)
-    server = uvicorn.Server(config)
-    fastapi_task = asyncio.create_task(server.serve())
-    demo.launch(server_name="0.0.0.0", server_port=7860)
-    await fastapi_task
-
 if __name__ == "__main__":
-    asyncio.run(run_servers())
+    demo.launch(server_name="0.0.0.0", server_port=7860)