feat: Add UI preview images, update frontend access port, and refactor Docker Compose for model service management.

siddharthksah · siddharthksah · commit aea8b9aafa89 · 2025-12-03T20:36:01.000+08:00
diff --git a/.gitignore b/.gitignore
@@ -12,4 +12,5 @@ models/audio/vocoder_artifacts/models/librifake_pretrained_lambda0.5_epoch_25.pt
 deepsafe_utils/__pycache__/visualizer.cpython-313.pyc
 .gitignore
 #ignore all .pyc files
-*.pyc   
+*.pyc   
+models/audio/vocoder_artifacts/temp_repo
diff --git a/Makefile b/Makefile
@@ -16,7 +16,7 @@ install:
 start:
 	@echo "Starting DeepSafe..."
 	docker-compose up -d
-	@echo "DeepSafe is running at http://localhost:80"
+	@echo "DeepSafe is running at http://localhost:8888"
 
 stop:
 	@echo "Stopping DeepSafe..."
diff --git a/README.md b/README.md
@@ -32,6 +32,13 @@ DeepSafe adopts a **microservices architecture** where each detection model runs
 -   **Dockerized Architecture**: Fully containerized services for easy deployment and isolation.
 -   **RESTful API**: Robust FastAPI backend with health checks, batch processing, and detailed logging.
 
+## 📸 UI Preview
+
+<div align="center">
+  <img src="docs/images/login_page.png" alt="Login Page" width="45%">
+  <img src="docs/images/dashboard.png" alt="Dashboard" width="45%">
+</div>
+
 ## 🏗️ Architecture
 
 DeepSafe orchestrates a fleet of specialized detectors via a central API gateway:
@@ -73,7 +80,7 @@ graph TD
     *This will build all containers and start the services. Initial build may take a few minutes.*
 
 3.  **Access the Dashboard**:
-    Open `http://localhost:80` in your browser.
+    Open `http://localhost:8888` in your browser.
 
 4.  **API Documentation**:
     Visit `http://localhost:8000/docs` for the interactive Swagger UI.
diff --git a/api/Dockerfile b/api/Dockerfile
@@ -23,6 +23,7 @@ RUN pip install uv && \
 
 # Copy application code
 COPY main.py .
+COPY database.py .
 
 # Expose the port the app runs on
 EXPOSE ${PORT}
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -14,11 +14,11 @@ services:
       - PORT=8000
       - META_MODEL_ARTIFACTS_DIR=/app/meta_model_artifacts
       - DEEPSAFE_CONFIG_FILE_PATH=/app/config/deepsafe_config.json
+      # - LOG_LEVEL=debug # Optional: for more verbose API logs
     depends_on:
       - npr_deepfakedetection
       - universalfakedetect
-      - cross_efficient_vit
-      - vocoder_artifacts
+      # Add other future video/audio model service names here
     networks:
       - deepsafe-network
 
@@ -37,6 +37,36 @@ services:
     networks:
       - deepsafe-network
 
+  # yermandy_clip_detection:
+  #   build: ./models/image/yermandy_clip_detection
+  #   container_name: deepsafe-yermandy-clip-detection
+  #   ports:
+  #     - "5002:5002"
+  #   restart: unless-stopped
+  #   environment:
+  #     - MODEL_PORT=5002
+  #     - MODEL_PATH=model_code/weights/model.ckpt
+  #     - PRELOAD_MODEL=false
+  #     - MODEL_TIMEOUT=600
+  #     - USE_GPU=false
+  #   networks:
+  #     - deepsafe-network
+
+  # wavelet_clip_detection:
+  #   build: ./models/image/wavelet_clip_detection
+  #   container_name: deepsafe-wavelet-clip-detection
+  #   ports:
+  #     - "5003:5003"
+  #   restart: unless-stopped
+  #   environment:
+  #     - MODEL_PORT=5003
+  #     - MODEL_PATH=model_code/weights/clip_wavelet_best.pth
+  #     - PRELOAD_MODEL=false
+  #     - MODEL_TIMEOUT=600
+  #     - USE_GPU=false
+  #   networks:
+  #     - deepsafe-network
+
   universalfakedetect:
     build: ./models/image/universalfakedetect
     container_name: deepsafe-universalfakedetect
