Update app.py

app.py

@@ -2,8 +2,9 @@ import os
 import zipfile
 import random
 import base64
-from io import BytesIO
+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
@@ -22,8 +23,8 @@ def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mu
     cuts = []
 
     for _ in range(num_cuts):
-        start_time = random.uniform(0, max(total_duration - duration * 60, 0))
-        end_time = min(start_time + duration * 60, total_duration)
+        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 = []
@@ -73,6 +74,13 @@ 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']
+
+with st.sidebar:
+    seleccionado = option_menu("Descargador", opciones, icons=['play'], menu_icon="cast", default_index=0)
+
+funciones = [youtube]
+
 if seleccionado:
     indice = opciones.index(seleccionado)
     funcion = funciones[indice]
@@ -81,7 +89,7 @@ if seleccionado:
     
     video_link = st.text_input("Enlace del Video", value='https://www.youtube.com/watch?v=or_wwcB9pGQ&t=226s')
     choice = st.radio('Seleccionar Proceso:', ['Cortar Video', 'Descargar Completo'], 0)
-
+    
     if choice == 'Cortar Video':
         num_cuts = st.slider('Seleccionar número de cortes', 1, 10, 2)
         duration_slider = st.slider('Seleccionar duración de los cortes', 10, 900, 60)
@@ -90,47 +98,21 @@ if seleccionado:
         use_9_16 = st.checkbox('Proporción 9:16')
 
         if st.button(f"Descargar y Cortar {seleccionado}"):
-            video_file, titulo_video, _, thumbnail_file = funcion(video_link)
-
-            st.video(video_file)  # Mostrar la previsualización del video
+            video_file, titulo_video, video_info, thumbnail_file = funcion(video_link)
             
             cut_files, cuts = random_cuts(video_file, titulo_video, num_cuts=num_cuts, duration=duration_slider, speed_factor=speed_factor, mute_audio=mute_audio, use_9_16=use_9_16)
             
             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("Descargar y Cortar"):
-            video_link = st.text_input("Enlace del Video", value='https://www.youtube.com/watch?v=or_wwcB9pGQ&t=226s')
-        
-            video_file, titulo_video, _, thumbnail_file = youtube(video_link)
-            st.video(video_file)  # Mostrar la previsualización del video
-            
-            num_cuts = st.slider('Seleccionar número de cortes', 1, 10, 2)
-            duration_slider = st.slider('Seleccionar duración de los cortes (minutos)', 1, 360, 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')
-            use_9_16 = st.checkbox('Proporción 9:16')
-        
-            cut_files, cuts = random_cuts(video_file, titulo_video, num_cuts=num_cuts, duration=duration_slider, speed_factor=speed_factor, mute_audio=mute_audio, use_9_16=use_9_16)
-            
-            for i, cut_file in enumerate(cut_files):
-                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("Descargar Completo"):
-        video_link = st.text_input("Enlace del Video", value='https://www.youtube.com/watch?v=or_wwcB9pGQ&t=226s')
-    
-        video_file, titulo_video, _, thumbnail_file = youtube(video_link)
-        st.video(video_file)  # Mostrar la previsualización del video
-        
-        file_size = os.path.getsize(video_file)
-        info_list = [titulo_video, video_file, thumbnail_file]
-        download_all_button([video_file])
-        session(info_list)
+        if st.button(f"Descargar {seleccionado}"):
+            video_file, titulo_video, video_info, thumbnail_file = funcion(video_link)
+            file_size = os.path.getsize(video_file)
+            info_list = [titulo_video, video_file, thumbnail_file]
+            download_all_button([video_file])
+            session(info_list)