fastapi added

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.10-slim
+
+# System deps for OpenCV and others
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    ffmpeg libsm6 libxext6 git \
+    && rm -rf /var/lib/apt/lists/*
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    OMP_NUM_THREADS=4
+
+WORKDIR /app
+
+COPY requirements.txt /app/requirements.txt
+RUN pip install --no-cache-dir -r /app/requirements.txt
+
+COPY . /app
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+
+

README.md

@@ -1,12 +1,9 @@
 ---
-title: Face Upscale
+title: Face Upscale API
 emoji: 🚀
 colorFrom: purple
 colorTo: purple
-sdk: gradio
-sdk_version: 5.45.0
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -25,6 +25,7 @@ os.makedirs('input', exist_ok=True)
 from fastapi import FastAPI, UploadFile, File, Form, Depends, HTTPException, status
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from fastapi.responses import FileResponse, JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
 
 # Mongo imports (optional)
 _mongo_client = None
@@ -843,6 +844,13 @@ if __name__ == "__main__":
     os.makedirs('input', exist_ok=True)
 
 fastapi_app = FastAPI(title="Face Upscale API")
+fastapi_app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 _bearer_scheme = HTTPBearer(auto_error=False)
 _api_bearer_token = os.getenv("API_BEARER_TOKEN", "changeme")
@@ -976,5 +984,24 @@ def download(filename: str, _: bool = Depends(_verify_bearer_token)):
             return FileResponse(p, filename=safe_name)
     raise HTTPException(status_code=404, detail="File not found")
 
-# Expose FastAPI app for Uvicorn/HF Spaces
+# Optional: mount Gradio UI if enabled, otherwise run API-only
+ENABLE_GRADIO_UI = os.getenv("ENABLE_GRADIO_UI", "0") == "1"
+if ENABLE_GRADIO_UI:
+    try:
+        import gradio as gr
+        # Minimal read-only UI showing models list
+        with gr.Blocks(title="Face Upscale API") as demo:
+            gr.Markdown("# Face Upscale API\nThis Space exposes FastAPI endpoints. UI is optional.")
+        try:
+            from fastapi.middleware.wsgi import WSGIMiddleware
+            # Mount at /gradio for compatibility; Hugging Face can still hit API at root
+            from gradio.routes import mount_gradio_app
+            fastapi_app = mount_gradio_app(fastapi_app, demo, path="/gradio")
+        except Exception:
+            # Fallback mounting
+            fastapi_app.mount("/gradio", WSGIMiddleware(demo.server_app))
+    except Exception:
+        pass
+
+# Expose FastAPI app for Uvicorn/HF Spaces (root ASGI)
 app = fastapi_app

requirements.txt

@@ -23,4 +23,5 @@ gdown # supports downloading the large file from Google Drive
 fastapi
 uvicorn[standard]
 pymongo
-python-multipart
+python-multipart
+gradio