Refactor Dockerfile and start script; remove unused dependencies and streamline setup

Dockerfile

@@ -28,7 +28,6 @@ RUN apt-get update && apt-get install -y \
     libprotobuf-dev \
     protobuf-compiler \
     git \
-    ninja-build \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy root requirements and install
@@ -44,15 +43,9 @@ RUN mkdir -p id-maker/assets && \
     wget -O id-maker/assets/TYBAH.TTF "https://raw.githubusercontent.com/Esmaill1/color-stealer/main/TYBAH.TTF" && \
     wget -O id-maker/assets/arialbd.ttf "https://raw.githubusercontent.com/Esmaill1/color-stealer/main/arialbd.ttf"
 
-# Build CodeFormer extensions (basicsr)
-WORKDIR /app/codeformer
-RUN python basicsr/setup.py build_ext --inplace
-
 # Set up storage directories and permissions
 WORKDIR /app
-RUN mkdir -p id-maker/storage/uploads id-maker/storage/processed id-maker/storage/results \
-             codeformer/static/uploads codeformer/static/results \
-             codeformer/weights codeformer/inputs codeformer/output && \
+RUN mkdir -p id-maker/storage/uploads id-maker/storage/processed id-maker/storage/results && \
     chmod -R 777 /app
 
 # Create the startup script to run both services

id-maker/config/settings.json

@@ -59,5 +59,8 @@
             60,
             60
         ]
+    },
+    "restoration": {
+        "fidelity": 1
     }
 }

requirements.txt

@@ -16,6 +16,7 @@ flask
 python-multipart
 jinja2
 MarkupSafe
+requests
 
 # Image Processing
 Pillow>=10.0.0
@@ -27,16 +28,5 @@ protobuf<=3.20.3
 scipy
 scikit-image
 
-# CodeFormer Utilities
-addict
-future
-lmdb
-pyyaml
-requests
-tqdm
-yapf
-lpips
-gdown
-
 # GUI (Optional for web but in id-maker reqs)
 pywebview

start.sh

@@ -1,17 +1,9 @@
 #!/bin/bash
 
-# 1. Start CodeFormer internally on port 8001
-export CODEFORMER_PORT=8001
-echo "Starting CodeFormer API on port $CODEFORMER_PORT..."
-cd /app/codeformer && python app.py &
-
-# 2. Wait for CodeFormer to initialize (downloads weights on first run)
-# We give it some time, though it might take longer depending on connection.
-sleep 5
-
-# 3. Start ID Maker on the main HF port (7860)
+# Start ID Maker on the main HF port (7860)
 export PORT=7860
-export CODEFORMER_API_URL=http://localhost:8001/api/restore
+# CODEFORMER_API_URL is NOT set here, so it will use the value in id-maker/config/settings.json
+# (which points to the external HF Space)
+
 echo "Starting ID Maker Web UI on port $PORT..."
-echo "CodeFormer URL: $CODEFORMER_API_URL"
 cd /app/id-maker && python web/server.py