Update app.py

app.py

@@ -28,7 +28,6 @@ if "tutorial_path" not in st.session_state:
     st.session_state.tutorial_path = None
 if "video_final" not in st.session_state:
     st.session_state.video_final = None
-
 # ----- BOTÃO PUXAR VÍDEO -----
 if url and st.button("🔄 Puxar vídeo do TikTok"):
     with st.spinner("Baixando vídeo do TikTok..."):
@@ -55,7 +54,6 @@ if video and not st.session_state.video_path:
         st.session_state.video_path = tmp.name
 
 video_path = st.session_state.video_path
-
 # ----- PLAYER DO VÍDEO ANTES DOS CORTES -----
 if video_path and not st.session_state.processando:
     st.subheader("🎥 Pré-visualização do vídeo original")
@@ -65,6 +63,8 @@ if video_path and not st.session_state.processando:
     excluir_inicio = st.number_input("Excluir a partir do segundo", min_value=0, value=8, key="excluir_inicio")
     excluir_fim = st.number_input("Excluir até o segundo", min_value=0, value=12, key="excluir_fim")
 
+    st.session_state.embaralhar_cortes = st.checkbox("✂️ Ativar cortes e embaralhamento aleatório", value=True, key="embaralhar_cortes")
+
     corte_min = st.number_input("Tamanho mínimo do corte (segundos)", min_value=1, value=2, key="corte_min")
     corte_max = st.number_input("Tamanho máximo do corte (segundos)", min_value=3, value=5, key="corte_max")
 
@@ -121,7 +121,6 @@ if st.session_state.processando:
     )
 
     status = st.empty()
-
     status.markdown("🔎 Analisando duração e resolução do vídeo...")
     barra.progress(5)
 
@@ -148,7 +147,6 @@ if st.session_state.processando:
     excluir_fim = st.session_state.excluir_fim
     corte_min = st.session_state.corte_min
     corte_max = st.session_state.corte_max
-
     velocidade_cortes = st.session_state.velocidade_cortes
 
     partes = []
@@ -159,29 +157,33 @@ if st.session_state.processando:
 
     cortes = []
     usados = set()
-
     barra.progress(15)
     status.markdown("✂️ Realizando cortes...")
 
-    for inicio, fim in partes:
-        pos = inicio
-        while pos < fim:
-            tam_possivel = min(corte_max, fim - pos)
-            if tam_possivel < corte_min:
-                break
-            tamanho = random.randint(corte_min, tam_possivel)
-            fim_corte = pos + tamanho
-
-            if (pos, fim_corte) not in usados:
-                cortes.append((int(pos), int(fim_corte)))
-                usados.add((pos, fim_corte))
-            pos = fim_corte
-
-    random.shuffle(cortes)
-
+    if st.session_state.embaralhar_cortes:
+        for inicio, fim in partes:
+            pos = inicio
+            while pos < fim:
+                tam_possivel = min(corte_max, fim - pos)
+                if tam_possivel < corte_min:
+                    break
+                tamanho = random.randint(corte_min, tam_possivel)
+                fim_corte = pos + tamanho
+
+                if (pos, fim_corte) not in usados:
+                    cortes.append((int(pos), int(fim_corte)))
+                    usados.add((pos, fim_corte))
+                pos = fim_corte
+
+        random.shuffle(cortes)
+    else:
+        # Apenas um corte simples direto
+        start = excluir_inicio
+        end = excluir_fim if excluir_fim > excluir_inicio else dur
+        cortes = [(start, end)]
     barra.progress(25)
 
-    # --- PROCESSAR OS CORTES SEM FILTRO ---
+    # --- PROCESSAR OS CORTES ---
     with tempfile.TemporaryDirectory() as tmpdir:
         arquivos_cortes = []
         total_cortes = len(cortes)
@@ -226,7 +228,6 @@ if st.session_state.processando:
         barra.progress(70)
         status.markdown("🔍 Aplicando zoom e espelhamento no vídeo unido...")
 
