Mazoura commited on
Commit
f244650
·
verified ·
1 Parent(s): adfc19d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -0
app.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, render_template, request, jsonify
2
+ from model import load_model, ethics_checker
3
+
4
+ print("⚠️ Loading Medical model, please wait...")
5
+ model = load_model()
6
+ print("✅ Model loaded successfully!")
7
+
8
+ app = Flask(__name__)
9
+
10
+ @app.route("/")
11
+ def home():
12
+ return render_template("index.html")
13
+
14
+ @app.route("/chat", methods=["POST"])
15
+ def chat():
16
+ data = request.get_json()
17
+ user_input = data.get("message", "")
18
+
19
+ # Generate model response
20
+ result = model.generate(user_input)[0]
21
+
22
+ # Ethics check
23
+ ok, cat = ethics_checker(user_input, result)
24
+ if not ok:
25
+ result = "⚠️ Cette question contient des informations personnelles. Reformule sans nom / identité."
26
+
27
+ return jsonify({"response": result, "ethics": ok, "category": cat})
28
+
29
+ if __name__ == "__main__":
30
+ # HOST must be 0.0.0.0 and PORT must be 7860 for Hugging Face
31
+ app.run(host="0.0.0.0", port=7860)