Update app.py

app.py

@@ -426,10 +426,35 @@ if st.button("Gerar Vídeo(s)"):
                                 st.download_button(f"📥 Baixar vídeo {n+1}", f, file_name=final_name)
 
                 progresso.progress(100)
-                st.success("✅ Todos os vídeos foram gerados com sucesso!")
-
-            except subprocess.CalledProcessError as e:
-                st.error(f"❌ Erro ao gerar vídeo:\n\n{e.stderr.decode(errors='ignore')}")
-
-            finally:
-                shutil.rmtree(temp_dir)
+                # Lista para armazenar os vídeos gerados
+videos_gerados = []
+
+# Dentro do loop de geração de cada vídeo (depois de gerar cada final_name ou antiflop_out):
+# Exemplo:
+videos_gerados.append(antiflop_out)  # ou final_name se não usar antiflop
+
+# Depois de todos os vídeos gerados:
+if videos_gerados:
+    st.markdown("### 📥 Baixar Todos os Vídeos")
+
+    # Monta os links
+    download_links = ""
+    for v in videos_gerados:
+        download_links += f"<a href='{v}' download='{v}'></a>\n"
+
+    # Botão que dispara o JS para baixar tudo
+    st.markdown(
+        f"""
+        <button onclick="baixarTodos()">📥 Baixar Todos</button>
+        <script>
+        function baixarTodos() {{
+            const links = Array.from(document.querySelectorAll('a[download]'));
+            for (let l of links) {{
+                l.click();
+            }}
+        }}
+        </script>
+        {download_links}
+        """,
+        unsafe_allow_html=True
+    )