Update app.py

app.py

@@ -9,6 +9,7 @@ from moviepy.editor import vfx
 from streamlit_option_menu import option_menu
 from youtube import youtube, download_youtube
 from moviepy.video.io.VideoFileClip import VideoFileClip
+import tempfile
 
 def session(info_list):
     st.video(info_list[1])
@@ -16,10 +17,8 @@ def session(info_list):
     if len(info_list) == 5:
         st.text(f"Duration: {info_list[3]} - {info_list[4]} seconds")
 
-def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mute_audio=False, use_aspect_ratio=None):
-    clip = VideoFileClip(video_file)
-    
-    total_duration = clip.duration
+def random_cuts(video_clip, title, num_cuts=1, duration=60, speed_factor=1.0, mute_audio=False, use_aspect_ratio=None):
+    total_duration = video_clip.duration
     cuts = []
 
     for _ in range(num_cuts):
@@ -35,36 +34,29 @@ def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mu
             continue
 
         output_file = f"{title}_cut_{i+1}.mp4"
-        clipped_clip = clip.subclip(start_time, end_time)
-        
-        if mute_audio:
-            clipped_clip = clipped_clip.set_audio(None)
-        
-        clipped_clip = clipped_clip.fx(vfx.speedx, speed_factor)
-        
-        if use_aspect_ratio:
-            original_width, original_height = clip.size
-        
-            if use_aspect_ratio == '16:9':
-                target_ratio = 16 / 9
-            elif use_aspect_ratio == '9:16':
-                target_ratio = 9 / 16
-            elif use_aspect_ratio == '21:9':
-                target_ratio = 21 / 9
-            else:
-                target_ratio = 16 / 9
-        
+        temp_filename = None
+
+        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_file:
+            temp_filename = temp_file.name
+
+            clipped_clip = video_clip.subclip(start_time, end_time)
+            
+            if mute_audio:
+                clipped_clip = clipped_clip.set_audio(None)
+            
+            clipped_clip = clipped_clip.fx(vfx.speedx, speed_factor)
+            
             if use_aspect_ratio:
-                original_width, original_height = clip.size
-                
+                original_width, original_height = video_clip.size
+
                 if use_aspect_ratio == '21:9':
                     target_ratio = 21 / 9
                 else:
                     target_ratio = 16 / 9 if use_aspect_ratio == '16:9' else 9 / 16
-                
+
                 target_width = int(original_height * target_ratio)
                 target_height = original_height
-            
+
                 if use_aspect_ratio == '21:9':
                     target_height = int(original_width / target_ratio)
                     margin = (original_height - target_height) // 2
@@ -72,12 +64,12 @@ def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mu
                 else:
                     margin = (original_width - target_width) // 2
                     clipped_clip = clipped_clip.crop(x1=margin, x2=original_width - margin)
-        
-        with st.spinner(f'Procesando video {i+1} de {num_cuts}...'):
-            clipped_clip.write_videofile(output_file, codec="libx264", audio_codec="aac", verbose=False)
+
+            with st.spinner(f'Procesando video {i+1} de {num_cuts}...'):
+                clipped_clip.write_videofile(temp_filename, codec="libx264", audio_codec="aac", verbose=False)
             
-        output_files.append(output_file)
-        prev_end_time = end_time
+            output_files.append(temp_filename)
+            prev_end_time = end_time
 
     return output_files, cuts
 
@@ -94,49 +86,42 @@ def download_all_button(cut_files):
     href = f'<a href="data:application/zip;base64,{b64}" download="{zip_filename}">Descargar Todos</a>'
     st.markdown(href, unsafe_allow_html=True)
 
-opciones = ['YouTube']
+st.title("Descargador de Videos")
+video_link = st.text_input("Enlace del Video", value='')
+video_input = st.file_uploader("Cargar Video", type=["mp4", "avi", "mov"])
 
