Update app.py

app.py

@@ -1,34 +1,43 @@
 import ffmpeg
-import streamlit as st
+import gradio as gr
 import time
- 
-# ts stores the time in seconds
-ts = time.time()
+import os
 
-def convert_m3u8_to_mp4(url, output_path):
+# Function to convert M3U8 to MP4
+def convert_m3u8_to_mp4(url):
     try:
-        # Download and convert the m3u8 stream to mp4 using FFmpeg
-        ffmpeg.input(url).output(output_path, vcodec='copy', acodec='copy').run()
-        return True
+        # Generate a unique output file name based on the current timestamp
+        timestamp = str(int(time.time()))
+        output_path = f"/tmp/{timestamp}.mp4"
+
+        # Convert the M3U8 stream to MP4
+        ffmpeg.input(url).output(output_path, vcodec='copy', acodec='copy', format='mp4').run()
+
+        # Return the output file path for download
+        return output_path
     except ffmpeg.Error as e:
-        st.error(f"An error occurred: {e}")
-        return False
+        return f"Error occurred during conversion: {e}"
+
+# Function for the Gradio app
+def process_conversion(url):
+    # Perform conversion and check for errors
+    result = convert_m3u8_to_mp4(url)
+    if os.path.exists(result):
+        return result
+    else:
+        return "Conversion failed. Please check the URL or try again."
 
-def main():
-    st.title("M3U8 Stream to MP4 Converter")
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# M3U8 Stream to MP4 Converter")
+    gr.Markdown("Enter the URL of a `.m3u8` stream below to convert it into an MP4 file.")
     
-    url = st.text_input("Enter the URL of the .m3u8 stream")
-    if url:
-        ts = str(time.time())
-        output_path = "/tmp/" + ts + ".mp4"
-        
-        if st.button("Convert"):
-            success = convert_m3u8_to_mp4(url, output_path)
-            if success:
-                st.success("Conversion successful!")
-                st.download_button("Download MP4", data=open(output_path, "rb"), file_name= ts +".mp4")
-            else:
-                st.error("Conversion failed.")
-                
-if __name__ == "__main__":
-    main()
+    url_input = gr.Textbox(label="M3U8 Stream URL", placeholder="Enter .m3u8 stream URL here...")
+    convert_button = gr.Button("Convert")
+    output_file = gr.File(label="Download MP4 File")
+
+    # Define the workflow when the button is clicked
+    convert_button.click(fn=process_conversion, inputs=url_input, outputs=output_file)
+
+# Launch the Gradio app
+demo.launch()