app.py

from flask import Flask, request

from api import db, ai

app = Flask(__name__)

@app.route("/call",methods=["POST"])
def callback_callmessage():
    """
    {
        "user_id": <UID>,
        "chat_id": <INT>,
        "content": <TEXT>
    }
    """
    uid: str = request.json["user_id"]
    chat_id: int = int(request.json["chat_id"])
    content: str = request.json["content"]

    # get chat history
    chat = db.getchat(uid, chat_id)

    # send add message query to db
    new_message = db.addmessage(uid, chat_id, content, "user")

    # new message append to chat(variable)
    chat.messages.append(new_message)

    reply = ai.run(chat)

    # send add reply query to db
    reply_message = db.addmessage(uid, chat_id, reply, "assistant")

    chat.messages.append(reply_message)
    return chat.get_content()

@app.route("/chats/ids",methods=["GET"])
def callback_get_user_chatids():
    user_id: str = request.args.get("id", type=str)
    return {"result": db.getuserschatid(user_id)}

@app.route("/chats/messages",methods=["GET"])
def callback_get_chat_messages():
    """
    {
        "user_id": <UID>,
        "chat_id": <INT>
    }
    """
    user_id: str = request.args.get("user_id", type=str)
    chat_id: int = request.args.get("chat_id", type=int)
    return db.getchat(user_id, chat_id).get_content()

@app.route("/chats/new",methods=["POST"])
def callback_new_chat():
    """
        {
            "user_id": <UID>
        }
    """
    user_id: str = request.json["user_id"]
    return db.newchat(user_id)

@app.route("/",methods=["GET"])
def callback_route_get():
    return "THIS IS GET TEST 0825."