Gradio Update

app.py

@@ -1,18 +1,10 @@
 import os
 import re
 import yt_dlp
+import requests
 import subprocess
 import gradio as gr
-
-# Install yt-dlp
-try:
-    
-    print("yt-dlp sudah terinstal.")
-except ImportError:
-    print("yt-dlp belum terinstal, menginstal sekarang...")
-    subprocess.run(['pip', 'install', 'yt-dlp', 'mutagen'])
-    import yt_dlp
-    print("yt-dlp berhasil diinstal.")
+from urllib.parse import urlparse
 
 # Deteksi orientasi video berdasarkan URL atau metadata
 def get_orientation(url):
@@ -26,13 +18,16 @@ def get_orientation(url):
         return None
 
 # Fungsi download video
-def video_download(url, video_folder = "/home/user/app/Video"):
-    if not os.path.exists(video_folder):
-        os.makedirs(video_folder)
+def video_download(url):
+    domain = urlparse(url).hostname.split('.')[-2].title()
     try:
-        result = subprocess.run(['yt-dlp', '--get-title', url], capture_output=True, text=True, check=True)
-        title = result.stdout.strip()
+        result = subprocess.run(['yt-dlp', '--print', '%(title)s\n%(uploader)s\n%(thumbnail)s', '--skip-download', url], capture_output=True, text=True, check=True)
+        title, uploader, thumbnail_url = result.stdout.strip().split('\n')
+        uploader = re.sub(r'[\\/:*?"<>|]', ' ', uploader)
         title = re.sub(r'[\\/:*?"<>|]', ' ', title)
+        if "ytimg" in thumbnail_url and thumbnail_url.endswith(".webp"):
+            thumbnail_url = thumbnail_url.replace("_webp", "").replace(".webp", ".jpg")
+        # print(f'Channel: {uploader}\nTitle: {title}\nThumbnail: {thumbnail_url}')
 
         if not title:
             raise ValueError("Tidak dapat mendapatkan judul video.")
@@ -46,61 +41,100 @@ def video_download(url, video_folder = "/home/user/app/Video"):
             print("Gagal mendeteksi orientasi video.")
             return None
 
-        video_file = os.path.join(video_folder, f"{title.replace('/', ' ')}.mp4")
-        subprocess.run(['yt-dlp', '-f', format_filter, '-o', video_file, url], check=True)
+        video_folder = os.path.join("Gradio", domain, uploader, "Video")
+        if not os.path.exists(video_folder):
+            os.makedirs(video_folder)
+        video_file = os.path.join(video_folder, f"{uploader} - {title}.mp4")
+
+        thumbnail_folder = os.path.join("Gradio", domain, uploader, "Thumbnail")
+        if not os.path.exists(thumbnail_folder):
+            os.makedirs(thumbnail_folder)
+        thumbnail_file = os.path.join(thumbnail_folder, f"{uploader} - {title}.jpg")
 
-        return video_file
+        if not os.path.exists(thumbnail_file):
+            subprocess.run(['wget', '-O', thumbnail_file, thumbnail_url])
+        if not os.path.exists(video_file):
+            subprocess.run(['yt-dlp', '-f', format_filter, '-o', video_file, url], check=True)
+
+        return video_file, thumbnail_file
 
     except subprocess.CalledProcessError as e:
-        print(f"Terjadi kesalahan saat menjalankan yt-dlp: {e}")
-        return None
+        return "", ""
     except Exception as e:
-        print(f"Terjadi kesalahan: {e}")
-        return None
+        return "", ""
 
 # Fungsi download MP3
-def audio_download(url, audio_folder = "/home/user/app/Audio"):
-    if not os.path.exists(audio_folder):
-        os.makedirs(audio_folder)
+def audio_download(url):
+    domain = urlparse(url).hostname.split('.')[-2].title()
     try:
-        if "youtube" not in url:
-            url = "https://www.youtube.com/watch?v=" + url.split("[")[-1].split("]")[0]
-        url = url.split("&")[0]
-
-        result = subprocess.run(['yt-dlp', '--get-title', url], capture_output=True, text=True, check=True)
-        title = result.stdout.strip()
+        result = subprocess.run(['yt-dlp', '--print', '%(title)s\n%(uploader)s\n%(thumbnail)s', '--skip-download', url], capture_output=True, text=True, check=True)
+        title, uploader, thumbnail_url = result.stdout.strip().split('\n')
+        uploader = re.sub(r'[\\/:*?"<>|]', ' ', uploader)
         title = re.sub(r'[\\/:*?"<>|]', ' ', title)
