Update app.py

app.py

@@ -1,85 +1,72 @@
-import gradio as gr
-from moviepy.video.io.VideoFileClip import VideoFileClip
+import streamlit as st
+import moviepy.editor as mp
+import tempfile
 import os
 
-# Función para obtener duración del vídeo
-def get_video_duration(video_path):
-    clip = VideoFileClip(video_path)
-    duration = clip.duration
-    clip.close()
-    return duration
-
-# Función para cortar y redimensionar el vídeo
-def edit_video(video_file, start_time, end_time, resolution):
-    if video_file is None:
-        return "Sube un vídeo primero.", None
-
-    input_path = video_file.name
-    clip = VideoFileClip(input_path)
-
-    # Asegurar que los tiempos son válidos
-    if start_time < 0 or end_time > clip.duration or start_time >= end_time:
-        clip.close()
-        return "Tiempos inválidos. Asegúrate de elegir un rango correcto.", None
-
-    # Cortar el clip según tiempos proporcionados
-    edited_clip = clip.subclip(start_time, end_time)
-
-    # Definir resoluciones estándar
-    resolutions = {
-        "480p": (854, 480),
-        "720p": (1280, 720),
-        "1080p": (1920, 1080),
-        "1440p": (2560, 1440),
-        "4K": (3840, 2160)
-    }
-
-    # Redimensionar el vídeo manteniendo aspecto original
-    target_resolution = resolutions[resolution]
-    edited_clip = edited_clip.resize(height=target_resolution[1])
-
-    # Guardar resultado
-    output_filename = f"edited_{resolution}_{os.path.basename(input_path)}"
-    edited_clip.write_videofile(output_filename, codec="libx264", audio_codec="aac")
-
-    edited_clip.close()
-    clip.close()
-
-    return "Vídeo editado correctamente.", output_filename
-
-# Interfaz de Gradio
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎬 Editor de Vídeos Interactivo")
-
-    video_input = gr.File(label="Sube tu vídeo aquí", file_types=["video"])
-    duration_display = gr.Textbox(label="Duración del vídeo (segundos)", interactive=False)
-
-    with gr.Row():
-        start_time = gr.Number(label="Tiempo inicial (segundos)", value=0)
-        end_time = gr.Number(label="Tiempo final (segundos)", value=10)
-
-    resolution = gr.Dropdown(label="Resolución de exportación",
-                             choices=["480p", "720p", "1080p", "1440p", "4K"],
-                             value="1080p")
-
-    edit_button = gr.Button("Editar Vídeo")
-    status_display = gr.Textbox(label="Estado", interactive=False)
-    video_output = gr.Video(label="Vídeo editado")
-
-    # Actualizar duración cuando se sube un vídeo
-    def update_duration(video):
-        if video is not None:
-            duration = get_video_duration(video.name)
-            return duration
-        return ""
-
-    video_input.change(fn=update_duration, inputs=video_input, outputs=duration_display)
-
-    # Editar vídeo al pulsar el botón
-    edit_button.click(
-        fn=edit_video,
-        inputs=[video_input, start_time, end_time, resolution],
-        outputs=[status_display, video_output]
-    )
-
-demo.launch()
+def main():
+    st.title("Simple Video Editor")
+
+    # File uploaders
+    video_file = st.file_uploader("Upload a video file (MP4)", type=["mp4"])
+    audio_file = st.file_uploader("Upload an audio file (MP3 or WAV)", type=["mp3", "wav"])
+
+    if video_file and audio_file:
+        # Save uploaded files to temporary locations
+        temp_video = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+        temp_video.write(video_file.read())
+        temp_video_path = temp_video.name
+
+        temp_audio = tempfile.NamedTemporaryFile(delete=False, suffix=f".{audio_file.name.split('.')[-1]}")
+        temp_audio.write(audio_file.read())
+        temp_audio_path = temp_audio.name
+
+        # Load video and audio
+        video = mp.VideoFileClip(temp_video_path)
+        audio = mp.AudioFileClip(temp_audio_path)
+
+        # Display video info
+        st.write(f"Video duration: {video.duration:.2f} seconds")
+        st.write(f"Audio duration: {audio.duration:.2f} seconds")
+
+        # Trim video
+        st.subheader("Trim Video")
+        video_start = st.slider("Video start time (seconds)", 0.0, video.duration, 0.0)
+        video_end = st.slider("Video end time (seconds)", 0.0, video.duration, video.duration)
+        trimmed_video = video.subclip(video_start, video_end)
+
+        # Trim audio
+        st.subheader("Trim Audio")
+        audio_start = st.slider("Audio start time (seconds)", 0.0, audio.duration, 0.0)
+        audio_end = st.slider("Audio end time (seconds)", 0.0, audio.duration, audio.duration)
+        trimmed_audio = audio.subclip(audio_start, audio_end)
+
+        # Combine video and audio
+        if st.button("Combine Video and Audio"):
+            final_clip = trimmed_video.set_audio(trimmed_audio)
+            
+            # Save the final video
+            output_path = "output_video.mp4"
+            final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac")
+
+            # Provide download link
+            st.success("Video processing complete!")
+            with open(output_path, "rb") as file:
+                btn = st.download_button(
+                    label="Download processed video",
+                    data=file,
+                    file_name="processed_video.mp4",
+                    mime="video/mp4"
+                )
+            
+            # Display video player
+            st.subheader("Preview Processed Video")
+            st.video(output_path)
+
+        # Clean up temporary files
+        video.close()
+        audio.close()
+        os.unlink(temp_video_path)
+        os.unlink(temp_audio_path)
+
+if __name__ == "__main__":
+    main()