File size: 1,955 Bytes
015dbc8 c42012b 015dbc8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os
from pathlib import Path
from dotenv import load_dotenv
from flask import Flask, jsonify, request
from flask_cors import CORS
from ai import generate_response
from auth import auth_blueprint, verify_request_token
BASE_DIR = Path(__file__).resolve().parent.parent
ENV_PATH = BASE_DIR / ".env"
load_dotenv(dotenv_path=ENV_PATH)
app = Flask(__name__)
CORS(app, origins=[os.getenv("FRONTEND_URL", "http://localhost:3000")])
app.register_blueprint(auth_blueprint, url_prefix="/auth")
def _validate_payload(payload):
if not payload:
return "Missing JSON payload."
if "message" not in payload:
return "Missing 'message'."
return None
@app.route("/chat", methods=["POST"])
def chat():
token_payload, token_error = verify_request_token()
if token_error:
message, status = token_error
return jsonify({"error": message}), status
user_id = token_payload.get("user_id")
if user_id is None:
return jsonify({"error": "Token missing user_id."}), 401
payload = request.get_json(silent=True)
error = _validate_payload(payload)
if error:
return jsonify({"error": error}), 400
message = payload["message"]
try:
reply = generate_response(user_id, message)
except Exception as exc: # pragma: no cover - runtime safeguard
return jsonify({"error": str(exc)}), 500
return jsonify({"reply": reply})
@app.route("/chat/history", methods=["GET"])
def chat_history():
token_payload, token_error = verify_request_token()
if token_error:
message, status = token_error
return jsonify({"error": message}), status
user_id = token_payload.get("user_id")
if user_id is None:
return jsonify({"error": "Token missing user_id."}), 401
history = db.get_conversation_history(str(user_id))
return jsonify({"conversations": history})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=5000)
|