Update app.py

app.py

@@ -1,22 +1,16 @@
 import os
+from io import BytesIO
+import zipfile
+import random
+import base64
 import streamlit as st
 from streamlit_option_menu import option_menu
 from youtube import youtube, download_youtube
 from others import *
-import random
-import base64
-import zipfile
-from io import BytesIO
 from moviepy.video.io.VideoFileClip import VideoFileClip
-from moviepy.editor import vfx 
-
-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")
+from moviepy.editor import vfx
 
-def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mute_audio=False, aspect_ratio='16:9'):
+def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mute_audio=False, use_9_16=False):
     clip = VideoFileClip(video_file)
     
     total_duration = clip.duration
@@ -28,10 +22,11 @@ def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mu
         cuts.append((start_time, end_time))
 
     output_files = []
-    prev_end_time = 0  # Para evitar repeticiones de tiempo de corte
+    prev_end_time = 0
+    
     for i, (start_time, end_time) in enumerate(cuts):
         if start_time == prev_end_time:
-            continue  # Evitar repeticiones de tiempo de corte
+            continue
 
         output_file = f"{title}_cut_{i+1}.mp4"
         clipped_clip = clip.subclip(start_time, end_time)
@@ -41,16 +36,16 @@ def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mu
         
         clipped_clip = clipped_clip.fx(vfx.speedx, speed_factor)
         
-        if aspect_ratio == '9:16':
-            
+        if use_9_16:
             original_width, original_height = clip.size
-            new_height = int(original_width * (9 / 16))
+            new_width = int(original_height * (9 / 16))
             
-            top_margin = (original_height - new_height) // 2
-            bottom_margin = original_height - top_margin
-            clipped_clip = clipped_clip.crop(y1=top_margin, y2=bottom_margin)
+            left_margin = (original_width - new_width) // 2
+            right_margin = original_width - left_margin
+            clipped_clip = clipped_clip.crop(x1=left_margin, x2=right_margin)
             
-            clipped_clip = clipped_clip.resize(width=new_height, height=original_width)
+            # Cambio en las dimensiones para la relación de aspecto 9:16
+            clipped_clip = clipped_clip.resize(width=new_width, height=original_height)
         
         with st.spinner(f'Procesando video {i+1} de {num_cuts}...'):
             clipped_clip.write_videofile(output_file, codec="libx264", audio_codec="aac", verbose=False)
@@ -60,15 +55,9 @@ def random_cuts(video_file, title, num_cuts=1, duration=60, speed_factor=1.0, mu
 
     return output_files, cuts
 
-opciones = ['YouTube']
-
-with st.sidebar:
-    seleccionado = option_menu("Descargador", opciones, icons=['play'], menu_icon="cast", default_index=0)
-
-funciones = [youtube]
-
 def download_all_button(cut_files):
     zip_buffer = BytesIO()
+    
     with zipfile.ZipFile(zip_buffer, 'w') as zip_file:
         for cut_file in cut_files:
             zip_file.write(cut_file, os.path.basename(cut_file))
@@ -79,6 +68,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]
@@ -93,13 +89,12 @@ if seleccionado:
         duration_slider = st.slider('Seleccionar duración de los cortes', 10, 900, 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')
-        aspect_ratio = st.radio('Seleccionar Formato de Aspecto:', ['16:9', '9:16'], 1)
+        use_9_16 = st.checkbox('Proporción 9:16')
 
         if st.button(f"Descargar y Cortar {seleccionado}"):
             video_file, titulo_video, video_info, thumbnail_file = funcion(video_link)
             
-            cut_height = 1280 if aspect_ratio == '9:16' else None
-            cut_files, cuts = random_cuts(video_file, titulo_video, num_cuts=num_cuts, duration=duration_slider, speed_factor=speed_factor, mute_audio=mute_audio, aspect_ratio=aspect_ratio)
+            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)
@@ -114,4 +109,4 @@ if seleccionado:
             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)