Update app.py

app.py

@@ -1,84 +1,52 @@
 import gradio as gr
-from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip
-from moviepy.audio.fx.all import audio_fadeout
+from moviepy.editor import VideoFileClip, concatenate_videoclips
 import random
-import subprocess
-import os
 
-# Función para convertir MOV a MP4
-def convertir_mov_a_mp4(ruta_mov):
-    ruta_mp4 = ruta_mov.replace('.MOV', '.mp4')
-    subprocess.run(['ffmpeg', '-i', ruta_mov, '-vcodec', 'h264', '-acodec', 'aac', ruta_mp4], check=True)
-    return ruta_mp4
-
-# Función para procesar los videos
-def process_videos(videos, rotations, song):
+def process_videos(videos, rotate_option):
     clips = []
-    errores = []  # Para almacenar los videos que no se pudieron procesar
-    
-    # Procesar cada video cargado
-    for idx, video_path in enumerate(videos):
-        try:
-            if video_path.endswith('.MOV'):
-                # Convertir MOV a MP4
-                video_path = convertir_mov_a_mp4(video_path)
-            if video_path.endswith(('.mp4')):
-                clip = VideoFileClip(video_path).subclip(2)  # Cortar los primeros 2 segundos
-
-                # Aplicar la rotación seleccionada por el usuario para cada video
-                if rotations[idx] == "90 grados":
-                    clip = clip.rotate(90)
-                elif rotations[idx] == "180 grados":
-                    clip = clip.rotate(180)
-                elif rotations[idx] == "270 grados":
-                    clip = clip.rotate(270)
-
-                clips.append(clip)
-            else:
-                errores.append(f"Archivo no compatible: {video_path}")
-        except Exception as e:
-            errores.append(f"Error al procesar {video_path}: {str(e)}")
     
-    # Si no se procesó ningún video, mostrar los errores
-    if len(clips) == 0:
-        return None, f"No se pudieron procesar los videos. Errores: {', '.join(errores)}"
+    # Cargar y recortar cada video a los primeros 3 segundos
+    for video in videos:
+        clip = VideoFileClip(video.name)
+        
+        # Recortar el video a los primeros 3 segundos
+        if clip.duration > 3:
+            clip = clip.subclip(0, 3)
+        
+        # Aplicar la rotación seleccionada
+        if rotate_option == "90 grados":
+            clip = clip.rotate(90)
+        elif rotate_option == "180 grados":
+            clip = clip.rotate(180)
+        elif rotate_option == "270 grados":
+            clip = clip.rotate(270)
+        
+        clips.append(clip)
     
     # Mezclar los clips en orden aleatorio
     random.shuffle(clips)
     
-    # Unir los clips
-    try:
-        final_clip = concatenate_videoclips(clips)
-
-        # Añadir la canción y hacer que termine con un fade out
-        if song:
-            song_audio = AudioFileClip(song)
-            song_audio = song_audio.subclip(0, final_clip.duration)
-            song_audio = audio_fadeout(song_audio, 3)  # Añadir fade out de 3 segundos al final de la canción
-            final_clip = final_clip.set_audio(song_audio)
-
-        # Guardar el video resultante
-        output_path = "output_video.mp4"
-        final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac")
-        
-        return output_path, None  # Devolver el archivo generado para la descarga y None para el error
-    except Exception as e:
-        return None, f"Error al concatenar los clips: {str(e)}"
+    # Unir los clips mezclados
+    final_clip = concatenate_videoclips(clips)
+    
+    # Guardar el video unido en un archivo
+    output_path = "output_video_rotated.mp4"
+    final_clip.write_videofile(output_path, codec="libx264")
+    
+    # Devolver la ruta del archivo final
+    return output_path
 
 # Crear la interfaz de Gradio
-video_inputs = [gr.Video(label=f"Sube el video {i+1}") for i in range(5)]  # Eliminar el argumento `type`
-rotation_inputs = [gr.Radio(["Sin rotación", "90 grados", "180 grados", "270 grados"], label=f"Rotar video {i+1}") for i in range(5)]
-
-demo = gr.Interface(
+iface = gr.Interface(
     fn=process_videos,
-    inputs=video_inputs + rotation_inputs + [gr.Audio(label="Sube una canción")],
-    outputs=[
-        gr.Video(label="Previsualiza tu video procesado"),  # Para mostrar el video en un reproductor
-        gr.File(label="Descargar video procesado")  # Para permitir la descarga del video final
+    inputs=[
+        gr.Video(type="file", label="Sube tus videos", file_count="multiple"),
+        gr.Radio(["Sin rotación", "90 grados", "180 grados", "270 grados"], label="Rotar video")
     ],
-    title="Mezclar y Unir Videos con Rotación y Música",
-    description="Sube varios videos, corta los primeros 2 segundos, ajusta la rotación de cada uno, y añade música.",
+    outputs=gr.File(label="Video unido de salida"),
+    title="Mezclar y Unir Videos con Rotación",
+    description="Sube varios videos en formato 9:16. El sistema cortará los primeros 3 segundos de cada video, los mezclará en un orden aleatorio, aplicará la rotación seleccionada y generará un nuevo archivo de video."
 )
 
 # Ejecutar la interfaz
-demo.launch(share=True)
+iface.launch()