-        # --- APLICAR ZOOM E ESPELHAMENTO NO VÍDEO UNIDO ---
         video_processado = os.path.join(tmpdir, "video_processado.mp4")
         zoom_factor = st.session_state.zoom / 100
         filtro_zoom = f"scale=iw*{zoom_factor}:ih*{zoom_factor},crop={largura}:{altura}"
@@ -253,13 +254,12 @@ if st.session_state.processando:
             resultado = subprocess.run(duracao_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
             duracao_video = float(resultado.stdout.decode().strip())
 
-            # Escolhe posição aleatória depois de 3 segundos e antes do final - 1s
+            # Escolhe posição aleatória entre 3s e final - 1s
             pos_tutorial = random.uniform(3, duracao_video - 1)
 
             parte1 = os.path.join(tmpdir, "parte1.mp4")
             parte2 = os.path.join(tmpdir, "parte2.mp4")
 
-            # Corta a parte 1
             subprocess.call([
                 "ffmpeg", "-y", "-i", video_processado,
                 "-t", str(pos_tutorial),
@@ -267,7 +267,6 @@ if st.session_state.processando:
                 parte1
             ])
 
-            # Corta a parte 2
             subprocess.call([
                 "ffmpeg", "-y", "-i", video_processado,
                 "-ss", str(pos_tutorial),
@@ -275,7 +274,6 @@ if st.session_state.processando:
                 parte2
             ])
 
-            # Prepara tutorial padronizado
             tutorial_pad = os.path.join(tmpdir, "tutorial_pad.mp4")
             subprocess.call([
                 "ffmpeg", "-y", "-i", st.session_state.tutorial_path,
@@ -285,7 +283,6 @@ if st.session_state.processando:
                 tutorial_pad
             ])
 
-            # Criar lista para concatenação
             lista_final_txt = os.path.join(tmpdir, "lista_final.txt")
             with open(lista_final_txt, "w") as f:
                 f.write(f"file '{parte1}'\n")
@@ -298,7 +295,6 @@ if st.session_state.processando:
                 "-c:v", "libx264", "-preset", "veryfast",
                 video_com_tutorial
             ])
-
         else:
             video_com_tutorial = video_processado
 
@@ -326,11 +322,9 @@ if st.session_state.processando:
             ])
         else:
             os.rename(video_final_temp, video_final)
-
         barra.progress(95)
         status.markdown("📋 Adicionando metadata...")
 
-        # METADATA ALEATÓRIA
         marcas = {
             "samsung": ["SM-S918B", "SM-A546E", "SM-M536B"],
             "xiaomi": ["Redmi Note 12", "Poco X5", "Mi 11 Lite"],
@@ -368,17 +362,14 @@ if st.session_state.processando:
 
         barra.progress(100)
 
-        # Copiar vídeo final para fora do tmpdir
         video_final_permanente = os.path.join(tempfile.gettempdir(), "video_final_final.mp4")
         shutil.copy(video_final_meta, video_final_permanente)
         st.session_state.video_final = video_final_permanente
 # --- PLAYER DOS VÍDEOS E BOTÕES ---
-
 if st.session_state.video_final:
 
     st.success("✅ Vídeo criado com sucesso!")
 
-    # Loader "Aguarde os players"
     load_players = st.empty()
     load_players.markdown(
         """
@@ -424,17 +415,14 @@ if st.session_state.video_final:
 
     load_players.empty()
 
-    # Botão para download (sem reset automático)
     with open(st.session_state.video_final, "rb") as f:
         st.download_button("⬇️ Baixar vídeo", f, file_name="video_novo.mp4")
 
-    # Botão separado para reset
     if st.button("🔄 Criar novo"):
         for key in list(st.session_state.keys()):
             del st.session_state[key]
         st.experimental_rerun()
 
-    # Botão para manter as mesmas configurações
     if st.button("🔄 Gerar novamente com os mesmos ajustes"):
         st.session_state.processando = True
         st.session_state.video_final = None