Optimize Dockerfile for Hugging Face Spaces deployment and update server configuration

- Refactored Dockerfile to improve caching and streamline the build process.
- Added environment variables for configuration, including PORT and logging settings.
- Updated server.py to utilize environment variables for configuration and improved logging.
- Enhanced CORS settings to allow requests from Hugging Face Spaces domains.
- Added werkzeug to requirements.txt for compatibility with the updated server implementation.

Dockerfile

@@ -1,39 +1,29 @@
-# Dockerfile
+# Dockerfile optimizado para Hugging Face Spaces
 FROM python:3.11-slim
 
-# Crear usuario no root
-RUN adduser --disabled-password --gecos '' appuser
+WORKDIR /code
 
-WORKDIR /app
-
-# Instalar dependencias del sistema si son necesarias
-RUN apt-get update && apt-get install -y --no-install-recommends \
+# Instalar dependencias del sistema
+RUN apt-get update && apt-get install -y \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Copiar archivos de la aplicación
+# Copiar solo requirements.txt primero para aprovechar la caché de Docker
 COPY requirements.txt .
-COPY server.py .
-COPY hypercorn.toml .
 
 # Instalar dependencias de Python
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Configurar permisos
-RUN chown -R appuser:appuser /app && \
-    mkdir -p /home/appuser/.cache/huggingface && \
-    chown -R appuser:appuser /home/appuser/.cache
+# Copiar el resto de los archivos
+COPY . .
 
-# Variables de entorno de seguridad
+# Variables de entorno para Hugging Face Spaces
 ENV PYTHONUNBUFFERED=1
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONPATH=/app
-
-# Cambiar al usuario no root
-USER appuser
+ENV PYTHONPATH=/code
+ENV PORT=7860
 
 # Exponer el puerto que Hugging Face Spaces espera
 EXPOSE 7860
 
-# Comando para ejecutar la aplicación con Hypercorn
-CMD ["hypercorn", "--config", "hypercorn.toml", "server:app"]
+# Comando para ejecutar la aplicación
+CMD ["hypercorn", "server:app", "--bind", "0.0.0.0:7860", "--workers", "1"]

requirements.txt

@@ -1,4 +1,5 @@
 quart==0.18.4
+werkzeug==2.3.7
 quart-schema==0.15.0
 quart-cors==0.7.0
 hypercorn==0.14.3

server.py

@@ -9,28 +9,33 @@ from quart_cors import cors
 import importlib.metadata
 import secrets
 import logging
-import hypercorn.asyncio
+import os
 
 from broker import SessionBroker, SessionDoesNotExist, ClientRequest, ClientResponse, ClientError
 
-# Configuración
-TIMEOUT: int = 60
-LOG_LEVEL: int = logging.INFO
-TRUSTED_HOSTS: list[str] = ["127.0.0.1"]
-MAX_MESSAGE_SIZE: int = 16 * 1024 * 1024  # 16MB
-RATE_LIMIT: int = 100  # requests per minute
-SESSION_TIMEOUT: int = 3600  # 1 hour
+# Configuración para Hugging Face Spaces
+PORT = int(os.getenv('PORT', 7860))
+TIMEOUT: int = int(os.getenv('TIMEOUT', 60))
+LOG_LEVEL: int = getattr(logging, os.getenv('LOG_LEVEL', 'INFO'))
+MAX_MESSAGE_SIZE: int = int(os.getenv('MAX_MESSAGE_SIZE', 16 * 1024 * 1024))
+RATE_LIMIT: int = int(os.getenv('RATE_LIMIT', 100))
+SESSION_TIMEOUT: int = int(os.getenv('SESSION_TIMEOUT', 3600))
+
+# Configurar logging
+logging.basicConfig(
+    level=LOG_LEVEL,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
 
-# Crear aplicación con CORS habilitado y configuración de seguridad
+# Crear aplicación con CORS habilitado
 app = Quart(__name__)
 app = cors(app, 
-    allow_origin="*",
+    allow_origin=["https://*.hf.space", "https://*.huggingface.co"],
     allow_methods=["GET", "POST", "OPTIONS"],
     allow_headers=["Content-Type"],
     max_age=3600
 )
 QuartSchema(app)
-app.logger.setLevel(LOG_LEVEL)
 
 broker = SessionBroker()
 
@@ -108,9 +113,11 @@ async def command(data: Command) -> Tuple[CommandResponse | ErrorResponse, int]:
 
 # Ejecutar aplicación
 def run():
-    config = hypercorn.Config()
-    config.bind = ["0.0.0.0:7860"]
-    asyncio.run(hypercorn.asyncio.serve(app, config))
+    app.run(
+        host='0.0.0.0',
+        port=PORT,
+        debug=False
+    )
 
 # Agregar un endpoint de health check
 @app.route("/health")