Update streamlit_app.py

streamlit_app.py

@@ -102,6 +102,8 @@ def process_video(uploaded_video, bg_image, bg_color, bg_type):
     run_id = uuid.uuid4().hex[:8]
     logger.info("=" * 80)
     logger.info(f"[RUN {run_id}] VIDEO PROCESSING STARTED at {datetime.utcnow().isoformat()}Z")
+    logger.info(f"[RUN {run_id}] Video size={len(uploaded_video.read()) / 1e6:.2f}MB, BG type={bg_type}")
+    uploaded_video.seek(0)  # Reset for later read
     logger.info("=" * 80)
     st.session_state.processing = True
     st.session_state.processed_video_bytes = None
@@ -114,15 +116,18 @@ def process_video(uploaded_video, bg_image, bg_color, bg_type):
             uploaded_video.seek(0)
             tmp_vid.write(uploaded_video.read())
             tmp_vid_path = tmp_vid.name
+        logger.info(f"[RUN {run_id}] Temporary video path: {tmp_vid_path}")
         # Stage 1: Create transparent video and extract audio
         transparent_path, audio_path = stage1_create_transparent_video(tmp_vid_path)
         if not transparent_path or not os.path.exists(transparent_path):
             raise RuntimeError("Stage 1 failed: Transparent video not created")
+        logger.info(f"[RUN {run_id}] Stage 1 completed: Transparent path={transparent_path}, Audio path={audio_path}")
         # Stage 2: Composite with background and restore audio
         background = bg_image.convert("RGB") if bg_type == "Image" else bg_color
         final_path = stage2_composite_background(transparent_path, audio_path, background, bg_type.lower())
         if not final_path or not os.path.exists(final_path):
             raise RuntimeError("Stage 2 failed: Final video not created")
+        logger.info(f"[RUN {run_id}] Stage 2 completed: Final path={final_path}")
         # Load final video into memory for download
         with open(final_path, 'rb') as f:
             st.session_state.processed_video_bytes = f.read()
@@ -141,14 +146,18 @@ def process_video(uploaded_video, bg_image, bg_color, bg_type):
         logger.info(f"[RUN {run_id}] Processing finished")
 # --- Main App Entry Point ---
 def main():
-    st.set_page_config(
-        page_title=APP_NAME,
-        page_icon="🎥",
-        layout="wide",
-        initial_sidebar_state="expanded"
-    )
-    initialize_session_state()
-    render_ui(process_video)
+    try:
+        st.set_page_config(
+            page_title=APP_NAME,
+            page_icon="🎥",
+            layout="wide",
+            initial_sidebar_state="expanded"
+        )
+        initialize_session_state()
+        render_ui(process_video)
+    except Exception as e:
+        logger.error(f"Main app error: {e}", exc_info=True)
+        st.error(f"App startup failed: {str(e)}. Check logs for details.")
 if __name__ == "__main__":
     setup_t4_environment()
     main()