adAstra144/sqlite-test

1from flask import Flask, request, jsonify2from flask_cors import CORS3import sqlite34from urllib.parse import urlparse5 6app = Flask(__name__)7CORS(app)8 9# Open whitelist DB10db_conn = sqlite3.connect("whitelist.db", check_same_thread=False)11 12def is_whitelisted(url):13    cursor = db_conn.cursor()14    domain = urlparse(url).netloc.lower()15    if domain.startswith("www."):16        domain = domain[4:]17    cursor.execute("SELECT 1 FROM whitelist WHERE domain = ?", (domain,))18    return cursor.fetchone() is not None19 20@app.route("/scan", methods=["POST"])21def scan():22    data = request.get_json()23    url = data.get("text", "")24 25    if is_whitelisted(url):26        return jsonify({27            "label": "Safe (Whitelisted)",28            "confidence": 1.029        })30    31    # Fake model response for testing32    return jsonify({33        "label": "Phishing",34        "confidence": 0.7535    })36 37if __name__ == "__main__":38    app.run(host="0.0.0.0", port=7860)39