Update app.py

app.py

@@ -135,11 +135,20 @@ if audio_file:
    # 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
+        # Define the directory for the video
         video_dir = "./generated_videos"
-        os.makedirs(video_dir, exist_ok=True)
+        
+        # Ensure the directory exists
+        if not os.path.exists(video_dir):
+            os.makedirs(video_dir, exist_ok=True)
+            logger.info(f"Created directory: {video_dir}")
+        else:
+            logger.info(f"Directory already exists: {video_dir}")
+
+        # Define the video file path
         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(
@@ -147,22 +156,35 @@ if st.session_state[f'generated_images_{session_id}'] and st.session_state[f'aud
             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.")
+
+        # Check if the video was generated successfully
+        if generated_video_path and os.path.exists(generated_video_path):
+            st.session_state[f'generated_video_{session_id}'] = generated_video_path
+            st.session_state[f'video_generated_{session_id}'] = True
+            st.success("Video generated successfully!")
+            logger.info(f"Video saved at: {generated_video_path}")
+        else:
+            st.error("Error generating video.")
+            logger.error("Error generating video.")
+
 # 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"
-        )
+    video_path = st.session_state[f'generated_video_{session_id}']
+
+    # Check if the video file exists
+    if os.path.exists(video_path):
+        st.video(video_path)  # Display the video
+
+        # Add a download button for the generated video
+        with open(video_path, "rb") as file:
+            st.download_button(
+                label="Download Video",
+                data=file.read(),
+                file_name=f"generated_video_{session_id}.mp4",
+                mime="video/mp4"
+            )
+    else:
+        st.error("The generated video file is missing or inaccessible.")
 else:
     st.warning("Please upload an audio file to proceed.")
     logger.warning("No audio file uploaded.")