Add debug info for model loading

app.py

@@ -170,28 +170,59 @@ BINARY_MODEL = None
 DISEASE_MODEL = None
 TEETH_HEALTH_MODEL = None
 
+# Debug: Check if files exist
+print(f"[DEBUG] Binary model exists: {os.path.exists(BINARY_MODEL_PATH)}")
+print(f"[DEBUG] Binary model size: {os.path.getsize(BINARY_MODEL_PATH) if os.path.exists(BINARY_MODEL_PATH) else 0} bytes")
+print(f"[DEBUG] Disease model exists: {os.path.exists(DISEASE_MODEL_PATH)}")
+print(f"[DEBUG] Disease model size: {os.path.getsize(DISEASE_MODEL_PATH) if os.path.exists(DISEASE_MODEL_PATH) else 0} bytes")
+
 # Load binary model
-try:
-    if os.path.exists(BINARY_MODEL_PATH):
-        print(f"[INFO] Loading binary model from {BINARY_MODEL_PATH}...")
+if os.path.exists(BINARY_MODEL_PATH):
+    print(f"[INFO] Loading binary model from {BINARY_MODEL_PATH}...")
+    try:
+        # Try different loading methods
+        print("[INFO] Attempt 1: Loading with compile=False...")
         BINARY_MODEL = tf.keras.models.load_model(BINARY_MODEL_PATH, compile=False)
-        print("[SUCCESS] Binary model loaded")
-    else:
-        raise FileNotFoundError(f"Binary model not found")
-except Exception as e:
-    print(f"[ERROR] Failed to load binary model: {e}")
+        print("[SUCCESS] Binary model loaded successfully")
+    except Exception as e1:
+        print(f"[WARNING] Attempt 1 failed: {e1}")
+        try:
+            print("[INFO] Attempt 2: Loading with custom_objects...")
+            BINARY_MODEL = tf.keras.models.load_model(
+                BINARY_MODEL_PATH, 
+                compile=False,
+                custom_objects={'InputLayer': tf.keras.layers.InputLayer}
+            )
+            print("[SUCCESS] Binary model loaded with custom_objects")
+        except Exception as e2:
+            print(f"[ERROR] Failed to load binary model: {e2}")
+            BINARY_MODEL = None
+else:
+    print(f"[ERROR] Binary model not found at {BINARY_MODEL_PATH}")
     BINARY_MODEL = None
 
 # Load disease model
-try:
-    if os.path.exists(DISEASE_MODEL_PATH):
-        print(f"[INFO] Loading disease model from {DISEASE_MODEL_PATH}...")
+if os.path.exists(DISEASE_MODEL_PATH):
+    print(f"[INFO] Loading disease model from {DISEASE_MODEL_PATH}...")
+    try:
+        print("[INFO] Attempt 1: Loading with compile=False...")
         DISEASE_MODEL = tf.keras.models.load_model(DISEASE_MODEL_PATH, compile=False)
-        print("[SUCCESS] Disease model loaded")
-    else:
-        raise FileNotFoundError(f"Disease model not found")
-except Exception as e:
-    print(f"[ERROR] Failed to load disease model: {e}")
+        print("[SUCCESS] Disease model loaded successfully")
+    except Exception as e1:
+        print(f"[WARNING] Attempt 1 failed: {e1}")
+        try:
+            print("[INFO] Attempt 2: Loading with custom_objects...")
+            DISEASE_MODEL = tf.keras.models.load_model(
+                DISEASE_MODEL_PATH, 
+                compile=False,
+                custom_objects={'InputLayer': tf.keras.layers.InputLayer}
+            )
+            print("[SUCCESS] Disease model loaded with custom_objects")
+        except Exception as e2:
+            print(f"[ERROR] Failed to load disease model: {e2}")
+            DISEASE_MODEL = None
+else:
+    print(f"[ERROR] Disease model not found at {DISEASE_MODEL_PATH}")
     DISEASE_MODEL = None
 
 # Load HuggingFace model
@@ -207,9 +238,17 @@ except Exception as e:
     print(f"[ERROR] Failed to load HuggingFace model: {e}")
     TEETH_HEALTH_MODEL = None
 
-# Verify models loaded
-if BINARY_MODEL is None or DISEASE_MODEL is None or TEETH_HEALTH_MODEL is None:
-    print("\n[ERROR] One or more models failed to load. Exiting...")
+# Final verification
+print("\n[INFO] Final model status:")
+print(f"Binary model: {'✅ LOADED' if BINARY_MODEL is not None else '❌ FAILED'}")
+print(f"Disease model: {'✅ LOADED' if DISEASE_MODEL is not None else '❌ FAILED'}")
+print(f"HuggingFace model: {'✅ LOADED' if TEETH_HEALTH_MODEL is not None else '❌ FAILED'}")
+
+# Exit if critical models missing
+if BINARY_MODEL is None or DISEASE_MODEL is None:
+    print("\n[ERROR] Critical TensorFlow models failed to load. Exiting...")
+    print("[ERROR] This is likely due to model compatibility issues.")
+    print("[ERROR] The models may need to be resaved with TensorFlow 2.12.")
     sys.exit(1)
 
 print("[INFO] All models loaded successfully\n")