+        if "ytimg" in thumbnail_url and thumbnail_url.endswith(".webp"):
+            thumbnail_url = thumbnail_url.replace("_webp", "").replace(".webp", ".jpg")
+        # print(f'Channel: {uploader}\nTitle: {title}\nThumbnail: {thumbnail_url}')
 
         if not title:
             raise ValueError("Tidak dapat mendapatkan judul audio.")
 
-        audio_file = os.path.join(audio_folder, f"{title.replace('/', ' ')}.mp3")
-        subprocess.run(['yt-dlp', '-x', '--audio-format', 'mp3', '--embed-thumbnail', '-o', audio_file, url], check=True)
-        
-        return audio_file
+        audio_folder = os.path.join("Gradio", domain, uploader, "Audio")
+        if not os.path.exists(audio_folder):
+            os.makedirs(audio_folder)
+        audio_file = os.path.join(audio_folder, f"{uploader} - {title}.mp3")
+
+        thumbnail_folder = os.path.join("Gradio", domain, uploader, "Thumbnail")
+        if not os.path.exists(thumbnail_folder):
+            os.makedirs(thumbnail_folder)
+        thumbnail_file = os.path.join(thumbnail_folder, f"{uploader} - {title}.jpg")
+
+        if not os.path.exists(thumbnail_file):
+            subprocess.run(['wget', '-O', thumbnail_file, thumbnail_url])
+        if not os.path.exists(audio_file):
+            subprocess.run(['yt-dlp', '-x', '--audio-format', 'mp3', '--embed-thumbnail', '-o', audio_file, url], check=True)
+
+        return audio_file, thumbnail_file
 
     except subprocess.CalledProcessError as e:
-        print(f"Terjadi kesalahan saat menjalankan yt-dlp: {e}")
-        return None
+        return "", ""
     except Exception as e:
-        print(f"Terjadi kesalahan: {e}")
-        return None
+        return "", ""
 
 # Fungsi proses input untuk Gradio
 def process_input(url, download_type):
     if download_type == "Video":
-        return video_download(url)
+        file, thumbnail = video_download(url.split("&")[0])
     else:
-        return audio_download(url)
+        file, thumbnail = audio_download(url.split("&")[0])
+
+    return (
+        gr.update(label=download_type, value=file, visible=True),
+        thumbnail,
+        gr.update(label=f"Download {download_type}", value=file, visible=True),
+        gr.update(label="Download Thumbnail", value=thumbnail, visible=True)
+        )
 
 # Gradio interface
-with gr.Blocks() as youtube:
+css_url = "https://gilbertclaus.pythonanywhere.com/templates/gradio.css"
+
+with gr.Blocks(title="Video/Audio Downloader", css=requests.get(css_url).text) as youtube:
     gr.Markdown("# YouTube Downloader - Video dan Audio")
-    input_text = gr.Textbox(label="Masukkan URL YouTube")
-    download_type = gr.Radio(["Video", "Audio"], label="Pilih Format Download", value="Video")
+    with gr.Row():
+        input_text = gr.Textbox(label="Masukkan URL YouTube")
+        download_type = gr.Radio(["Video", "Audio"], label="Pilih Format Download", value="Video")
     download_button = gr.Button("Download")
-    output_file = gr.File(label="Download File")
-
-    download_button.click(process_input, inputs=[input_text, download_type], outputs=output_file)
+    with gr.Row():
+        thumbnail_file = gr.Image(label="Thumbnail")
+        with gr.Column():
+            output_file = gr.File(visible=False)
+            with gr.Row():
+                download_thumbnail = gr.DownloadButton(visible=False)
+                download_file = gr.DownloadButton(visible=False)
+
+    download_button.click(
+        process_input,
+        inputs=[input_text, download_type],
+        outputs=[output_file, thumbnail_file, download_file, download_thumbnail]
+        )
 
 youtube.launch(share=True)