Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,47 +1,73 @@
 import streamlit as st
 import subprocess
 import os
-import time
-
-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"):
-    if not url:
-        st.error("Please enter a valid URL.")
-    else:
-        try:
-            start_time = time.time()
-            
-            # Run FFmpeg command
-            command = [
+import zipfile
+import uuid
+import shutil
+
+st.set_page_config(page_title="MP4 → HLS Converter", layout="centered")
+
+st.title("🎬 MP4 → HLS (m3u8) Converter")
+st.caption("Uses FFmpeg • No re-encode • Streaming-safe")
+
+uploaded_file = st.file_uploader("Upload MP4 file", type=["mp4"])
+
+if uploaded_file:
+    job_id = str(uuid.uuid4())
+    workdir = f"/tmp/{job_id}"
+    os.makedirs(workdir, exist_ok=True)
+
+    input_path = os.path.join(workdir, "input.mp4")
+    output_path = os.path.join(workdir, "output.m3u8")
+    zip_path = os.path.join(workdir, "hls.zip")
+
+    # Save file
+    with open(input_path, "wb") as f:
+        f.write(uploaded_file.read())
+
+    st.success("File uploaded")
+
+    if st.button("Convert to HLS"):
+        with st.spinner("Running FFmpeg…"):
+            cmd = [
                 "ffmpeg",
-                "-y",                 # Overwrite output if exists
-                "-i", url,            # Input HLS URL
-                "-c", "copy",         # Copy codecs without re-encoding
-                output_file
+                "-y",
+                "-i", input_path,
+                "-codec", "copy",
+                "-start_number", "0",
+                "-hls_time", "6",
+                "-hls_list_size", "0",
+                "-f", "hls",
+                output_path
             ]
-            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}")
+
+            result = subprocess.run(
+                cmd,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE
+            )
+
+            if result.returncode != 0:
+                st.error("FFmpeg failed")
+                st.text(result.stderr.decode())
+                shutil.rmtree(workdir)
+                st.stop()
+
+        # Zip output
+        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zipf:
+            for file in os.listdir(workdir):
+                if file.endswith(".ts") or file.endswith(".m3u8"):
+                    zipf.write(
+                        os.path.join(workdir, file),
+                        arcname=file
+                    )
+
+        st.success("Conversion complete")
+
+        with open(zip_path, "rb") as f:
+            st.download_button(
+                label="⬇️ Download HLS (ZIP)",
+                data=f,
+                file_name="test2.zip",
+                mime="application/zip"
+            )