Update app.py

app.py

@@ -6,23 +6,63 @@ import torch
 import os
 
 def load_model_and_data():
+    """Load the trained model and necessary data"""
+    import os
+    
+    # Debug: Check what files exist
+    print("Checking for files...")
+    print(f"Current directory: {os.getcwd()}")
+    print(f"Files in current directory: {os.listdir('.')}")
+    
+    if os.path.exists('model_artifacts'):
+        print(f"Files in model_artifacts/: {os.listdir('model_artifacts')}")
+    else:
+        print("ERROR: model_artifacts/ folder does not exist!")
+    
     try:
+        # Check each file individually
+        files_to_check = [
+            'model_artifacts/hybrid_model.pkl',
+            'model_artifacts/loader.pkl',
+            'model_artifacts/movies.pkl'
+        ]
+        
+        for file_path in files_to_check:
+            if not os.path.exists(file_path):
+                print(f"ERROR: Missing file: {file_path}")
+            else:
+                file_size = os.path.getsize(file_path) / (1024*1024)  # MB
+                print(f"Found: {file_path} ({file_size:.2f} MB)")
+        
+        # Load files
         with open('model_artifacts/hybrid_model.pkl', 'rb') as f:
             model = pickle.load(f)
+        print("✓ Loaded hybrid_model.pkl")
         
         with open('model_artifacts/loader.pkl', 'rb') as f:
             loader = pickle.load(f)
+        print("✓ Loaded loader.pkl")
         
         with open('model_artifacts/movies.pkl', 'rb') as f:
             movies = pickle.load(f)
+        print("✓ Loaded movies.pkl")
         
+        # Get list of users
         user_ids = sorted(loader.user_id_map.keys())
+        print(f"✓ Model loaded successfully! {len(user_ids)} users available")
         
         return model, loader, movies, user_ids
+    except FileNotFoundError as e:
+        print(f"ERROR: File not found - {e}")
+        print("Make sure all pkl files are in the model_artifacts/ folder")
+        return None, None, None, []
     except Exception as e:
-        print(f"Error loading model: {e}")
+        print(f"ERROR loading model: {type(e).__name__}: {e}")
+        import traceback
+        traceback.print_exc()
         return None, None, None, []
 
+
 print("Loading model and data...")
 model, loader, movies_df, user_ids = load_model_and_data()
 print(f"Model loaded! Available users: {len(user_ids)}")