Update src/streamlit_app.py

src/streamlit_app.py

@@ -46,11 +46,15 @@ with loading_placeholder:
     </div>
     """, unsafe_allow_html=True)
 
+# Set environment variables before any imports
+os.environ["KERAS_BACKEND"] = "torch"
+os.environ["OPENCV_LOG_LEVEL"] = "ERROR"
+os.environ["QT_QPA_PLATFORM"] = "offscreen"  # For headless OpenCV
+
 # Import dependencies with comprehensive error handling
 try:
     import numpy as np
     import pandas as pd
-    import cv2
     import time
     import tempfile
     import json
@@ -61,8 +65,27 @@ try:
     from typing import NamedTuple
     import subprocess
     
-    # Set environment variables
-    os.environ["KERAS_BACKEND"] = "torch"
+    # Import OpenCV with headless configuration and fallback
+    cv2 = None
+    try:
+        import cv2
+        # Ensure OpenCV uses headless backend
+        if hasattr(cv2, 'setNumThreads'):
+            cv2.setNumThreads(1)
+        # Test basic functionality
+        _ = cv2.__version__
+    except ImportError as opencv_error:
+        loading_placeholder.empty()
+        st.error(f"❌ OpenCV import failed: {opencv_error}")
+        st.error("Please run the setup script: `setup_windows.bat` or `./setup_windows.ps1`")
+        st.info("💡 Alternative: pip uninstall opencv-python && pip install opencv-python-headless")
+        st.stop()
+    except Exception as opencv_error:
+        loading_placeholder.empty()
+        st.error(f"❌ OpenCV configuration error: {opencv_error}")
+        st.error("This might be a headless display issue. Please use opencv-python-headless.")
+        st.info("💡 Try: pip install opencv-python-headless==4.8.1.78")
+        st.stop()
     
     # Deep learning imports
     import keras