Update api/app.py

api/app.py

@@ -4,11 +4,18 @@ import os
 import sys
 
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-from api.predict import predict_review
+from api.predict import predict_review, load_resources
 
 app = Flask(__name__)
 CORS(app)
 
+print("loading models on startup...")
+try:
+    load_resources()
+    print("models loaded")
+except Exception as e:
+    print(f"couldnt preload models: {e}")
+
 @app.route('/health', methods=['GET'])
 def health():
     return jsonify({"status": "ok"}), 200
@@ -31,10 +38,15 @@ def predict():
         
         result = predict_review(reviewText)
         
+        if 'error' in result:
+            return jsonify({"error": result['error']}), 400
+        
         return jsonify({
             "prediction": result['prediction'],
             "confidence": result['confidence'],
-            "is_fake": result['is_fake']
+            "is_fake": result['is_fake'],
+            "length_category": result.get('length_category'),
+            "token_count": result.get('token_count')
         }), 200
         
     except Exception as e:
@@ -42,4 +54,4 @@ def predict():
 
 if __name__ == '__main__':
     print("starting api server")
-    app.run(host='0.0.0.0', port=5000, debug=False)
+    app.run(host='0.0.0.0', port=5000, debug=False)