Update handler.py

handler.py

@@ -18,35 +18,41 @@ class EndpointHandler:
         review_tfidf = self.vectorizer.transform([review])
         predicted_rating = self.model.predict(review_tfidf)[0]
         return int(predicted_rating)
-def __call__(self, inputs):
-    try:
-        inputs_dict = json.loads(inputs)
 
-        if 'inputs' not in inputs_dict:
-            return json.dumps({"error": "No valid inputs provided"})
+    # __call__ method should be part of the class
+    def __call__(self, inputs):
+        try:
+            # Parse the input JSON string
+            inputs_dict = json.loads(inputs)
 
-        inputs_data = inputs_dict['inputs']
+            # Check if 'inputs' key exists
+            if 'inputs' not in inputs_dict:
+                return json.dumps({"error": "No 'inputs' key provided in the JSON input."})
 
-        if 'review' not in inputs_data:
-            return json.dumps({"error": "No valid review provided"})
+            inputs_data = inputs_dict['inputs']
 
-        review = inputs_data['review']
+            # Check if 'review' key exists
+            if 'review' not in inputs_data:
+                return json.dumps({"error": "No 'review' key provided in the 'inputs' object."})
 
-        if not review:
-            return json.dumps({"error": "No valid review provided"})
+            review = inputs_data['review']
 
-        predicted_rating = self.predict_rating(review)
+            # Validate that the review is a non-empty string
+            if not isinstance(review, str) or not review.strip():
+                return json.dumps({"error": "Review must be a non-empty string."})
 
-        response = {
-            "review": review,
-            "predicted_rating": predicted_rating
-        }
+            predicted_rating = self.predict_rating(review)
 
-        return json.dumps(response)
+            response = {
+                "review": review,
+                "predicted_rating": predicted_rating
+            }
 
-    except json.JSONDecodeError:
-        return json.dumps({"error": "Invalid JSON format"})
+            return json.dumps(response)
 
-    except Exception as e:
-        return json.dumps({"error": str(e)})
+        except json.JSONDecodeError:
+            return json.dumps({"error": "Invalid JSON format in input."})
+
+        except Exception as e:
+            return json.dumps({"error": str(e)})