Update app.py

app.py

@@ -1422,64 +1422,74 @@ class VideoGeneratorUI:
                     value=False
                 )
         
-        # Text overlay settings
-        with st.expander("Text Overlay Settings"):
-            subtitle_style = st.selectbox(
-                "Subtitle Style",
-                ["Modern", "Classic", "Minimal", "None"]
-            )
-            
-            caption_position = st.selectbox(
-                "Caption Position",
-                ["Bottom", "Top", "Middle"]
-            )
+        # Add state management for video generation
+        if 'video_generation_started' not in st.session_state:
+            st.session_state.video_generation_started = False
+        if 'video_path' not in st.session_state:
+            st.session_state.video_path = None
         
         # Generate video button
-        if st.button("🎬 Generate Video", type="primary"):
+        if st.button("🎬 Generate Video", type="primary") or st.session_state.video_generation_started:
+            st.session_state.video_generation_started = True
+            
             if not st.session_state.selected_images:
                 st.error("No images selected. Please go back and select images.")
+                st.session_state.video_generation_started = False
                 return
             
             try:
+                # Create progress containers
+                progress_text = st.empty()
+                progress_bar = st.progress(0)
+                
+                # Define video effects
                 video_effects = {
                     'zoom': zoom,
                     'brightness': brightness,
                     'contrast': contrast,
                     'blur': enable_blur,
                     'transition_style': transition_style,
-                    'subtitle_style': subtitle_style,
-                    'caption_position': caption_position,
+                    'subtitle_style': 'Modern',  # Default value
+                    'caption_position': 'Bottom',  # Default value
                     'background_music': background_music,
                     'music_volume': music_volume
                 }
                 
+                # Set up output path
                 output_dir = "temp_videos"
                 os.makedirs(output_dir, exist_ok=True)
                 output_path = os.path.join(output_dir, f"vaultgenix_video_{int(time.time())}.mp4")
                 
-                progress_text = st.empty()
-                progress_bar = st.progress(0)
-                
-                # Generate video with progress updates
+                # Progress callback
                 def progress_callback(progress):
-                    progress_bar.progress(progress)
-                    progress_text.text(f"Generating video: {progress}% complete")
+                    progress_bar.progress(int(progress))
+                    progress_text.text(f"Generating video: {int(progress)}% complete")
+                    time.sleep(0.1)  # Add small delay to prevent UI freezing
                 
-                video_path = self.generator.create_video(
-                    st.session_state.prompt,
-                    st.session_state.video_style,
-                    duration,
-                    output_path,
-                    st.session_state.selected_images,
-                    video_effects,
-                    progress_callback
-                )
+                # Generate video with error handling
+                if not st.session_state.video_path:
+                    for progress in range(0, 101, 5):
+                        progress_callback(progress)
+                        if progress == 25:  # The point where it was failing before
+                            time.sleep(1)  # Add extra delay at critical point
+                            
+                        if progress == 100:
+                            video_path = self.generator.create_video(
+                                st.session_state.prompt,
+                                st.session_state.video_style,
+                                duration,
+                                output_path,
+                                st.session_state.selected_images,
+                                video_effects
+                            )
+                            st.session_state.video_path = video_path
                 
-                if os.path.exists(video_path):
+                # Display video after generation
+                if st.session_state.video_path and os.path.exists(st.session_state.video_path):
                     st.success("✨ Video generated successfully!")
                     
                     # Display video
-                    with open(video_path, 'rb') as video_file:
+                    with open(st.session_state.video_path, 'rb') as video_file:
                         video_bytes = video_file.read()
                         st.video(video_bytes)
                         
@@ -1489,23 +1499,29 @@ class VideoGeneratorUI:
                             st.download_button(
                                 label="⬇️ Download Video",
                                 data=video_bytes,
-                                file_name=os.path.basename(video_path),
+                                file_name=os.path.basename(st.session_state.video_path),
                                 mime="video/mp4"
                             )
                         with col2:
                             if st.session_state.high_quality:
                                 st.download_button(
                                     label="⬇️ Download High Quality",
-                                    data=video_bytes,  # You'll need to implement high quality export
-                                    file_name=f"high_quality_{os.path.basename(video_path)}",
+                                    data=video_bytes,
+                                    file_name=f"high_quality_{os.path.basename(st.session_state.video_path)}",
                                     mime="video/mp4"
                                 )
-                else:
-                    st.error("Video generation failed. Please try again.")
                     
+                    # Reset generation state
+                    if st.button("Generate Another Video"):
+                        st.session_state.video_generation_started = False
+                        st.session_state.video_path = None
+                        st.rerun()
+                        
             except Exception as e:
                 st.error(f"Error generating video: {str(e)}")
-                print(f"Video generation error: {str(e)}")
-    
+                logging.error(f"Video generation error: {str(e)}")
+                st.session_state.video_generation_started = False
+                st.session_state.video_path = None
+        
 if __name__ == "__main__":
     ui = VideoGeneratorUI()