Update app.py

app.py

@@ -1,12 +1,20 @@
+import os
+import tempfile
 import gradio as gr
 import yt_dlp
-import tempfile
-import os
 
 
 def get_formats(url):
+    if not url or not url.strip():
+        return gr.update(choices=[], value=None)
+
     try:
-        ydl_opts = {"quiet": True, "noplaylist": True}
+        ydl_opts = {
+            "quiet": True,
+            "noplaylist": True,
+            "skip_download": True,
+        }
+
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=False)
 
@@ -14,20 +22,29 @@ def get_formats(url):
         for f in info.get("formats", []):
             format_id = f.get("format_id")
             ext = f.get("ext", "unknown")
-            resolution = f.get("resolution") or f.get("format_note") or "audio"
+            resolution = (
+                f.get("resolution")
+                or f.get("format_note")
+                or ("audio" if f.get("vcodec") == "none" else "unknown")
+            )
+
             if format_id:
                 formats.append((f"{format_id} - {ext} - {resolution}", format_id))
 
         return gr.update(choices=formats, value=formats[0][1] if formats else None)
-    except Exception:
-        return gr.update(choices=[], value=None)
+
+    except Exception as e:
+        raise gr.Error(f"Không lấy được format: {str(e)}")
 
 
 def download_video(url, format_id):
-    if not url or not format_id:
-        raise gr.Error("Vui lòng nhập URL và chọn format.")
+    if not url:
+        raise gr.Error("Vui lòng nhập URL.")
+    if not format_id:
+        raise gr.Error("Vui lòng chọn format.")
 
     temp_dir = tempfile.mkdtemp()
+
     ydl_opts = {
         "format": format_id,
         "outtmpl": os.path.join(temp_dir, "%(title)s.%(ext)s"),
@@ -35,27 +52,33 @@ def download_video(url, format_id):
         "noplaylist": True,
     }
 
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info = ydl.extract_info(url, download=True)
-        filename = ydl.prepare_filename(info)
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            filename = ydl.prepare_filename(info)
+
+        if not os.path.exists(filename):
+            raise gr.Error("Không tìm thấy file sau khi tải.")
 
-    return filename, filename
+        return filename, filename
+
+    except Exception as e:
+        raise gr.Error(f"Tải thất bại: {str(e)}")
 
 
 with gr.Blocks() as demo:
     gr.Markdown("## Audio/Video Downloader with Preview")
-    gr.Markdown(
-        "Chỉ tải nội dung khi bạn có quyền hợp pháp để tải và sử dụng nội dung đó."
-    )
+    gr.Markdown("Chỉ dùng với nội dung bạn có quyền tải và sử dụng.")
 
     url_input = gr.Textbox(label="Video URL")
     format_dropdown = gr.Dropdown(label="Select download format", choices=[])
+    load_btn = gr.Button("Load Formats")
     download_btn = gr.Button("Download")
 
     output_file = gr.File(label="Download File")
     output_video = gr.Video(label="Preview Media")
 
-    url_input.change(
+    load_btn.click(
         fn=get_formats,
         inputs=url_input,
         outputs=format_dropdown