Update Dockerfile

Dockerfile

@@ -1,40 +1,40 @@
-# Use official Python 3.10 slim image for a smaller footprint
 FROM python:3.10-slim
 
-# Set environment variables
-ENV TRANSFORMERS_CACHE=/app/cache \
-    HF_HOME=/app/cache \
-    USE_QUANTIZATION=0 \
-    DEVICE=-1 \
-    MAX_TEXT_LENGTH=5000
-
 # Set working directory
 WORKDIR /app
 
-# Create the /app directory explicitly to avoid copy errors
-RUN mkdir -p /app
-
-# Copy requirements file
-COPY requirements.txt .
+# Create necessary directories
+RUN mkdir -p /app/cache /app/logs
 
-# Install system dependencies and Python packages
-# Note: libgomp1 is required for PyTorch
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV HF_HOME=/app/cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV MAX_TEXT_LENGTH=5000
+ENV GRADIO_SERVER_NAME=0.0.0.0
+ENV GRADIO_SERVER_PORT=7860
+
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
-    gcc \
-    libgomp1 \
-    && rm -rf /var/lib/apt/lists/* \
-    && pip install --no-cache-dir --upgrade pip \
+    build-essential \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements and install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip \
     && pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py .
 
-# Create cache and logs directories with write permissions
-RUN mkdir -p /app/cache /app/logs \
-    && chmod -R 777 /app/cache /app/logs
+# Create a non-root user for security
+RUN useradd -m -u 1000 user
+RUN chown -R user:user /app
+USER user
 
-# Expose port for FastAPI
-EXPOSE 8000
+# Expose port
+EXPOSE 7860
 
-# Run Uvicorn server
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+# Run the application
+CMD ["python", "app.py"]