File size: 2,820 Bytes
2cfffe2 80d56e2 3d9712d da9c720 2cfffe2 da9c720 80d56e2 da9c720 80d56e2 da9c720 80d56e2 da9c720 80d56e2 da9c720 80d56e2 da9c720 80d56e2 da9c720 80d56e2 da9c720 80d56e2 da9c720 2cfffe2 da9c720 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | """
Alture AI — Hugging Face ZeroGPU Production Backend Entrypoint
============================================================
Official Hugging Face ZeroGPU production runner using user's proven monkeypatching pattern.
"""
import os
import sys
import torch
import gradio as gr
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import Response
try:
import spaces
@spaces.GPU
def dummy_gpu_function():
return "ZeroGPU is active and ready"
except ImportError:
def dummy_gpu_function():
return "Local CPU mode"
from deployment.backend.main import app as main_fastapi_app
# Create Gradio demo to satisfy ZeroGPU compiler
with gr.Blocks(title="Alture AI Backend API") as demo:
gr.Markdown("# Alture AI — Production ZeroGPU Backend Engine")
gr.Markdown("FastAPI REST endpoints available under `/api/v1` and `/v1`.")
btn = gr.Button("⚡ Verify ZeroGPU Pipeline", variant="primary")
out = gr.Textbox(label="Pipeline Output")
btn.click(dummy_gpu_function, outputs=out)
# Monkeypatch Gradio's internal FastAPI app creator (proven portfolio pattern)
original_create_app = gr.routes.App.create_app
def custom_create_app(*args, **kwargs):
app = original_create_app(*args, **kwargs)
# Configure CORS & preflight middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.middleware("http")
async def cors_preflight_middleware(request, call_next):
if request.method == "OPTIONS":
return Response(
status_code=200,
headers={
"Access-Control-Allow-Origin": "*",
"Access-Control-Allow-Methods": "*",
"Access-Control-Allow-Headers": "*",
}
)
response = await call_next(request)
response.headers["Access-Control-Allow-Origin"] = "*"
response.headers["Access-Control-Allow-Methods"] = "*"
response.headers["Access-Control-Allow-Headers"] = "*"
return response
# Inject main FastAPI app router
app.include_router(main_fastapi_app.router)
# Reorder routes so /api and /v1 routes take precedence
api_routes = []
other_routes = []
for route in app.router.routes:
path = getattr(route, 'path', '')
if path.startswith("/api") or path.startswith("/v1") or path.startswith("/health"):
api_routes.append(route)
else:
other_routes.append(route)
app.router.routes = api_routes + other_routes
return app
gr.routes.App.create_app = custom_create_app
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860)
|