Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import yt_dlp
 import os
+from pathlib import Path
 
 def download_video(url, quality, format_choice, download_playlist, cookies_file):
     try:
@@ -10,9 +11,13 @@ def download_video(url, quality, format_choice, download_playlist, cookies_file)
             'noplaylist': not download_playlist,  # Tải playlist nếu được chọn
         }
         
-        # Sử dụng file cookie nếu được cung cấp
-        if cookies_file and os.path.exists(cookies_file):
-            ydl_opts['cookiefile'] = cookies_file
+        # Sử dụng file cookie nếu được upload
+        if cookies_file is not None:
+            # Lưu file cookie tạm thời
+            cookie_path = "temp_cookies.txt"
+            with open(cookie_path, "wb") as f:
+                f.write(cookies_file)
+            ydl_opts['cookiefile'] = cookie_path
         
         # Điều chỉnh chất lượng video
         if quality == "Best":
@@ -44,6 +49,10 @@ def download_video(url, quality, format_choice, download_playlist, cookies_file)
             info = ydl.extract_info(url, download=True)
             filename = ydl.prepare_filename(info)
             
+        # Xóa file cookie tạm thời sau khi tải xong
+        if cookies_file is not None and os.path.exists(cookie_path):
+            os.remove(cookie_path)
+            
         return f"Video đã tải thành công: {filename}"
     
     except Exception as e:
@@ -71,15 +80,15 @@ interface = gr.Interface(
             value=False,
             info="Check để tải toàn bộ playlist"
         ),
-        gr.Textbox(
-            label="Đường dẫn file cookie (nếu cần)",
-            placeholder="Ví dụ: cookies.txt",
-            info="Dùng file cookie để tải video yêu cầu đăng nhập (YouTube, v.v.)"
+        gr.File(
+            label="Upload file cookie (nếu cần)",
+            type="binary",
+            info="Upload file cookie (.txt) để tải video yêu cầu đăng nhập"
         )
     ],
     outputs=gr.Textbox(label="Kết quả"),
-    title="Trình tải video nâng cao",
-    description="Dán URL video và tùy chỉnh. Nếu video yêu cầu đăng nhập, cung cấp file cookie."
+    title="Trình tải video trên Hugging Face Spaces",
+    description="Dán URL video và tùy chỉnh. Upload file cookie nếu video yêu cầu đăng nhập."
 )
 
 # Khởi chạy ứng dụng