Add Nginx reverse proxy setup for Hugging Face Spaces

Dockerfile

@@ -11,6 +11,7 @@ RUN apt-get update && apt-get install -y \
     libx11-6 \
     libx11-dev \
     libxrender1 \
+    nginx \
     && rm -rf /var/lib/apt/lists/*
 
 # Create matplotlib config directory with proper permissions
@@ -32,6 +33,13 @@ RUN useradd -m -d /home/appuser -s /bin/bash appuser && \
     chown -R appuser:appuser /home/appuser && \
     chmod 666 /home/appuser/.ocpvscode
 
+# Configure Nginx
+COPY nginx.conf /etc/nginx/nginx.conf
+RUN mkdir -p /run/nginx && \
+    chown -R appuser:appuser /run/nginx && \
+    chown -R appuser:appuser /var/log/nginx && \
+    chown -R appuser:appuser /var/lib/nginx
+
 # Install uv and create virtual environment
 RUN pip install uv && \
     uv venv /opt/venv
@@ -62,5 +70,9 @@ ENV HOME=/home/appuser
 # Expose port 7860 for Hugging Face Spaces
 EXPOSE 7860
 
-# Run the application
-CMD ["python", "-c", "import cadviewer; from nicegui import app; app.native.start_args['port'] = 7860; cadviewer.ui.run(native=False, host='0.0.0.0', port=7860)"] 
+# Create startup script
+COPY start.sh /code/start.sh
+RUN chmod +x /code/start.sh
+
+# Run the startup script
+CMD ["/code/start.sh"] 

cadviewer.py

@@ -198,14 +198,8 @@ with ui.splitter().classes(
             space_url = os.getenv('SPACE_URL', '')
             logger.info(f"Space URL: {space_url}")
             
-            # Construct the viewer URL
-            if space_url:
-                # We're in a Hugging Face Space
-                viewer_url = f"{space_url}/proxy/3939/viewer"
-            else:
-                # Local development
-                viewer_url = "http://localhost:3939/viewer"
-                
+            # Construct the viewer URL - always use /viewer since Nginx handles the routing
+            viewer_url = "/viewer"
             logger.info(f"Using viewer URL: {viewer_url}")
             
             ocpcv = (

nginx.conf

@@ -0,0 +1,44 @@
+worker_processes 1;
+error_log /dev/stdout info;
+pid /run/nginx/nginx.pid;
+
+events {
+    worker_connections 1024;
+}
+
+http {
+    access_log /dev/stdout;
+
+    upstream nicegui {
+        server 127.0.0.1:7861;
+    }
+
+    upstream viewer {
+        server 127.0.0.1:3939;
+    }
+
+    server {
+        listen 7860;
+        server_name localhost;
+
+        # Main app (NiceGUI)
+        location / {
+            proxy_pass http://nicegui;
+            proxy_http_version 1.1;
+            proxy_set_header Upgrade $http_upgrade;
+            proxy_set_header Connection "upgrade";
+            proxy_set_header Host $host;
+            proxy_set_header X-Real-IP $remote_addr;
+        }
+
+        # Viewer route
+        location /viewer/ {
+            proxy_pass http://viewer/viewer/;
+            proxy_http_version 1.1;
+            proxy_set_header Upgrade $http_upgrade;
+            proxy_set_header Connection "upgrade";
+            proxy_set_header Host $host;
+            proxy_set_header X-Real-IP $remote_addr;
+        }
+    }
+} 

start.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+# Start Nginx
+nginx
+
+# Start the application on port 7861 (Nginx will proxy from 7860)
+python -c "import cadviewer; from nicegui import app; app.native.start_args['port'] = 7861; cadviewer.ui.run(native=False, host='0.0.0.0', port=7861)"