changes

Dockerfile

@@ -18,6 +18,11 @@ COPY . .
 ENV YT_AUDIO_DIR=/tmp/ytdl
 RUN mkdir -p /tmp/ytdl /app/downloads && chmod 777 /tmp/ytdl /app/downloads
 
+# Add entrypoint to allow runtime DNS/Proxy configuration
+COPY docker-entrypoint.sh /usr/local/bin/docker-entrypoint.sh
+RUN chmod +x /usr/local/bin/docker-entrypoint.sh
+ENTRYPOINT ["/usr/local/bin/docker-entrypoint.sh"]
+
 EXPOSE 7860
 
 ENV PORT=7860

app.py

@@ -30,6 +30,10 @@ def download_youtube_audio(url: HttpUrl, audio_format: Literal["mp3", "m4a", "wa
     - audio_format: Desired audio format (requires ffmpeg in the container)
 
     The file will be saved under /app/downloads. Ensure the container has write access.
+
+    Networking:
+    - If outbound DNS is restricted, set env DNS_SERVERS (space-separated, e.g. "8.8.8.8 1.1.1.1").
+    - To route via a proxy, set env YT_PROXY (e.g. http://user:pass@proxy:port).
     """
     # Ensure output directory exists
     output_dir = "/app/downloads"
@@ -55,6 +59,10 @@ def download_youtube_audio(url: HttpUrl, audio_format: Literal["mp3", "m4a", "wa
         "nocheckcertificate": True,
     }
 
+    proxy_url = os.environ.get("YT_PROXY")
+    if proxy_url:
+        ydl_opts["proxy"] = proxy_url
+
     info_title: Optional[str] = None
     downloaded_id: Optional[str] = None
 

docker-entrypoint.sh

@@ -0,0 +1,24 @@
+#!/bin/sh
+set -e
+
+# Optionally configure DNS servers at runtime if provided
+if [ -n "$DNS_SERVERS" ]; then
+  {
+    for s in $DNS_SERVERS; do
+      echo "nameserver $s"
+    done
+  } > /etc/resolv.conf 2>/dev/null || true
+fi
+
+# Optionally configure proxy for outbound requests
+if [ -n "$YT_PROXY" ]; then
+  export HTTPS_PROXY="$YT_PROXY"
+  export HTTP_PROXY="$YT_PROXY"
+fi
+
+# Ensure output directory exists and is writable
+OUTDIR="${YT_AUDIO_DIR:-/tmp/ytdl}"
+mkdir -p "$OUTDIR" 2>/dev/null || true
+chmod 777 "$OUTDIR" 2>/dev/null || true
+
+exec "$@"