Update app.py

app.py

@@ -2,21 +2,23 @@ import streamlit as st
 import tempfile
 import subprocess
 import os
+import re
 
 st.set_page_config(page_title="Video Speed & Compression Tool", layout="centered")
 
-st.title("🎬 Video Speed & Compression Tool (Ultra Fast Lossless Mode)")
+st.title("🎬 Video Speed & Compression Tool (Custom Speed + Always Compress)")
 
 uploaded_file = st.file_uploader("Upload a video file", type=["mp4", "mov", "avi", "mkv"])
 
-speed = st.selectbox(
-    "Select Playback Speed",
-    options=[1.2, 1.5, 2.0],
-    index=1
+# Let user choose any custom speed
+speed = st.number_input(
+    "Set Playback Speed (e.g., 1.2 for 20% faster, 0.5 for half speed)",
+    min_value=0.25,
+    max_value=4.0,
+    value=1.5,
+    step=0.05
 )
 
-compress = st.toggle("Compress Video?")
-
 if uploaded_file is not None:
     st.video(uploaded_file)
 
@@ -31,30 +33,48 @@ if uploaded_file is not None:
             video_speed = 1 / speed
             audio_speed = speed
 
-            if compress:
-                # Re-encode with compression
-                cmd = [
-                    "ffmpeg", "-i", input_path,
-                    "-filter_complex", f"[0:v]setpts={video_speed}*PTS[v];[0:a]atempo={audio_speed}[a]",
-                    "-map", "[v]", "-map", "[a]",
-                    "-vcodec", "libx264", "-crf", "28", "-preset", "fast",
-                    "-movflags", "+faststart",
-                    "-y", output_path
-                ]
-            else:
-                # Lossless mode (copy video, re-encode only audio, make it browser-playable)
-                cmd = [
-                    "ffmpeg", "-i", input_path,
-                    "-filter_complex", f"[0:v]setpts={video_speed}*PTS[v];[0:a]atempo={audio_speed}[a]",
-                    "-map", "[v]", "-map", "[a]",
-                    "-c:v", "copy", "-c:a", "aac", "-b:a", "192k",
-                    "-movflags", "+faststart",  # ensures browser playback
-                    "-y", output_path
-                ]
-
-            subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-
-            st.success("✅ Processing complete!")
+            # Always compress
+            cmd = [
+                "ffmpeg", "-i", input_path,
+                "-filter_complex", f"[0:v]setpts={video_speed}*PTS[v];[0:a]atempo={audio_speed}[a]",
+                "-map", "[v]", "-map", "[a]",
+                "-vcodec", "libx264", "-crf", "28", "-preset", "fast",
+                "-movflags", "+faststart",
+                "-y", output_path
+            ]
+
+            # Progress bar setup
+            progress_text = st.empty()
+            progress_bar = st.progress(0)
+
+            # Get video duration first
+            probe = subprocess.run(
+                ["ffprobe", "-v", "error", "-show_entries", "format=duration",
+                 "-of", "default=noprint_wrappers=1:nokey=1", input_path],
+                stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True
+            )
+            try:
+                total_duration = float(probe.stdout.strip())
+            except:
+                total_duration = None
+
+            # Run ffmpeg with live output parsing
+            process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1)
+
+            for line in process.stdout:
+                if "time=" in line and total_duration:
+                    # Extract current time processed
+                    match = re.search(r"time=(\d+:\d+:\d+\.\d+)", line)
+                    if match:
+                        h, m, s = match.group(1).split(":")
+                        elapsed = int(h) * 3600 + int(m) * 60 + float(s)
+                        percent = min(int((elapsed / total_duration) * 100), 100)
+                        progress_bar.progress(percent)
+                        progress_text.text(f"Processing... {percent}%")
+
+            process.wait()
+            progress_bar.progress(100)
+            progress_text.text("✅ Processing complete!")
 
             # Show processed video
             st.video(output_path)