Update streamlit_app.py

streamlit_app.py

@@ -21,59 +21,46 @@
 import numpy as np
 from PIL import Image
 import torch
-
 # Import UI components
 from ui import render_ui
-
 # Import pipeline functions
-from pipeline.video_pipeline import (..., ...)  # Remove dump_env
+from pipeline.video_pipeline import (
     stage1_create_transparent_video,
     stage2_composite_background,
     setup_t4_environment,
-    check_gpu,
-    dump_env
+    check_gpu
 )
-
 # --- Constants ---
 APP_NAME = "Advanced Video Background Replacer"
 LOG_FILE = "/tmp/app.log"
 LOG_MAX_BYTES = 5 * 1024 * 1024
 LOG_BACKUPS = 5
-
 # --- Logging Setup ---
 def setup_logging(level: int = logging.INFO) -> logging.Logger:
     logger = logging.getLogger(APP_NAME)
     logger.setLevel(level)
     logger.propagate = False
-
     # Clear previous handlers on rerun
     for h in list(logger.handlers):
         logger.removeHandler(h)
-
     # Console handler
     ch = logging.StreamHandler(sys.stdout)
     ch.setLevel(level)
     ch.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
-
     # Rotating file handler
     fh = logging.handlers.RotatingFileHandler(
         LOG_FILE, maxBytes=LOG_MAX_BYTES, backupCount=LOG_BACKUPS, encoding="utf-8"
     )
     fh.setLevel(level)
     fh.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
-
     logger.addHandler(ch)
     logger.addHandler(fh)
     return logger
-
 logger = setup_logging()
-
 # --- Global Exception Hook ---
 def custom_excepthook(type, value, tb):
     logger.error(f"Unhandled: {type.__name__}: {value}\n{''.join(traceback.format_tb(tb))}", exc_info=True)
-
 sys.excepthook = custom_excepthook
-
 # --- Session State Initialization ---
 def initialize_session_state():
     defaults = {
@@ -98,11 +85,8 @@ def initialize_session_state():
     for k, v in defaults.items():
         if k not in st.session_state:
             st.session_state[k] = v
-
     if st.session_state.gpu_available is None:
-        dump_env(logger)
         st.session_state.gpu_available = check_gpu(logger)
-
 # --- Set Log Level ---
 def set_log_level(name: str):
     name = (name or "INFO").upper()
@@ -113,19 +97,16 @@ def set_log_level(name: str):
     logger.setLevel(lvl)
     st.session_state.log_level_name = name
     logger.info(f"Log level set to {name}")
-
 # --- Main Processing Function ---
 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("=" * 80)
-
     st.session_state.processing = True
     st.session_state.processed_video_bytes = None
     st.session_state.last_error = None
     t0 = time.time()
-
     try:
         # Materialize uploaded video to temp file
         suffix = Path(uploaded_video.name).suffix or ".mp4"
@@ -133,27 +114,21 @@ 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
-
         # 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")
-
         # 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")
-
         # Load final video into memory for download
         with open(final_path, 'rb') as f:
             st.session_state.processed_video_bytes = f.read()
-
         total = time.time() - t0
         logger.info(f"[RUN {run_id}] SUCCESS size={len(st.session_state.processed_video_bytes)/1e6:.2f}MB, total Δ={total:.2f}s")
         return True
-
     except Exception as e:
         total = time.time() - t0
         error_msg = f"[RUN {run_id}] Processing Error: {str(e)} (Δ {total:.2f}s)\n\nCheck logs for details."
@@ -161,11 +136,9 @@ def process_video(uploaded_video, bg_image, bg_color, bg_type):
         logger.error(traceback.format_exc())
         st.session_state.last_error = error_msg
         return False
-
     finally:
         st.session_state.processing = False
         logger.info(f"[RUN {run_id}] Processing finished")
-
 # --- Main App Entry Point ---
 def main():
     st.set_page_config(
@@ -174,10 +147,8 @@ def main():
         layout="wide",
         initial_sidebar_state="expanded"
     )
-
     initialize_session_state()
     render_ui(process_video)
-
 if __name__ == "__main__":
     setup_t4_environment()
-    main()
+    main()