@@ -51,7 +81,48 @@ services:
     networks:
       - deepsafe-network
 
-  # --- Video Model Services ---
+  # trufor:
+  #   build: ./models/image/trufor
+  #   container_name: deepsafe-trufor
+  #   ports:
+  #     - "5005:5005"
+  #   restart: unless-stopped
+  #   environment:
+  #     - MODEL_PORT=5005
+  #     - PRELOAD_MODEL=false
+  #     - MODEL_TIMEOUT=600
+  #     - USE_GPU=false
+  #   networks:
+  #     - deepsafe-network
+
+  # spsl_deepfake_detection:
+  #   build: ./models/image/spsl_deepfake_detection
+  #   container_name: deepsafe-spsl-deepfake-detection
+  #   ports:
+  #     - "5006:5006"
+  #   restart: unless-stopped
+  #   environment:
+  #     - MODEL_PORT=5006
+  #     - PRELOAD_MODEL=false
+  #     - MODEL_TIMEOUT=600
+  #     - USE_GPU=false
+  #   networks:
+  #     - deepsafe-network
+
+  # ucf_deepfake_detection:
+  #   build: ./models/image/ucf_deepfake_detection
+  #   container_name: deepsafe-ucf-deepfake-detection
+  #   ports:
+  #     - "5007:5007"
+  #   restart: unless-stopped
+  #   environment:
+  #     - MODEL_PORT=5007
+  #     - PRELOAD_MODEL=false
+  #     - MODEL_TIMEOUT=600
+  #     - USE_GPU=false
+  #   networks:
+  #     - deepsafe-network
+
   cross_efficient_vit:
     build: ./models/video/cross_efficient_vit
     container_name: deepsafe-cross-efficient-vit
@@ -60,34 +131,42 @@ services:
     restart: unless-stopped
     environment:
       - MODEL_PORT=7001
-      - PRELOAD_MODEL=false
-      - MODEL_TIMEOUT=900
-      - USE_GPU=false
+      - PRELOAD_MODEL=false 
+      - MODEL_TIMEOUT=900   # Video models might be heavier, allow longer idle time
+      - USE_GPU=false       # Ensure CPU operation
+      # The following are set within the cross_efficient_vit/Dockerfile,
+      # but can be overridden here if needed for different weight files for the same image:
+      # - DEFAULT_MODEL_VARIANT="cross_efficient_vit" 
+      # - CROSS_EFFICIENT_VIT_MODEL_PATH="/app/model_code/cross-efficient-vit/pretrained_models/cross_efficient_vit.pth"
+      # - EFFICIENT_VIT_MODEL_PATH="/app/model_code/efficient-vit/pretrained_models/efficient_vit.pth"
+      # - CROSS_EFFICIENT_VIT_CONFIG_PATH="/app/model_code/cross-efficient-vit/configs/architecture.yaml"
+      # - EFFICIENT_VIT_CONFIG_PATH="/app/model_code/efficient-vit/configs/architecture.yaml"
+      # - FRAMES_PER_VIDEO="15" # If you want to control frame sampling from docker-compose
     networks:
       - deepsafe-network
+    # Optional: Add resource limits for CPU/memory if this model is resource-intensive
+    # deploy:
+    #   resources:
+    #     limits:
+    #       cpus: '2.0' 
+    #       memory: 4G
 
