Update app.py

app.py

@@ -1,4 +1,4 @@
-# api.py
+# app.py
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import pandas as pd
@@ -81,14 +81,18 @@ def initialize():
     global FEATURE_COLUMNS
     try:
         # Load models
+        print("Loading models...")
         rf_model = joblib.load('phishing_detector_rf_grega.pkl')
         xgb_model = joblib.load('phishing_detector_xgb_grega.pkl')
         scaler = joblib.load('feature_scaler_grega.pkl')
+        print("Models loaded successfully!")
         
         # Load feature columns
+        print("Loading feature columns...")
         url = "https://raw.githubusercontent.com/GregaVrbancic/Phishing-Dataset/master/dataset_small.csv"
         df = pd.read_csv(url)
         FEATURE_COLUMNS = df.drop('phishing', axis=1).columns.tolist()
+        print(f"Feature columns loaded: {len(FEATURE_COLUMNS)} features")
         
         return rf_model, xgb_model, scaler
     except Exception as e:
@@ -100,7 +104,7 @@ print("Initializing models...")
 rf_model, xgb_model, scaler = initialize()
 if not rf_model:
     print("Failed to initialize models!")
-    exit()
+    # We'll continue and return error messages in the API
 
 # Fast DNS lookup with caching
 @lru_cache(maxsize=5000)
@@ -391,8 +395,12 @@ def predict_phishing_fast(url):
     }
 
 # API endpoint for checking URLs
-@app.route('/api/check', methods=['POST'])
+@app.route('/api/check', methods=['POST', 'OPTIONS'])
 def check_url():
+    # Handle OPTIONS request for CORS
+    if request.method == 'OPTIONS':
+        return '', 200
+    
     try:
         # Get JSON data from request
         data = request.get_json()
@@ -434,24 +442,41 @@ def health_check():
     return jsonify({
         'status': 'healthy',
         'message': 'Phishing detection API is running',
-        'models_loaded': rf_model is not None and xgb_model is not None and scaler is not None
+        'models_loaded': rf_model is not None and xgb_model is not None and scaler is not None,
+        'numpy_version': np.__version__
     })
 
-# Root endpoint
+# Root endpoint with API documentation
 @app.route('/', methods=['GET'])
 def root():
     return jsonify({
         'message': 'Phishing Detection API',
         'version': '1.0.0',
+        'numpy_version': np.__version__,
         'endpoints': {
             'check': '/api/check (POST)',
             'health': '/health (GET)'
         },
         'usage': {
             'check': 'POST {"url": "https://example.com"} to /api/check'
+        },
+        'example': {
+            'request': {
+                'url': 'http://paypal.secure.login-update.com'
+            },
+            'response': {
+                'success': True,
+                'data': {
+                    'url': 'http://paypal.secure.login-update.com',
+                    'is_phishing': True,
+                    'confidence': 'High',
+                    'avg_phishing_probability': 0.9137
+                }
+            }
         }
     })
 
 # Run the app
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000, debug=True)
+    port = int(os.environ.get('PORT', 7860))
+    app.run(host='0.0.0.0', port=port)