Update app.py

app.py

@@ -6,13 +6,16 @@ import pyperclip
 
 logging.basicConfig(level=logging.INFO)
 
-
+# Use the current working directory (Hugging Face compatible)
 DOWNLOAD_FOLDER = './output'
 
-
+# Ensure the directory exists
 if not os.path.exists(DOWNLOAD_FOLDER):
     os.makedirs(DOWNLOAD_FOLDER)
 
+# Specify the path for the cookies file (assuming it's in the same directory as this script)
+COOKIES_PATH = './cookies.txt'
+
 def create_readme():
     """Creates a readme.txt file with the specified text"""
     readme_path = os.path.join(DOWNLOAD_FOLDER, 'readme.txt')
@@ -24,14 +27,16 @@ def create_readme():
     except Exception as e:
         return f"Failed to create readme.txt: {str(e)}"
 
-def download_video(url, cookies_path=None):
+def download_video(url):
     try:
         output_path = os.path.join(DOWNLOAD_FOLDER, '%(title)s.mp4')
         command = ['yt-dlp', '-f', 'mp4', '-o', output_path, url]
 
-    
-        if cookies_path:
-            command.extend(['--cookies', cookies_path])
+        # Add cookies to the yt-dlp command
+        if os.path.exists(COOKIES_PATH):
+            command.extend(['--cookies', COOKIES_PATH])
+        else:
+            return f"Error: Cookies file {COOKIES_PATH} not found."
 
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         stdout, stderr = process.communicate()
@@ -99,32 +104,29 @@ def clear_input():
     """Clear the input and output"""
     return "", ""
 
-
+# Create the Gradio interface
 with gr.Blocks() as iface:
-
+    # Input for URL
     url_input = gr.Textbox(label="Enter YouTube URL", placeholder="Paste URL here")
-    
-  
-    cookies_input = gr.Textbox(label="Cookies File Path", placeholder="Enter path to cookies.txt (optional)")
 
-  
+    # Output for status
     status_output = gr.Textbox(label="Status", interactive=False)
 
- 
+    # Buttons for actions
     download_button = gr.Button("Download Video")
     android_button = gr.Button("Create Android Ringtone")
     iphone_button = gr.Button("Create iPhone Ringtone")
     paste_button = gr.Button("Paste URL from Clipboard")
     clear_button = gr.Button("Clear")
 
-  
-    download_button.click(download_video, inputs=[url_input, cookies_input], outputs=status_output)
+    # Button click actions
+    download_button.click(download_video, inputs=url_input, outputs=status_output)
     android_button.click(android_download, outputs=status_output)
     iphone_button.click(iphone_download, outputs=status_output)
     paste_button.click(paste_from_clipboard, outputs=url_input)
     clear_button.click(clear_input, outputs=[url_input, status_output])
 
-
+# Launch the interface
 iface.launch(share=True)
 
 
@@ -132,7 +134,3 @@ iface.launch(share=True)
 
 
 
-
-
-
-