Update app.py

app.py

@@ -9,18 +9,15 @@ logging.basicConfig(level=logging.INFO)
 DOWNLOAD_FOLDER = './output'
 COOKIES_PATH = './cookies.txt'
 
-# Ensure the download folder exists
-os.makedirs(DOWNLOAD_FOLDER, exist_ok=True)
+if not os.path.exists(DOWNLOAD_FOLDER):
+    os.makedirs(DOWNLOAD_FOLDER)
 
 def create_readme():
-    readme_path = os.path.join(DOWNLOAD_FOLDER, 'readme.txt')
+    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.\n"
-                "iPhone: drag and drop the .m4r file in iTunes and sync device."
-            )
-        return f"readme.txt created at: {readme_path}"
+        with open(path, 'w') as f:
+            f.write("Android: Copy .mp3 to ringtones folder and reboot. iPhone: Drag .m4r into iTunes and sync.")
+        return f"readme.txt created at: {path}"
     except Exception as e:
         return f"Failed to create readme.txt: {str(e)}"
 
@@ -32,54 +29,54 @@ def get_video_filename():
 
 def download_video(url):
     try:
-        output_template = os.path.join(DOWNLOAD_FOLDER, '%(title)s.%(ext)s')
-        command = ['yt-dlp', '-f', 'mp4', '-o', output_template, url]
+        output_path = os.path.join(DOWNLOAD_FOLDER, '%(title)s.%(ext)s')
+        command = ['yt-dlp', '-f', 'mp4', '-o', output_path, url]
 
         if os.path.exists(COOKIES_PATH):
-            command += ['--cookies', COOKIES_PATH]
+            command.extend(['--cookies', COOKIES_PATH])
         else:
             return f"Error: Cookies file {COOKIES_PATH} not found."
 
-        logging.info(f"Running command: {' '.join(command)}")
+        logging.info(f"Running: {' '.join(command)}")
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         stdout, stderr = process.communicate()
 
         logging.info(stdout.decode())
-        if process.returncode != 0:
-            return f"Error downloading video: {stderr.decode()}"
-
-        video_filename = get_video_filename()
-        if not video_filename:
-            return "Error: Downloaded video file not found."
-
-        return f"Video downloaded: {video_filename}\n" + create_readme()
+        logging.error(stderr.decode())
 
+        if process.returncode == 0:
+            video_file = get_video_filename()
+            if video_file:
+                return f"Video downloaded: {os.path.basename(video_file)}\n" + create_readme()
+            else:
+                return "Error: Video file not found."
+        else:
+            return f"Download error: {stderr.decode()}"
     except Exception as e:
-        return f"Exception during video download: {str(e)}"
-        
+        return f"Failed to download video: {str(e)}"
+
 def android_download():
     try:
-        video_filename = get_video_filename(f'{DOWNLOAD_FOLDER}/%(title)s.mp4')
-        output_file = f'{DOWNLOAD_FOLDER}/AndroidRingtone.mp3'
-        if video_filename:
-            command = ['ffmpeg', '-i', video_filename, '-t', '20', '-q:a', '0', '-map', 'a', output_file]
-            process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-            stdout, stderr = process.communicate()
-
-            if process.returncode == 0:
-                return "Android ringtone created successfully", output_file
-            else:
-                return f"Error creating Android ringtone: {stderr.decode()}", None
+        video_file = get_video_filename()
+        if not video_file:
+            return "Error: No video file found.", None
+
+        output_file = os.path.join(DOWNLOAD_FOLDER, 'AndroidRingtone.mp3')
+        command = ['ffmpeg', '-i', video_file, '-t', '20', '-q:a', '0', '-map', 'a', output_file]
+        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        stdout, stderr = process.communicate()
+
+        if process.returncode == 0:
+            return "Android ringtone created successfully", output_file
         else:
-            return "Error: Video file not found for Android ringtone.", None
+            return f"FFmpeg error: {stderr.decode()}", None
     except Exception as e:
-        return f"Failed to create Android ringtone: {str(e)}", None
-
+        return f"Error creating Android ringtone: {str(e)}", None
 
 def iphone_download():
     try:
-        input_file = f'{DOWNLOAD_FOLDER}/AndroidRingtone.mp3'
-        output_file = f'{DOWNLOAD_FOLDER}/iPhoneRingtone.m4r'
+        input_file = os.path.join(DOWNLOAD_FOLDER, 'AndroidRingtone.mp3')
+        output_file = os.path.join(DOWNLOAD_FOLDER, 'iPhoneRingtone.m4r')
         command = ['ffmpeg', '-i', input_file, '-t', '20', '-acodec', 'aac', '-b:a', '128k', '-f', 'mp4', output_file]
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         stdout, stderr = process.communicate()
@@ -87,46 +84,40 @@ def iphone_download():
         if process.returncode == 0:
             return "iPhone ringtone created successfully", output_file
         else:
-            return f"Error creating iPhone ringtone: {stderr.decode()}", None
+            return f"FFmpeg error: {stderr.decode()}", None
     except Exception as e:
-        return f"Failed to create iPhone ringtone: {str(e)}", None
-
+        return f"Error creating iPhone ringtone: {str(e)}", None
 
 def paste_from_clipboard():
-    try:
-        return pyperclip.paste()
-    except Exception as e:
-        return f"Clipboard error: {str(e)}"
+    return pyperclip.paste()
 
 def clear_input():
-    return "", ""
-
-# Gradio UI
+    return "", "", None, None
 
 with gr.Blocks() as iface:
-    gr.Markdown("## 🎵 YouTube Ringtone Creator")
-    url_input = gr.Textbox(label="Enter YouTube URL", placeholder="Paste URL here")
-    status_output = gr.Textbox(label="Status", interactive=False)
-
-    android_download_link = gr.File(label="Android Ringtone", interactive=False)
-    iphone_download_link = gr.File(label="iPhone Ringtone", interactive=False)
+    gr.Markdown("# 🎵 YouTube to Ringtone Converter")
 
+    url_input = gr.Textbox(label="YouTube URL", placeholder="Paste URL here")
+    status_output = gr.Textbox(label="Status", interactive=False)
 
-    android_file = gr.File(label="Download Android Ringtone (.mp3)")
-    iphone_file = gr.File(label="Download iPhone Ringtone (.m4r)")
+    with gr.Row():
+        download_button = gr.Button("📥 Download Video")
+        android_button = gr.Button("📱 Android Ringtone")
+        iphone_button = gr.Button("🍏 iPhone Ringtone")
 
+    with gr.Row():
+        paste_button = gr.Button("📋 Paste from Clipboard")
+        clear_button = gr.Button("🧹 Clear")
 
-    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")
+    with gr.Row():
+        android_file = gr.File(label="⬇️ Android Ringtone (.mp3)", interactive=False)
+        iphone_file = gr.File(label="⬇️ iPhone Ringtone (.m4r)", interactive=False)
 
-    download_button.click(fn=download_video, inputs=url_input, outputs=status_output)
+    download_button.click(download_video, inputs=url_input, outputs=status_output)
     android_button.click(android_download, outputs=[status_output, android_file])
     iphone_button.click(iphone_download, outputs=[status_output, iphone_file])
-    paste_button.click(fn=paste_from_clipboard, outputs=url_input)
-    clear_button.click(fn=clear_input, outputs=[url_input, status_output])
+    paste_button.click(paste_from_clipboard, outputs=url_input)
+    clear_button.click(clear_input, outputs=[url_input, status_output, android_file, iphone_file])
 
 iface.launch(share=True)
 
@@ -136,6 +127,7 @@ iface.launch(share=True)
 
 
 
+