-with st.sidebar:
-    seleccionado = option_menu("Descargador", opciones, icons=['play'], menu_icon="cast", default_index=0)
+aspect_ratio_options = ['16:9', '9:16', '21:9']
+aspect_ratio_choice = st.selectbox('Seleccionar Proporción:', aspect_ratio_options, index=2)
+resolution_options = [144, 240, 360, 720]
+resolution_choice = st.selectbox('Seleccionar Resolución:', resolution_options, index=2)
 
-funciones = [youtube]
+choice = st.radio('Seleccionar Proceso:', ['Cortar Video', 'Descargar Completo'], 1)
 
-if seleccionado:
-    indice = opciones.index(seleccionado)
-    funcion = funciones[indice]
-    st.title(f"{seleccionado} Descargador de Videos")
-    st.write(f"Descarga y corta una parte aleatoria de {seleccionado}.")  
+if st.button("Procesar"):
+    if video_input is not None:
+        video_clip = VideoFileClip(video_input)
+        video_clip = video_clip.subclip(0, video_clip.duration)  # Ajusta la duración si es necesario
 
-    video_link = st.text_input("Enlace del Video", value='https://www.youtube.com/watch?v=urhsbKIF9pE')
-    aspect_ratio_options = ['16:9', '9:16', '21:9']
-    aspect_ratio_choice = st.selectbox('Seleccionar Proporción:', aspect_ratio_options, index=2)
-    resolution_options = [144, 240, 360, 720]
-    resolution_choice = st.selectbox('Seleccionar Resolución:', resolution_options, index=2)
+        if choice == 'Cortar Video':
+            num_cuts = st.slider('Seleccionar número de cortes', 1, 10, 1)
+            duration_minutes_slider = st.slider('Seleccionar duración de los cortes (minutos)', 1, 180, 1)
+            duration_seconds = duration_minutes_slider * 60
+            speed_factor = st.slider('Seleccionar velocidad del video', -2.0, 3.0, 1.0)
+            mute_audio = st.checkbox('Eliminar sonido de los videos cortados')
+
+            cuts = random_cuts(video_clip, "video_cortado", num_cuts=num_cuts, duration=duration_seconds, speed_factor=speed_factor, mute_audio=mute_audio, use_aspect_ratio=aspect_ratio_choice)
+
+            output_files, _ = cuts
+            download_all_button(output_files)
 
-    choice = st.radio('Seleccionar Proceso:', ['Cortar Video', 'Descargar Completo'], 1)
-    
-    if choice == 'Cortar Video':
-        num_cuts = st.slider('Seleccionar número de cortes', 1, 10, 1)
-        duration_minutes_slider = st.slider('Seleccionar duración de los cortes (minutos)', 1, 180, 1)
-        duration_seconds = duration_minutes_slider * 60
-        speed_factor = st.slider('Seleccionar velocidad del video', -2.0, 3.0, 1.0)
-        mute_audio = st.checkbox('Eliminar sonido de los videos cortados')
-        
-        if st.button(f"Descargar y Cortar {seleccionado}"):
-            video_file, titulo_video, video_info, thumbnail_file = funcion(video_link, resolusi_input=resolution_choice)
-            cut_files, cuts = random_cuts(video_file, titulo_video, num_cuts=num_cuts, duration=duration_seconds, speed_factor=speed_factor, mute_audio=mute_audio, use_aspect_ratio=aspect_ratio_choice)
-            
-            for i, cut_file in enumerate(cut_files):
-                file_size = os.path.getsize(cut_file)
-                info_list = [f"{titulo_video}_cut_{i+1}", cut_file, thumbnail_file, cuts[i][0], cuts[i][1]]
-                session(info_list)
-        
-            download_all_button(cut_files)
-            
     else:  
         if st.button(f"Descargar {seleccionado}"):
             video_file, titulo_video, video_info, thumbnail_file = funcion(video_link, resolusi_input=resolution_choice)
             file_size = os.path.getsize(video_file)
             info_list = [titulo_video, video_file, thumbnail_file]
             download_all_button([video_file])
-            session(info_list)
+            session(info_list)
+        for temp_filename in output_files:
+            os.remove(temp_filename)
+
+        st.success("Proceso completado y archivos temporales eliminados.")