Create app.py

app.py

@@ -0,0 +1,77 @@
+import gradio as gr
+import yt_dlp
+import os
+
+def download_video(url, quality, format_choice, download_playlist):
+    try:
+        # Tùy chọn cấu hình cho yt-dlp
+        ydl_opts = {
+            'outtmpl': 'downloads/%(title)s.%(ext)s',  # Đường dẫn lưu file
+            'noplaylist': not download_playlist,  # Tải playlist nếu được chọn
+        }
+        
+        # Điều chỉnh chất lượng video
+        if quality == "Best":
+            ydl_opts['format'] = 'bestvideo+bestaudio/best'
+        elif quality == "Medium":
+            ydl_opts['format'] = 'bestvideo[height<=720]+bestaudio/best[height<=720]'
+        else:  # Low
+            ydl_opts['format'] = 'bestvideo[height<=480]+bestaudio/best[height<=480]'
+        
+        # Điều chỉnh định dạng đầu ra
+        if format_choice == "MP4":
+            ydl_opts['merge_output_format'] = 'mp4'
+        elif format_choice == "MKV":
+            ydl_opts['merge_output_format'] = 'mkv'
+        else:  # Audio Only (MP3)
+            ydl_opts['format'] = 'bestaudio'
+            ydl_opts['postprocessors'] = [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '192',
+            }]
+        
+        # Tạo thư mục downloads nếu chưa tồn tại
+        if not os.path.exists('downloads'):
+            os.makedirs('downloads')
+            
+        # Tải video
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            filename = ydl.prepare_filename(info)
+            
+        return f"Video đã tải thành công: {filename}"
+    
+    except Exception as e:
+        return f"Lỗi: {str(e)}"
+
+# Tạo giao diện Gradio
+interface = gr.Interface(
+    fn=download_video,
+    inputs=[
+        gr.Textbox(label="Dán URL video vào đây", placeholder="https://..."),
+        gr.Dropdown(
+            choices=["Best", "Medium", "Low"],
+            label="Chất lượng video",
+            value="Best",
+            info="Best: Cao nhất có thể, Medium: 720p, Low: 480p"
+        ),
+        gr.Dropdown(
+            choices=["MP4", "MKV", "Audio Only (MP3)"],
+            label="Định dạng đầu ra",
+            value="MP4",
+            info="Chọn định dạng file xuất ra"
+        ),
+        gr.Checkbox(
+            label="Tải toàn bộ playlist (nếu có)",
+            value=False,
+            info="Check để tải toàn bộ playlist thay vì chỉ một video"
+        )
+    ],
+    outputs=gr.Textbox(label="Kết quả"),
+    title="Trình tải video nâng cao",
+    description="Dán URL video từ bất kỳ trang nào hỗ trợ (YouTube, Facebook, v.v.) và tùy chỉnh cách tải."
+)
+
+# Khởi chạy ứng dụng
+interface.launch()