Update app.py

app.py

@@ -42,26 +42,34 @@ PREDICTOR = autogluon.tabular.TabularPredictor.load(
 
 def do_predict(height, width, depth, page_count):
     try:
-        row = {
-            "Height": float(height),
-            "Width": float(width),
-            "Depth": float(depth),
-            "Page Count": int(page_count),
-        }
-        X = pandas.DataFrame([row], columns=FEATURE_COLS)
-
+        # Validate inputs
+        inputs = validate_and_scale_inputs(height, width, depth, page_count)
+        
+        X = pandas.DataFrame([inputs], columns=FEATURE_COLS)
+        
+        # Get both prediction and probabilities
+        prediction = PREDICTOR.predict(X)
         proba = PREDICTOR.predict_proba(X)
+        
+        print(f"Prediction: {prediction.iloc[0]}")
+        print(f"Probabilities: {proba.iloc[0].to_dict()}")
+        
         row0 = proba.iloc[0]
-
-        return dict(
+        
+        result = dict(
             sorted(
                 {str(cls): float(val) for cls, val in row0.items()}.items(),
                 key=lambda kv: kv[1],
                 reverse=True,
             )
         )
+        
+        print(f"Final result: {result}")
+        return result
+        
     except Exception as e:
-        return {"Error": f"Invalid input: {e}"}
+        print(f"Prediction error: {e}")
+        return {"Error": f"Prediction failed: {str(e)}"}
 
 
 EXAMPLES = [