Update app.py

app.py

@@ -6,23 +6,21 @@ import pyperclip
 
 logging.basicConfig(level=logging.INFO)
 
-# Use the current working directory (Hugging Face compatible)
 DOWNLOAD_FOLDER = './output'
+COOKIES_PATH = './cookies.txt'
 
-# Ensure the directory exists
+# Ensure the download folder 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')
-    
     try:
         with open(readme_path, 'w') as readme_file:
-            readme_file.write("Android copy the .mp3 to the ringtones folder and reboot device. iPhone drag and drop the .m4r file in iTunes and sync device.")
+            readme_file.write(
+                "Android: copy the .mp3 to the ringtones folder and reboot device.\n"
+                "iPhone: drag and drop the .m4r file in iTunes and sync device."
+            )
         return f"readme.txt created at: {readme_path}"
     except Exception as e:
         return f"Failed to create readme.txt: {str(e)}"
@@ -32,19 +30,16 @@ def download_video(url):
         output_path = os.path.join(DOWNLOAD_FOLDER, '%(title)s.mp4')
         command = ['yt-dlp', '-f', 'mp4', '-o', output_path, url]
 
-        # Add cookies to the yt-dlp command if the cookies.txt exists
         if os.path.exists(COOKIES_PATH):
             command.extend(['--cookies', COOKIES_PATH])
         else:
             return f"Error: Cookies file {COOKIES_PATH} not found."
 
-        # Debugging: Print the command to see if cookies are added correctly
         logging.info(f"Running command: {' '.join(command)}")
 
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         stdout, stderr = process.communicate()
 
-        # Debugging: Print the output to help identify the error
         logging.info(f"STDOUT: {stdout.decode()}")
         logging.error(f"STDERR: {stderr.decode()}")
 
@@ -63,44 +58,47 @@ def android_download():
     try:
         video_filename = get_video_filename(f'{DOWNLOAD_FOLDER}/%(title)s.mp4')
         if video_filename:
-            command = ['ffmpeg', '-i', video_filename, '-t', '20', '-q:a', '0', '-map', 'a', f'{DOWNLOAD_FOLDER}/AndroidRingtone.mp3']
+            android_path = f'{DOWNLOAD_FOLDER}/AndroidRingtone.mp3'
+            command = ['ffmpeg', '-i', video_filename, '-t', '20', '-q:a', '0', '-map', 'a', android_path]
             process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-            stdout, stderr = process.communicate()
+            process.communicate()
 
             if process.returncode == 0:
-                return "Android ringtone created successfully"
+                return "Android ringtone created successfully", android_path
             else:
-                return f"Error creating Android ringtone: {stderr.decode()}"
+                return f"Error creating Android ringtone: {process.stderr.read().decode()}", None
         else:
-            return "Error: Video file not found for Android ringtone."
+            return "Error: Video file not found for Android ringtone.", None
     except Exception as e:
-        return f"Failed to create Android ringtone: {str(e)}"
+        return f"Failed to create Android ringtone: {str(e)}", None
 
 def iphone_download():
     try:
-        command = ['ffmpeg', '-i', f'{DOWNLOAD_FOLDER}/AndroidRingtone.mp3', '-t', '20', '-acodec', 'aac', '-b:a', '128k', '-f', 'mp4', f'{DOWNLOAD_FOLDER}/iPhoneRingtone.m4r']
+        iphone_path = f'{DOWNLOAD_FOLDER}/iPhoneRingtone.m4r'
+        command = [
+            'ffmpeg',
+            '-i', f'{DOWNLOAD_FOLDER}/AndroidRingtone.mp3',
+            '-t', '20',
+            '-acodec', 'aac',
+            '-b:a', '128k',
+            '-f', 'mp4',
+            iphone_path
+        ]
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        stdout, stderr = process.communicate()
+        process.communicate()
 
         if process.returncode == 0:
-            return "iPhone ringtone created successfully"
+            return "iPhone ringtone created successfully", iphone_path
         else:
-            return f"Error creating iPhone ringtone: {stderr.decode()}"
+            return f"Error creating iPhone ringtone: {process.stderr.read().decode()}", None
     except Exception as e:
-        return f"Failed to create iPhone ringtone: {str(e)}"
+        return f"Failed to create iPhone ringtone: {str(e)}", None
 
 def get_video_filename(output_path):
     """Finds and returns the actual downloaded filename"""
-    title = os.path.basename(output_path).replace('%(title)s', '').strip()
-    video_filename = os.path.join(DOWNLOAD_FOLDER, f"{title}.mp4")
-
-    if os.path.exists(video_filename):
-        return video_filename
-    else:
-        for file in os.listdir(DOWNLOAD_FOLDER):
-            if file.endswith(".mp4"):
-                return os.path.join(DOWNLOAD_FOLDER, file)
-
+    for file in os.listdir(DOWNLOAD_FOLDER):
+        if file.endswith(".mp4"):
+            return os.path.join(DOWNLOAD_FOLDER, file)
     return None
 
 def paste_from_clipboard():
@@ -111,35 +109,34 @@ def clear_input():
     """Clear the input and output"""
     return "", ""
 
-# Create the Gradio interface
+# Gradio UI
 with gr.Blocks() as iface:
-    # Input for URL
+    gr.Markdown("## 🎵 YouTube Ringtone Creator")
     url_input = gr.Textbox(label="Enter YouTube URL", placeholder="Paste URL here")
-
-    # Output for status
     status_output = gr.Textbox(label="Status", interactive=False)
 
-    # Buttons for actions
+    android_download_link = gr.File(label="Download Android Ringtone", interactive=False)
+    iphone_download_link = gr.File(label="Download iPhone Ringtone", interactive=False)
+
     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")
 
-    # 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)
+    android_button.click(android_download, outputs=[status_output, android_download_link])
+    iphone_button.click(iphone_download, outputs=[status_output, iphone_download_link])
     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)
 
 
 
 
 
+