Add yt-audio gradio app

.gitignore

@@ -0,0 +1 @@
+venv

Dockerfile

@@ -0,0 +1,23 @@
+# Use Python 3.10 base image
+FROM python:3.10-slim
+
+# Create a non-root user for security
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:${PATH}"
+
+# Set working directory
+WORKDIR /app
+
+# Install Python dependencies first for layer caching
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Copy application code
+COPY --chown=user . .
+
+# Expose Gradio's default port
+EXPOSE 7860
+
+# Launch the application with Spaces-compatible settings
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,82 @@
+import gradio as gr
+from pytubefix import YouTube
+import tempfile
+import base64
+from io import BytesIO
+
+def download_audio(url):
+    try:
+        # Initialize YouTube object
+        yt = YouTube(url, 'WEB')
+        title = yt.title
+        
+        # Get best audio stream
+        audio_stream = yt.streams.filter(only_audio=True).order_by('abr').desc().first()
+        if not audio_stream:
+            return ["Error: No audio stream found", None, None]
+
+        # Download to in-memory buffer
+        buffer = BytesIO()
+        audio_stream.stream_to_buffer(buffer)
+        buffer.seek(0)
+        
+        # Create temporary file
+        with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as tmp_file:
+            tmp_file.write(buffer.read())
+            tmp_path = tmp_file.name
+
+        # Create HTML audio player with base64 encoding
+        with open(tmp_path, "rb") as f:
+            audio_bytes = f.read()
+        audio_base64 = base64.b64encode(audio_bytes).decode("utf-8")
+        audio_html = f"""
+        <audio controls autoplay style="width: 100%">
+            <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
+            Your browser does not support the audio element.
+        </audio>
+        """
+
+        return [
+            f"Success: {title}",
+            tmp_path,
+            audio_html
+        ]
+    
+    except Exception as e:
+        return [f"Error: {str(e)}", None, None]
+
+with gr.Blocks(title="YouTube Audio Downloader") as demo:
+    gr.Markdown("# YouTube Audio Downloader")
+    
+    with gr.Row():
+        url_input = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube URL here...")
+        
+        with gr.Column(scale=0.3):
+            clear_btn = gr.Button("Clear")
+            submit_btn = gr.Button("Submit")
+    
+    status_output = gr.Textbox(label="Status", interactive=False)
+    file_download = gr.File(label="Audio File")
+    audio_player = gr.HTML(label="Audio Player")
+    
+    def clear():
+        return [None, None, None, None]
+    
+    clear_btn.click(
+        fn=clear,
+        outputs=[url_input, status_output, file_download, audio_player]
+    )
+    
+    submit_btn.click(
+        fn=download_audio,
+        inputs=url_input,
+        outputs=[status_output, file_download, audio_player]
+    )
+
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        allowed_paths=["*"]
+    )

requirements.txt

@@ -0,0 +1,54 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.8.0
+certifi==2024.12.14
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+exceptiongroup==1.2.2
+fastapi==0.115.6
+ffmpy==0.5.0
+filelock==3.17.0
+fsspec==2024.12.0
+gradio==5.12.0
+gradio-client==1.5.4
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.1
+idna==3.10
+jinja2==3.1.5
+markdown-it-py==3.0.0
+markupsafe==2.1.5
+mdurl==0.1.2
+numpy==2.2.2
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+pydantic==2.10.5
+pydantic-core==2.27.2
+pydub==0.25.1
+pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytubefix==8.12.0
+pytz==2024.2
+pyyaml==6.0.2
+requests==2.32.3
+rich==13.9.4
+ruff==0.9.2
+safehttpx==0.1.6
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.41.3
+tomlkit==0.13.2
+tqdm==4.67.1
+typer==0.15.1
+typing-extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==14.2

sample.py

@@ -0,0 +1,72 @@
+import gradio as gr
+import yt_dlp
+import os
+import browser_cookie3
+from huggingface_hub import hf_hub_download, login
+
+
+
+if "cookies.txt" not in os.listdir("."):
+    hf_hub_download("not-lain/secrets","cookies.txt",local_dir=".",repo_type="dataset")
+    
+
+def get_cookies():
+    try:
+        # Try to get cookies from multiple browsers
+        cookies = []
+        for get_cookies in [
+            browser_cookie3.firefox,
+            browser_cookie3.chrome,
+            browser_cookie3.brave,
+        ]:
+            try:
+                cookies.extend([c for c in get_cookies(domain_name=".youtube.com")])
+            except:
+                continue
+        return cookies
+    except:
+        return None
+
+
+def download_video(url):
+    try:
+        downloads_dir = "/downloads"
+        if not os.path.exists(downloads_dir):
+            os.makedirs(downloads_dir, exist_ok=True)
+
+        # Use a pre-provided cookies file
+        cookie_file = "cookies.txt"
+        cookie_args = {}
+        if os.path.exists(cookie_file):
+            cookie_args["cookiefile"] = cookie_file
+
+        ydl_opts = {
+            "outtmpl": os.path.join(downloads_dir, "%(title)s.%(ext)s"),
+            "format": "mp4/bestvideo+bestaudio/best",
+            "merge_output_format": "mp4",
+            **cookie_args,
+            "paths": {"home": "/downloads", "temp": "/tmp"},
+        }
+
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            video_path = ydl.prepare_filename(info)
+            if os.path.exists(video_path):
+                return f"Successfully downloaded: {info['title']}", video_path
+            else:
+                return f"Download completed but file not found: {info['title']}", None
+    except Exception as e:
+        return f"Error: {str(e)}", None
+
+
+iface = gr.Interface(
+    fn=download_video,
+    inputs=gr.Textbox(label="YouTube URL"),
+    outputs=[gr.Textbox(label="Status"), gr.Video(label="Downloaded Video")],
+    title="YouTube Video Downloader",
+    description="Cookies created using the following extension : https://chromewebstore.google.com/detail/get-cookiestxt-clean/ahmnmhfbokciafffnknlekllgcnafnie?hl=en \n\nEnter a YouTube URL to download and view the video",
+    examples=["https://www.youtube.com/watch?v=UiyDmqO59QE"],
+)
+
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0", allowed_paths=["/"], server_port=7860)