from flask import Flask, render_template, request, jsonify
from model import load_model, ethics_checker

print("⚠️ Loading Medical model, please wait...")
model = load_model()
print("✅ Model loaded successfully!")

app = Flask(__name__)

@app.route("/")
def home():
    return render_template("index.html")

@app.route("/chat", methods=["POST"])
def chat():
    data = request.get_json()
    user_input = data.get("message", "")

    # Generate model response
    result = model.generate(user_input)[0]

    # Ethics check
    ok, cat = ethics_checker(user_input, result)
    if not ok:
        result = "⚠️ Cette question contient des informations personnelles. Reformule sans nom / identité."

    return jsonify({"response": result, "ethics": ok, "category": cat})

if __name__ == "__main__":
    # HOST must be 0.0.0.0 and PORT must be 7860 for Hugging Face
    app.run(host="0.0.0.0", port=7860)