Update Dockerfile

Dockerfile

@@ -1,42 +1,54 @@
-# Use the official Python 3.11 slim image as the base
 FROM python:3.11-slim
 
-# Create non-root user
-RUN useradd -ms /bin/bash appuser
-USER appuser
-WORKDIR /home/appuser
-
-# Set venv path
+# System environment
 ENV VENV_PATH="/home/appuser/venv"
 ENV PATH="$VENV_PATH/bin:$PATH"
+ENV HF_HOME="/home/appuser/.cache/huggingface"
 
-# Install system dependencies
+# Create user
+RUN useradd -ms /bin/bash appuser
 USER root
+
+# Install system dependencies
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
         ffmpeg \
         espeak-ng \
         wget \
+        git \
+        curl \
         python3-venv && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
-# Switch back to non-root user
+# Switch to non-root user
 USER appuser
+WORKDIR /home/appuser
 
-# Create and activate virtual environment
+# Setup virtual environment
 RUN python3 -m venv $VENV_PATH && \
-    $VENV_PATH/bin/pip install --no-cache-dir --upgrade pip && \
-    $VENV_PATH/bin/pip install --no-cache-dir audiblez && \
-    $VENV_PATH/bin/python -m spacy download xx_ent_wiki_sm
+    $VENV_PATH/bin/pip install --upgrade pip && \
+    $VENV_PATH/bin/pip install audiblez huggingface_hub
 
 # Download Kokoro model and voices
-RUN mkdir -p /home/appuser/.local/share/audiblez/ && \
-    wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/kokoro-v0_19.onnx -P /home/appuser/.local/share/audiblez/ && \
-    wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/voices.json -P /home/appuser/.local/share/audiblez/
-
-# Create working and output directories
-WORKDIR /home/appuser/ebooks
-RUN mkdir -p /home/appuser/m4b
-
-# Convert and copy m4b files
-CMD for book in *.epub; do $VENV_PATH/bin/audiblez "$book" -v af_aoede; done && cp *.m4b /home/appuser/m4b/
+RUN mkdir -p /home/appuser/.local/share/audiblez && \
+    wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/kokoro-v0_19.onnx -P /home/appuser/.local/share/audiblez && \
+    wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/voices.json -P /home/appuser/.local/share/audiblez
+
+# Create ebook and output folders
+RUN mkdir -p /home/appuser/ebooks /home/appuser/m4b
+
+# Copy EPUBs from Hugging Face Space (example with 2 test books, update as needed)
+RUN wget https://huggingface.co/spaces/MoiMoi-01/AudiobookMakerAudiblezTrial/resolve/main/ebooks/MyDescendantBeggedMeToHelpHimJustAfterIBecameAGod.epub -P /home/appuser/ebooks && \
+    wget https://huggingface.co/spaces/MoiMoi-01/AudiobookMakerAudiblezTrial/resolve/main/ebooks/MySwordIntentCanBeInfinitelyImproved.epub -P /home/appuser/ebooks
+
+# Final command to convert and upload m4b files
+CMD for book in /home/appuser/ebooks/*.epub; do \
+        $VENV_PATH/bin/audiblez "$book" -v af_sky; \
+    done && \
+    cp *.m4b /home/appuser/m4b && \
+    echo "Uploading to Hugging Face..." && \
+    $VENV_PATH/bin/huggingface-cli login --token $HF_TOKEN && \
+    cd /home/appuser/m4b && \
+    for file in *.m4b; do \
+        huggingface-cli upload MoiMoi-01/AudiobookMakerAudiblezTrial "$file" --path_in_repo m4b/"$file"; \
+    done