Update app.py

app.py

@@ -1,62 +1,27 @@
-import os
-import zipfile
-import random
-import base64
-from others import *
-import streamlit as st
-from io import BytesIO
-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])
-    st.text(f"Video Information: {info_list[0]}")
-    if len(info_list) == 5:
-        st.text(f"Duration: {info_list[3]} - {info_list[4]} seconds")
-
-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):
-        start_time = random.uniform(0, max(total_duration - duration, 0))
-        end_time = min(start_time + duration, total_duration)
-        cuts.append((start_time, end_time))
-
-    output_files = []
-    prev_end_time = 0
-    
-    for i, (start_time, end_time) in enumerate(cuts):
-        if start_time == prev_end_time:
-            continue
-
-        output_file = f"{title}_cut_{i+1}.mp4"
-        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
+        
+            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
+        
             if use_aspect_ratio:
-                original_width, original_height = video_clip.size
-
+                original_width, original_height = 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
@@ -64,12 +29,12 @@ def random_cuts(video_clip, 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(temp_filename, codec="libx264", audio_codec="aac", verbose=False)
+        
+        with st.spinner(f'Procesando video {i+1} de {num_cuts}...'):
+            clipped_clip.write_videofile(output_file, codec="libx264", audio_codec="aac", verbose=False)
             
-            output_files.append(temp_filename)
-            prev_end_time = end_time
+        output_files.append(output_file)
+        prev_end_time = end_time
 
     return output_files, cuts
 
@@ -86,42 +51,49 @@ 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)
 
-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"])
-
-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)
-
-choice = st.radio('Seleccionar Proceso:', ['Cortar Video', 'Descargar Completo'], 1)
+opciones = ['YouTube']
 
-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
+with st.sidebar:
+    seleccionado = option_menu("Descargador", opciones, icons=['play'], menu_icon="cast", default_index=0)
 
-        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')
+funciones = [youtube]
 
-            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)
+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}.")  
 
-            output_files, _ = cuts
-            download_all_button(output_files)
+    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)
 
+    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)
-        for temp_filename in output_files:
-            os.remove(temp_filename)
-
-        st.success("Proceso completado y archivos temporales eliminados.")
+            session(info_list)