Update handler.py

handler.py

@@ -1,9 +1,9 @@
-
 from typing import Dict, List, Any
 import pickle
 import os
 import __main__
-
+import numpy as np
+import pandas as pd
 
 class ContentBasedRecommender:
     def __init__(self, train_data):
@@ -16,8 +16,8 @@ class ContentBasedRecommender:
 
         return np.random.choice(recommended_books, size=min(k, len(recommended_books)), replace=False).tolist()
 
-
 __main__.ContentBasedRecommender = ContentBasedRecommender
+
 class EndpointHandler:
     def __init__(self, path=""):
         model_path = os.path.join(path, "model.pkl")
@@ -25,8 +25,15 @@ class EndpointHandler:
             self.model = pickle.load(f)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        user_id = data.pop("user_id", None)
-        k = data.pop("k", 10)  # Default to 10 if not provided
+        # Extract the 'inputs' from the data
+        inputs = data.get('inputs', {})
+        
+        # If inputs is a string (for single user_id input), convert it to a dict
+        if isinstance(inputs, str):
+            inputs = {'user_id': inputs}
+        
+        user_id = inputs.get('user_id')
+        k = inputs.get('k', 10)  # Default to 10 if not provided
 
         if user_id is None:
             return [{"error": "user_id is required"}]