Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,71 +1,47 @@
 import streamlit as st
-import requests
-import m3u8
+import subprocess
 import os
 import time
-from moviepy.editor import concatenate_videoclips, VideoFileClip
 
-st.title("HLS to MP4 Downloader")
+st.title("HLS to MP4 Downloader (FFmpeg)")
 
 url = st.text_input("Enter HLS stream URL (.m3u8):")
+output_file = st.text_input("Output MP4 filename:", "output.mp4")
 
-if st.button("Download as MP4"):
+if st.button("Download"):
     if not url:
         st.error("Please enter a valid URL.")
     else:
         try:
             start_time = time.time()
-
-            # Load m3u8 playlist
-            playlist = m3u8.load(url)
-            segments = playlist.segments
-            if not segments:
-                st.error("No segments found in playlist.")
-            else:
-                st.info(f"Found {len(segments)} segments. Downloading...")
-                ts_files = []
-                total_bytes = 0
-
-                # Download each segment
-                for i, seg in enumerate(segments):
-                    seg_url = seg.absolute_uri
-                    ts_filename = f"seg_{i}.ts"
-                    ts_files.append(ts_filename)
-                    r = requests.get(seg_url, stream=True)
-                    with open(ts_filename, "wb") as f:
-                        for chunk in r.iter_content(chunk_size=8192):
-                            if chunk:
-                                f.write(chunk)
-                                total_bytes += len(chunk)
-                    st.write(f"Downloaded segment {i+1}/{len(segments)}")
-
-                # Convert TS segments to MP4 using moviepy
-                st.info("Merging segments into MP4...")
-                clips = [VideoFileClip(f) for f in ts_files]
-                final_clip = concatenate_videoclips(clips)
-                output_file = "output.mp4"
-                final_clip.write_videofile(output_file, codec="libx264", audio_codec="aac")
-                final_clip.close()
-
-                # Clean up TS files
-                for f in ts_files:
-                    os.remove(f)
-
-                end_time = time.time()
-                duration = end_time - start_time
-                file_size_mb = os.path.getsize(output_file) / (1024 * 1024)
-
-                metadata = {
-                    "File Name": output_file,
-                    "File Path": os.path.abspath(output_file),
-                    "File Size (MB)": round(file_size_mb, 2),
-                    "Download Time (s)": round(duration, 2),
-                    "URL": url,
-                    "Total Segments": len(segments)
-                }
-
-                st.success(f"Downloaded and merged to '{output_file}' successfully!")
-                st.json(metadata)
-
+            
+            # Run FFmpeg command
+            command = [
+                "ffmpeg",
+                "-y",                 # Overwrite output if exists
+                "-i", url,            # Input HLS URL
+                "-c", "copy",         # Copy codecs without re-encoding
+                output_file
+            ]
+            st.info("Downloading stream using FFmpeg...")
+            subprocess.run(command, check=True)
+
+            end_time = time.time()
+            duration = end_time - start_time
+            file_size_mb = os.path.getsize(output_file) / (1024 * 1024)
+
+            metadata = {
+                "File Name": output_file,
+                "File Path": os.path.abspath(output_file),
+                "File Size (MB)": round(file_size_mb, 2),
+                "Download Time (s)": round(duration, 2),
+                "URL": url
+            }
+
+            st.success(f"Downloaded successfully to '{output_file}'!")
+            st.json(metadata)
+
+        except subprocess.CalledProcessError as e:
+            st.error(f"FFmpeg failed: {e}")
         except Exception as e:
             st.error(f"Error: {e}")