Update app.py

app.py

@@ -1,12 +1,10 @@
-import streamlit as st
+import gradio as gr
 import ffmpeg
 import os
 from PIL import Image
 import re
 import tempfile
 
-st.set_page_config(layout="wide", page_title="Video Conversion Tool")
-
 # Supported formats
 supported_formats = sorted(['3GP', 'ASF', 'AVI', 'DIVX', 'FLV', 'M2TS', 'M4V', 'MKV', 'MOV', 'MP4', 'MPEG', 'MPG', 'MTS', 'TS', 'VOB', 'WEBM', 'WMV', 'XVID'])
 audio_formats = sorted(['MP3', 'WAV', 'AAC', 'FLAC', 'OGG', 'M4A', 'ALAC', 'WMA', 'AIFF', 'OPUS', 'APE', 'CAF', 'PCM', 'DTS', 'TTA', 'AMR', 'MID', 'SPX', 'WV', 'RA', 'TAK'])
@@ -39,7 +37,7 @@ def convert_video(video, target_format, conversion_type, time_in_seconds=None):
         video_path = os.path.join(temp_dir, f"{sanitized_base_name}.mp4")  # Saving as mp4 by default for now
 
         with open(video_path, "wb") as f:
-            f.write(video.getbuffer())  # Save the uploaded video to a local file
+            f.write(video)  # Save the uploaded video to a local file
 
         if conversion_type == 'Video to Video':
             output_file = f"flowly_ai_video_converter_{sanitized_base_name}.{target_format.lower()}"
@@ -86,64 +84,27 @@ def update_format_choices(conversion_type):
         return image_formats
     return []
 
-def main():
-    st.title("Video Conversion Tool")
-    st.write("Convert videos to audio, GIFs, images, or other formats easily with this powerful tool.")
-
-    # Create two columns
-    col1, col2 = st.columns([1, 1])
-
-    with col1:
-        # Upload video file
-        video_file = st.file_uploader("Upload a Video", type=supported_formats)
-        if video_file:
-            st.video(video_file)
-
-    with col2:
-        if video_file:
-            # Create a temporary file and save uploaded video
-            temp_video_path = os.path.join(tempfile.mkdtemp(), video_file.name)
-            with open(temp_video_path, "wb") as f:
-                f.write(video_file.getbuffer())
-
-            # Get video duration
-            video_duration = get_video_duration(temp_video_path)
-
-            # Select conversion type
-            conversion_type = st.selectbox(
-                "Select Conversion Type", 
-                ['Video to Video', 'Video to Audio', 'Video to GIF', 'Video to Image']
-            )
-
-            # Update format choices based on conversion type
-            target_format_choices = update_format_choices(conversion_type)
-            target_format = st.selectbox("Select Target Format", target_format_choices)
-
-            # If 'Video to Image' conversion, ask for time in seconds
-            if conversion_type == 'Video to Image':
-                time_in_seconds = st.slider(
-                    label="Time (in seconds) for image extraction",
-                    min_value=0, 
-                    max_value=int(video_duration),
-                    value=0,
-                    step=1
-                )
-            else:
-                time_in_seconds = None
-
-            if st.button("Convert"):
-                with st.spinner("Converting..."):
-                    output_file = convert_video(video_file, target_format, conversion_type, time_in_seconds)
-                    
-                    if "Error" in output_file:
-                        st.error(output_file)
-                    else:
-                        st.success(f"Conversion Successful! Download the file:")
-                        st.download_button(
-                            label="Download Converted File", 
-                            data=open(output_file, 'rb').read(), 
-                            file_name=output_file
-                        )
+def convert_function(video_file, conversion_type, target_format, time_in_seconds=None):
+    # Convert the video based on user input
+    video_file_content = video_file.read()
+    return convert_video(video_file_content, target_format, conversion_type, time_in_seconds)
+
+def gradio_interface():
+    # Set up Gradio interface
+    with gr.Blocks() as demo:
+        gr.Markdown("# Video Conversion Tool")
+        gr.Markdown("Convert videos to audio, GIFs, images, or other formats easily with this powerful tool.")
+        
+        with gr.Row():
+            video_file = gr.File(label="Upload Video File", type="file", file_types=supported_formats)
+            conversion_type = gr.Dropdown(label="Select Conversion Type", choices=['Video to Video', 'Video to Audio', 'Video to GIF', 'Video to Image'])
+            target_format = gr.Dropdown(label="Select Target Format", choices=supported_formats)
+            time_in_seconds = gr.Slider(label="Time (in seconds) for image extraction", minimum=0, maximum=300, step=1, value=0)
+            output_file = gr.File(label="Converted File", interactive=False)
+
+        video_file.change(fn=convert_function, inputs=[video_file, conversion_type, target_format, time_in_seconds], outputs=output_file)
+
+    demo.launch()
 
 if __name__ == "__main__":
-    main()
+    gradio_interface()