Update app.py

app.py

@@ -1,5 +1,5 @@
 # Install required packages
-import os  # Added this import at the top
+import os
 import subprocess
 import sys
 
@@ -7,27 +7,54 @@ import sys
 # and install packages if needed
 if not os.path.exists("/.dockerenv") and not os.path.exists("/kaggle"):
     try:
+        # Try importing the required packages
         import gradio
-        import deepface
         import cv2
         import numpy as np
         import matplotlib
         import PIL
+        
+        # Special handling for TensorFlow and DeepFace dependencies
+        try:
+            import tensorflow as tf
+            tf_version = tf.__version__
+            print(f"TensorFlow version: {tf_version}")
+            
+            # If TensorFlow version is >=2.16, we need to install tf-keras
+            if tf_version >= "2.16.0":
+                print("Installing tf-keras for compatibility with newer TensorFlow...")
+                subprocess.check_call([sys.executable, "-m", "pip", "install", "tf-keras"])
+                
+            # Now try to import deepface
+            import deepface
+        except ImportError as e:
+            print(f"Error importing dependencies: {str(e)}")
+            print("Installing deepface with specific dependencies...")
+            # First downgrade tensorflow to a compatible version if needed
+            subprocess.check_call([sys.executable, "-m", "pip", "install", "tensorflow<2.16.0"])
+            # Then install deepface
+            subprocess.check_call([sys.executable, "-m", "pip", "install", "deepface"])
+            
     except ImportError:
         print("Installing required packages...")
         subprocess.check_call([sys.executable, "-m", "pip", "install",
-                              "gradio", "deepface", "opencv-python-headless", "numpy", "matplotlib", "pillow"])
+                              "gradio", "opencv-python-headless", "numpy", "matplotlib", "pillow"])
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "tensorflow<2.16.0"])  # Use older version
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "deepface"])
 
+# Now import the required modules
 import gradio as gr
 import json
 import cv2
 import numpy as np
-from deepface import DeepFace
-import matplotlib.pyplot as plt
 from PIL import Image
 import tempfile
 import pandas as pd
 import shutil
+import matplotlib.pyplot as plt
+
+# Import DeepFace after ensuring dependencies are properly installed
+from deepface import DeepFace
 
 def verify_faces(img1, img2, threshold=0.70, model="VGG-Face"):
     # Save uploaded images to temporary files