swap to gradio and docker added

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.10-slim
+
+WORKDIR /home
+
+ENV PYTHONPATH=.
+
+COPY requirements.txt /home/requirements.txt
+COPY youtube_downloader /home/youtube_downloader
+RUN pip3 install -r /home/requirements.txt
+
+EXPOSE 7860
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+
+CMD ["gradio", "/home/youtube_downloader/app.py"]

README.md

@@ -4,8 +4,14 @@
 
 A simple python app that lets you painlessly download youtube / shorts video files without needing to visit chintzy sites online.  Can be easily run locally.  Try it out now in your browser at [![HuggingFace Space](https://img.shields.io/badge/🤗-HuggingFace%20Space-cyan.svg)](https://huggingface.co/spaces/neonwatty/youtube_downloader).
 
-To run the app install the associated `requirements.txt` (in a venv) and run
+To run the app install the associated `requirements.txt` and run
 
 ```python
-python -m streamlit run youtube_downloader/app.py
+python youtube_downloader/app.py
+```
+
+Or run via Docker
+
+```sh
+docker compose up
 ```

docker-compose.yml

@@ -0,0 +1,8 @@
+services:
+  youtube_downloader:
+    build:
+      context: .
+    image: youtube_downloader
+    container_name: youtube_downloader
+    ports:
+      - 7860:7860

requirements.txt

@@ -1,3 +1,2 @@
-pytube
-moviepy  # make sure ffmpeg is installed on your machine!
-streamlit
+yt-dlp
+gradio==4.38.1

tests/__init__.py

@@ -2,4 +2,4 @@
 
 cwd = os.getcwd()
 CONTAINER_NAME = "youtube_downloader"
-STREAMLIT_FILE = "youtube_downloader/app.py"
+APP_FILE = "youtube_downloader/app.py"

tests/test_app.py

@@ -0,0 +1,27 @@
+import subprocess
+import pytest
+import time
+from tests import APP_FILE
+
+
+@pytest.fixture(scope="module")
+def start_app():
+    cmd = f"python {APP_FILE}"
+    process = subprocess.Popen(
+        cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE
+    )
+    time.sleep(5)
+    yield process
+    process.terminate()
+    process.wait()
+
+
+def test_streamlit(subtests, start_app):
+    with subtests.test(msg="server up"):
+        assert start_app.poll() is None, "app failed to start"
+
+    with subtests.test(msg="streamlit down"):
+        start_app.terminate()
+        time.sleep(2)
+        assert start_app.poll() is not None, "app failed to stop"
+