Upload 6 files

app.py

@@ -0,0 +1,61 @@
+import os
+import fn
+import gradio as gr
+
+fn.load_model()
+
+last_url = None
+last_video = None
+last_audio = None
+last_chat = None
+last_info = None
+
+def run(url, mode):
+    global last_url, last_video, last_audio, last_chat, last_info
+
+    if url != last_url:
+        last_url = url
+        if os.path.exists(last_video):
+            os.remove(last_video)
+        last_video = None
+        if os.path.exists(last_audio):
+            os.remove(last_audio)
+        last_audio = None
+        last_chat = None
+        last_info = None
+
+    result = fn.run(url, mode)
+
+    if mode == 'video':
+        ext = os.path.splitext(result)[1]
+        os.rename(result, 'last_video' + ext)
+        last_video = 'last_video' + ext
+    elif mode == 'audio':
+        ext = os.path.splitext(result)[1]
+        os.rename(result, 'last_audio' + ext)
+        last_audio = 'last_audio' + ext
+    elif mode == 'chat':
+        last_chat = result
+    elif mode == 'info':
+        last_info = result
+
+    return last_video, last_audio, last_chat, last_info
+
+with gr.Blocks() as demo:
+    title = gr.Markdown('# ytdlp')
+    url = gr.Textbox(label="URL", interactive=True)
+    mode = gr.Dropdown(value='video', choices=['video', 'audio', 'chat', 'info'], label="mode", allow_custom_value=False, interactive=True)
+    run_button = gr.Button(value='Run')
+    video = gr.Video(label="Video", interactive=False)
+    audio = gr.Audio(label="Audio", interactive=False, type="filepath")
+    chat = gr.Textbox(label="Chat", interactive=False, show_copy_button=True)
+    info = gr.Textbox(label="Info", interactive=False, show_copy_button=True)
+
+    run_button.click(
+        fn=run,
+        inputs=[url, mode],
+        outputs=[video, audio, chat, info],
+    )
+
+if __name__ == '__main__':
+    demo.launch()

fn.py

@@ -0,0 +1,52 @@
+import os
+import glob
+import json
+from yt_dlp import YoutubeDL
+
+def load_model():
+    pass
+    
+def run(url, mode = None):
+    if not url.startswith('https://'):
+        url = f'https://www.youtube.com/watch?v={url}'
+    video_id = url[-11:]
+
+    # cleanup
+    for file in glob.glob(f'{video_id}.*'):
+        os.remove(file)
+
+    if mode == 'info':
+        with YoutubeDL() as ydl:
+            info = ydl.extract_info(url, download=False)
+            return json.dumps(info)
+    elif mode == 'chat':
+        args = {
+            'outtmpl': '%(id)s'+'.%(ext)s',
+            'writesubtitles': True,
+            'skip_download': True,
+        }
+        with YoutubeDL(args) as ydl:
+            result = ydl.download([url])
+            print(result)
+            for file in glob.glob(f'{video_id}.*'):
+                return file
+    elif mode == 'audio':
+        args = {
+            'outtmpl': '%(id)s'+'.%(ext)s',
+            'format': 'bestaudio',
+        }
+        with YoutubeDL(args) as ydl:
+            result = ydl.download([url])
+            print(result)
+            for file in glob.glob(f'{video_id}.*'):
+                return file
+    else:
+        args = {
+            'outtmpl': '%(id)s'+'.%(ext)s',
+            'format': 'best',
+        }
+        with YoutubeDL(args) as ydl:
+            result = ydl.download([url])
+            print(result)
+            for file in glob.glob(f'{video_id}.*'):
+                return file

install.bat

@@ -0,0 +1,56 @@
+@echo off
+
+rem -------------------------------------------
+rem NOT guaranteed to work on Windows
+
+set REPOS=https://huggingface.co/spaces/aka7774/ytdlp
+set APPDIR=ytdlp
+set VENV=venv
+
+rem -------------------------------------------
+
+set INSTALL_DIR=%~dp0
+cd /d %INSTALL_DIR%
+
+:git_clone
+set DL_URL=%REPOS%
+set DL_DST=%APPDIR%
+git clone %DL_URL% %APPDIR%
+if exist %DL_DST% goto install_python
+
+set DL_URL=https://github.com/git-for-windows/git/releases/download/v2.41.0.windows.3/PortableGit-2.41.0.3-64-bit.7z.exe
+set DL_DST=PortableGit-2.41.0.3-64-bit.7z.exe
+curl -L -o %DL_DST% %DL_URL%
+if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
+%DL_DST% -y
+del %DL_DST%
+
+set GIT=%INSTALL_DIR%PortableGit\bin\git
+%GIT% clone %REPOS%
+
+:install_python
+set DL_URL=https://github.com/indygreg/python-build-standalone/releases/download/20240107/cpython-3.10.13+20240107-i686-pc-windows-msvc-shared-install_only.tar.gz
+set DL_DST="%INSTALL_DIR%python.tar.gz"
+curl -L -o %DL_DST% %DL_URL%
+if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
+tar -xzf %DL_DST%
+
+set PYTHON=%INSTALL_DIR%python\python.exe
+set PATH=%PATH%;%INSTALL_DIR%python310\Scripts
+
+:install_venv
+cd %APPDIR%
+%PYTHON% -m venv %VENV%
+set PYTHON=%VENV%\Scripts\python.exe
+
+:install_pip
+set DL_URL=https://bootstrap.pypa.io/get-pip.py
+set DL_DST=%INSTALL_DIR%get-pip.py
+curl -o %DL_DST% %DL_URL%
+if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
+%PYTHON% %DL_DST%
+
+%PYTHON% -m pip install gradio
+%PYTHON% -m pip install -r requirements.txt
+
+pause

main.py

@@ -0,0 +1,55 @@
+import os
+import sys
+import time
+import signal
+import psutil
+import io
+
+from fastapi import FastAPI, Request, status, Form, UploadFile
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel, Field
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import JSONResponse
+
+import fn
+import gradio as gr
+from app import demo
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=['*'],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+gr.mount_gradio_app(app, demo, path="/gradio")
+
+fn.load_model()
+
+@app.post("/raw1")
+async def api_raw1(file: UploadFile = Form(...)):
+    try:
+        file_content = await file.read()
+        file_stream = io.BytesIO(file_content)
+
+		raw1, raw2 = fn.raw(file_stream)
+
+        return Response(content=raw1, media_type="audio/wav")
+    except Exception as e:
+        return {"error": str(e)}
+
+@app.post("/raw2")
+async def api_raw2(file: UploadFile = Form(...)):
+    try:
+        file_content = await file.read()
+        file_stream = io.BytesIO(file_content)
+
+		raw1, raw2 = fn.raw(file_stream)
+
+        return Response(content=raw2, media_type="audio/wav")
+    except Exception as e:
+        return {"error": str(e)}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+yt-dlp
+python-multipart

venv.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/bash
+
+python3 -m venv venv
+curl -kL https://bootstrap.pypa.io/get-pip.py | venv/bin/python
+
+venv/bin/python -m pip install gradio
+venv/bin/python -m pip install -r requirements.txt