Update app.py

app.py

@@ -132,41 +132,37 @@ if audio_file:
         progress_bar.empty()
         logger.info("All images generated successfully.")
 
-    # Generate video when all images are generated
-    if st.session_state[f'generated_images_{session_id}'] and st.session_state[f'audio_{session_id}'] and not st.session_state[f'video_generated_{session_id}']:
-        with st.spinner("Generating video... Please wait."):
-            # Create a temporary directory for the video
-            temp_dir = tempfile.gettempdir()
-            video_filename = f"generated_video_{session_id}.mp4"
-            video_path = os.path.join(temp_dir, video_filename)
-
-            # Map images to segments
-            image_paths = [img[1] for img in st.session_state[f'generated_images_{session_id}']]
-            generated_video_path = generate_video(
-                audio_file=st.session_state[f'audio_{session_id}'], 
-                images=image_paths, 
-                segments=st.session_state[f'segments_{session_id}']
-            )
-            st.session_state[f'generated_video_{session_id}'] = generated_video_path
-            st.session_state[f'video_generated_{session_id}'] = True  # Set the flag to True
-            st.success("Video generated successfully!")
-            logger.info("Video generated successfully.")
-
-    # Display the generated video
-    if st.session_state[f'generated_video_{session_id}']:
-        st.video(st.session_state[f'generated_video_{session_id}'])
-        
-        # Add a download button for the generated video
-        with open(st.session_state[f'generated_video_{session_id}'], "rb") as file:
-            st.download_button(
-                label="Download Video",
-                data=file,
-                file_name=f"generated_video_{session_id}.mp4",
-                mime="video/mp4"
-            )
-
+   # Generate video when all images are generated
+if st.session_state[f'generated_images_{session_id}'] and st.session_state[f'audio_{session_id}'] and not st.session_state[f'video_generated_{session_id}']:
+    with st.spinner("Generating video... Please wait."):
+        # Create a directory for the video
+        video_dir = "./generated_videos"
+        os.makedirs(video_dir, exist_ok=True)
+        video_filename = f"generated_video_{session_id}.mp4"
+        video_path = os.path.join(video_dir, video_filename)
+        # Map images to segments
+        image_paths = [img[1] for img in st.session_state[f'generated_images_{session_id}']]
+        generated_video_path = generate_video(
+            audio_file=st.session_state[f'audio_{session_id}'], 
+            images=image_paths, 
+            segments=st.session_state[f'segments_{session_id}']
+        )
+        st.session_state[f'generated_video_{session_id}'] = generated_video_path
+        st.session_state[f'video_generated_{session_id}'] = True  # Set the flag to True
+        st.success("Video generated successfully!")
+        logger.info("Video generated successfully.")
+# Display the generated video
+if st.session_state[f'generated_video_{session_id}']:
+    st.video(st.session_state[f'generated_video_{session_id}'])
+    
+    # Add a download button for the generated video
+    with open(st.session_state[f'generated_video_{session_id}'], "rb") as file:
+        st.download_button(
+            label="Download Video",
+            data=file,
+            file_name=f"generated_video_{session_id}.mp4",
+            mime="video/mp4"
+        )
 else:
     st.warning("Please upload an audio file to proceed.")
-    logger.warning("No audio file uploaded.")
-
-
+    logger.warning("No audio file uploaded.")