HF Space: Dockerized Flask app

Dockerfile

@@ -0,0 +1,30 @@
+# ---- Base image ----
+FROM python:3.11-slim
+
+# ---- System setup (add OS libs your deps may need) ----
+# Add or remove packages as needed. Keep it small to speed builds.
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# ---- App files ----
+WORKDIR /app
+# Copy only requirements first to leverage Docker layer caching
+COPY requirements.txt /app/requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Now copy the rest of your code
+COPY . /app
+
+# ---- Runtime env ----
+# HF Spaces provides a PORT env var; default to 7860 for local runs
+ENV PORT=7860
+ENV PYTHONUNBUFFERED=1
+
+# Make sure Flask binds publicly; gunicorn will use $PORT
+EXPOSE 7860
+
+# ---- Start command ----
+# Gunicorn will import your Flask app named "app" from run.py
+# If your app variable has a different name, change run:app accordingly.
+CMD ["gunicorn", "run:app", "--workers", "2", "--threads", "8", "--timeout", "120", "--bind", "0.0.0.0:${PORT}"]

app/__init__.py

@@ -9,7 +9,13 @@ def create_app():
     # Configure CORS
     CORS(app, resources={
         r"/*": {
-            "origins": ["http://localhost:5000", "http://127.0.0.1:5000"],
+            "origins": [
+                r"https://.*\.hf\.space",        # any HF Space subdomain
+                "https://<your-username>.github.io",
+                "https://www.<your-domain>.com",
+                "http://localhost:5000",
+                "http://127.0.0.1:5000".
+            ],
             "methods": ["GET", "POST", "OPTIONS"],
             "allow_headers": ["Content-Type"]
         }
@@ -27,4 +33,4 @@ def create_app():
     from app.routes import main
     app.register_blueprint(main)
     
-    return app 
+    return app

run.py

@@ -9,7 +9,9 @@ if __name__ == '__main__':
     # Parse command line arguments
     parser = argparse.ArgumentParser(description='Run DP-SGD Explorer')
     parser.add_argument('--port', type=int, default=5000, help='Port to run the server on (default: 5000)')
-    parser.add_argument('--host', type=str, default='127.0.0.1', help='Host to run the server on (default: 127.0.0.1)')
+    # parser.add_argument('--host', type=str, default='127.0.0.1', help='Host to run the server on (default: 127.0.0.1)')
+    parser.add_argument("--host", type=str, default=os.getenv("HOST", "0.0.0.0"),
+                        help="Host to run the server on (default: 0.0.0.0)")
     args = parser.parse_args()
     
     # Enable debug mode for development
@@ -20,4 +22,4 @@ if __name__ == '__main__':
     print(f"Starting server on http://{args.host}:{args.port}")
     
     # Run the application
-    app.run(host=args.host, port=args.port, debug=True) 
+    app.run(host=args.host, port=args.port, debug=True)

space.yml

@@ -0,0 +1,7 @@
+title: DPSGDTool
+emoji: 🛡️
+colorFrom: blue
+colorTo: purple
+sdk: docker
+pinned: false
+license: mit