Update app.py

app.py

@@ -34,8 +34,9 @@ async def run_subprocess(cmd, use_fdkaac=False):
     print(f"[DEBUG] Command succeeded:\n{stdout.decode()}\n")
     return stdout.decode(), stderr.decode()
 
-async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate):
+async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate, progress):
     if use_mp3:
+        progress.update(1, desc="Converting audio to MP3...")
         output_audio = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp3")
         ffmpeg_audio_cmd = [
             'ffmpeg', '-y', '-i', input_path, '-vn',
@@ -73,24 +74,27 @@ async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only,
     video_output = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp4")
 
     # Extract mono WAV audio
+    progress.update(1, desc="Splitting Audio and Video...")
     await run_subprocess([
         'ffmpeg', '-y', '-i', input_path, '-ac', '1', '-ar', '8000', audio_wav
     ])
 
     # Compress WAV with fdkaac
+    progress.update(1, desc="Converting audio to AAC which is lower quality...")
     await run_subprocess([
         './fdkaac', '-b', '1k', '-C', '-f', '2', '-G', '1', '-w', '8000',
         '-o', audio_output, audio_wav
     ], use_fdkaac=True)
 
     # Compress video without audio
+    progress.update(1, desc="Making the video quality lower...")
     video_cmd = [
         'ffmpeg', '-y', '-hwaccel', accel, '-i', input_path
     ]
 
     if downscale:
         video_cmd += ['-vf', 'scale=-2:144']
-
+    
     if custom_bitrate:
         video_cmd += ['-b:v', f"{int(video_bitrate)}k"]
     else:
@@ -102,26 +106,16 @@ async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only,
     await run_subprocess(video_cmd)
 
     if audio_only:
-        # User wants audio only, no merged video
-        # Clean intermediates except audio_output
-        for f in [audio_wav, video_output]:
-            try:
-                os.remove(f)
-            except FileNotFoundError:
-                pass
         return audio_output, None
 
-    # Merge audio + video into final file
+    # Merging video + audio
+    progress.update(1, desc="Mixing audio and video together...")
     merged_output = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp4")
     await run_subprocess([
-        'ffmpeg', '-y',
-        '-i', video_output,
-        '-i', audio_output,
-        '-c', 'copy',
-        merged_output
+        'ffmpeg', '-y', '-i', video_output, '-i', audio_output, '-c', 'copy', merged_output
     ])
 
-    # Cleanup intermediate files
+    # Cleanup intermediates
     for f in [audio_wav, audio_output, video_output]:
         try:
             os.remove(f)
@@ -130,34 +124,47 @@ async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only,
 
     return None, merged_output
 
-async def process_conversion(use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate):
+async def process_conversion(use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate, progress=gr.Progress()):
     try:
-        if use_youtube:
-            if not youtube_url:
-                return "Error: YouTube URL required.", None
-            yt_uuid = str(uuid.uuid4())
-            yt_out = os.path.join(UPLOAD_FOLDER, yt_uuid + ".%(ext)s")
-            yt_cmd = ['yt-dlp', '-o', yt_out, '-f', 'b', youtube_url]
-            await run_subprocess(yt_cmd)
-            pattern = os.path.join(UPLOAD_FOLDER, yt_uuid + ".*")
-            files = glob.glob(pattern)
-            if not files:
-                return "Download failed.", None
-            input_path = files[0]
-        else:
-            if not video_file:
-                return "No video provided.", None
-            ext = os.path.splitext(video_file.name)[1]
-            input_path = os.path.join(UPLOAD_FOLDER, f"{uuid.uuid4()}{ext}")
-            shutil.copy2(video_file.name, input_path)
-
-        audio_out, video_out = await convert_video_task(
-            input_path, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate
-        )
+        with progress.tqdm(total=5, desc="Starting...") as pbar:
+            if use_youtube:
+                pbar.update(1, desc="Downloading video...")
+                if not youtube_url:
+                    return "Error: YouTube URL required.", None
+                yt_uuid = str(uuid.uuid4())
+                yt_out = os.path.join(UPLOAD_FOLDER, yt_uuid + ".%(ext)s")
+                yt_cmd = ['yt-dlp', '-o', yt_out, '-f', 'b', youtube_url]
+                await run_subprocess(yt_cmd)
+                pattern = os.path.join(UPLOAD_FOLDER, yt_uuid + ".*")
+                files = glob.glob(pattern)
+                if not files:
+                    return "Download failed.", None
+                input_path = files[0]
+            else:
+                pbar.update(1, desc="Preparing input video...")
+                if not video_file:
+                    return "No video provided.", None
+                ext = os.path.splitext(video_file.name)[1]
+                input_path = os.path.join(UPLOAD_FOLDER, f"{uuid.uuid4()}{ext}")
+                shutil.copy2(video_file.name, input_path)
+
+            if use_mp3:
+                pbar.update(1, desc="Converting audio to MP3...")
+            else:
+                pbar.update(1, desc="Converting audio to AAC which is lower quality...")
 
-        if audio_only:
-            return audio_out, audio_out
-        return video_out, video_out
+            audio_out, video_out = await convert_video_task(
+                input_path, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate, progress=pbar
+            )
+
+            if not use_mp3:
+                pbar.update(1, desc="Making the video quality lower...")
+
+            pbar.update(1, desc="Finishing up...")
+
+            if audio_only:
+                return audio_out, audio_out
+            return video_out, video_out
     except Exception as e:
         return f"Error: {str(e)}", None
 
@@ -174,9 +181,9 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="rose")) as demo:
 
     with gr.Group():
         with gr.Row():
-            use_youtube = gr.Checkbox(label="🔗 Use YouTube URL (This is useless on HF)", value=False)
+            use_youtube = gr.Checkbox(label="🔗 Use YouTube URL (Useless on Huggingface, works when ran locally probably.)", value=False)
             youtube_url = gr.Textbox(label="YouTube URL", placeholder="Paste YouTube URL here")
-        video_file = gr.File(label="Upload Video File")
+        video_file = gr.File(label="📁 Upload Video File")
 
     gr.Markdown("### ⚙️ **Conversion Settings**")
     with gr.Group():