Update app.py

app.py

@@ -2,41 +2,46 @@
 import os
 import subprocess
 import sys
+import importlib
+import pkg_resources
+
+def install_package(package, version=None):
+    package_spec = f"{package}=={version}" if version else package
+    print(f"Installing {package_spec}...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir", package_spec])
+    
+def ensure_package(package, version=None):
+    try:
+        if version:
+            pkg_resources.require(f"{package}=={version}")
+        else:
+            importlib.import_module(package)
+        print(f"{package} is already installed with the correct version.")
+    except (ImportError, pkg_resources.VersionConflict):
+        install_package(package, version)
 
 # Check if running in a standard environment (not Colab/Jupyter)
-# and install packages if needed
 if not os.path.exists("/.dockerenv") and not os.path.exists("/kaggle"):
-    try:
-        # First ensure numpy is correctly installed
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "numpy==1.23.5"])
-        
-        # Try importing the required packages
-        import gradio
-        import cv2
-        import numpy as np
-        import matplotlib
-        import PIL
-        
-        # Special handling for TensorFlow and DeepFace dependencies
+    print("Setting up environment...")
+    
+    # First, uninstall problematic packages to start clean
+    for pkg in ["tensorflow", "numpy", "deepface", "ml-dtypes"]:
         try:
-            # Install specific TensorFlow version
-            subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "tensorflow==2.12.0"])
-            import tensorflow as tf
-            
-            # Now try to import deepface
-            subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "deepface"])
-            import deepface
-        except ImportError as e:
-            print(f"Error importing dependencies: {str(e)}")
-            print("Installing deepface with specific dependencies...")
-            subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "deepface"])
-            
-    except ImportError:
-        print("Installing required packages...")
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall",
-                              "numpy==1.23.5", "gradio", "opencv-python-headless", "matplotlib", "pillow"])
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "tensorflow==2.12.0"])
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--force-reinstall", "deepface"])
+            subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "-y", pkg])
+        except:
+            pass
+    
+    # Install packages in the correct order with compatible versions
+    ensure_package("numpy", "1.22.4")  # Lower version than before to avoid ml_dtypes issues
+    ensure_package("protobuf", "3.20.3")  # Important for TensorFlow compatibility
+    ensure_package("tensorflow", "2.10.1")  # Earlier version with fewer dependencies
+    
+    # Install other dependencies
+    for pkg in ["gradio", "opencv-python-headless", "matplotlib", "pillow", "pandas"]:
+        ensure_package(pkg)
+    
+    # Install deepface last after all dependencies are set up
+    ensure_package("deepface")
 
 # Now import the required modules
 import gradio as gr