-  # --- Audio Model Services ---
-  vocoder_artifacts:
-    build: ./models/audio/vocoder_artifacts
-    container_name: deepsafe-vocoder-artifacts
-    ports:
-      - "8001:8001"
-    restart: unless-stopped
-    environment:
-      - MODEL_PORT=8001
-      - PRELOAD_MODEL=false
-      - MODEL_TIMEOUT=600
-      - USE_GPU=false
-    volumes:
-      - ./models/audio/vocoder_artifacts/models:/app/models:ro
-    networks:
-      - deepsafe-network
+  # --- Placeholder for future audio model service ---
+  # example_audio_model_1:
+  #   build: ./models/audio/example_audio_model_1
+  #   container_name: deepsafe-example-audio-model-1
+  #   ports:
+  #     - "8001:8001"
+  #   restart: unless-stopped
+  #   # ... environment ...
+  #   networks:
+  #     - deepsafe-network
 
   frontend:
     build: ./frontend
     container_name: deepsafe-frontend
     ports:
-      - "80:80"
+      - "8888:80"
     restart: unless-stopped
     depends_on:
       - api
@@ -96,4 +175,4 @@ services:
 
 networks:
   deepsafe-network:
-    driver: bridge
+    driver: bridge
diff --git a/docs/images/dashboard.png b/docs/images/dashboard.png
diff --git a/docs/images/login_page.png b/docs/images/login_page.png
diff --git a/models/video/cross_efficient_vit/Dockerfile b/models/video/cross_efficient_vit/Dockerfile
@@ -28,8 +28,7 @@ ENV QT_QPA_PLATFORM=offscreen
 COPY requirements.txt .
 
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
-    pip install uv && \
-    uv pip install --system --no-cache-dir \
+    pip install --no-cache-dir \
     torch==1.11.0 \
     torchvision==0.12.0 \
     torchaudio==0.11.0 \
@@ -38,11 +37,11 @@ RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
 # NumPy version will be handled by requirements.txt now
 # REMOVE THIS LINE: RUN pip install --no-cache-dir numpy==1.21.6
 
-RUN uv pip install --system --no-cache-dir opencv-python-headless==4.6.0.66
+RUN pip install --no-cache-dir opencv-python-headless==4.6.0.66
 
 # Install other Python dependencies from requirements.txt
 # This will install numpy as per the constraint in requirements.txt
-RUN uv pip install --system --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 # ... (rest of your Dockerfile remains the same) ...
 
@@ -58,14 +57,14 @@ RUN mkdir -p /app/model_code/efficient-vit/pretrained_models && \
 
 # Move downloaded weights to their respective locations
 RUN if [ -f /app/model_code/gdrive_weights/efficient_vit.pth ]; then \
-    mv /app/model_code/gdrive_weights/efficient_vit.pth /app/model_code/efficient-vit/pretrained_models/efficient_vit.pth; \
+        mv /app/model_code/gdrive_weights/efficient_vit.pth /app/model_code/efficient-vit/pretrained_models/efficient_vit.pth; \
     else \
-    echo "Warning: efficient_vit.pth not found in gdrive_weights, assuming it might be part of the repo or handled differently." ; \
+        echo "Warning: efficient_vit.pth not found in gdrive_weights, assuming it might be part of the repo or handled differently." ; \
     fi && \
     if [ -f /app/model_code/gdrive_weights/cross_efficient_vit.pth ]; then \
-    mv /app/model_code/gdrive_weights/cross_efficient_vit.pth /app/model_code/cross-efficient-vit/pretrained_models/cross_efficient_vit.pth; \
+        mv /app/model_code/gdrive_weights/cross_efficient_vit.pth /app/model_code/cross-efficient-vit/pretrained_models/cross_efficient_vit.pth; \
     else \
-    echo "Warning: cross_efficient_vit.pth not found in gdrive_weights, assuming it might be part of the repo or handled differently." ; \
+        echo "Warning: cross_efficient_vit.pth not found in gdrive_weights, assuming it might be part of the repo or handled differently." ; \
     fi && \
     rm -rf /app/model_code/gdrive_weights
 
