pycaps / src /ui /step5_view.py
Franco Zanardi
feature: allow edit templates
6774799
import streamlit as st
from utils import go_to_step, reset_all, release_lock_slot_if_needed, acquire_lock_slot, display_video
def render_step5():
st.header("Your Video is Ready!")
release_lock_slot_if_needed()
if 'final_video_path' in st.session_state and st.session_state.final_video_path:
display_video(st.session_state.final_video_path)
else:
st.error("Could not find the final video.")
col1, col2 = st.columns(2)
with col1:
if st.button("⬅️ Choose Another Style", use_container_width=True):
lock_file = acquire_lock_slot()
if not lock_file:
st.warning("🚧 All our processing slots are currently busy. Please check back in a few minutes.")
else:
st.session_state.lock_file_path = lock_file
keys_to_delete = ['processed_doc', 'final_video_path', 'edit_requested']
for key in keys_to_delete:
if key in st.session_state:
del st.session_state[key]
go_to_step(2)
st.rerun()
with col2:
if st.button("🏠 Start with a New Video", use_container_width=True):
reset_all()
st.rerun()