Ryan Chen
45a5e92aee
Reviewed-on: #4 Co-authored-by: Ryan Chen <ryan@torrtle.co> Co-committed-by: Ryan Chen <ryan@torrtle.co>
103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
import os
|
|
|
|
from quart import Quart, request, jsonify, render_template, send_from_directory
|
|
from tortoise.contrib.quart import register_tortoise
|
|
|
|
from quart_jwt_extended import JWTManager
|
|
|
|
from main import consult_simba_oracle
|
|
from blueprints.conversation.logic import (
|
|
get_the_only_conversation,
|
|
add_message_to_conversation,
|
|
)
|
|
|
|
app = Quart(
|
|
__name__,
|
|
static_folder="raggr-frontend/dist/static",
|
|
template_folder="raggr-frontend/dist",
|
|
)
|
|
|
|
app.config["JWT_SECRET_KEY"] = os.getenv("JWT_SECRET_KEY", "SECRET_KEY")
|
|
jwt = JWTManager(app)
|
|
|
|
# Initialize Tortoise ORM
|
|
register_tortoise(
|
|
app,
|
|
db_url=os.getenv("DATABASE_URL", "sqlite://raggr.db"),
|
|
modules={"models": ["blueprints.conversation.models"]},
|
|
generate_schemas=True,
|
|
)
|
|
|
|
|
|
# Serve React static files
|
|
@app.route("/static/<path:filename>")
|
|
async def static_files(filename):
|
|
return await send_from_directory(app.static_folder, filename)
|
|
|
|
|
|
# Serve the React app for all routes (catch-all)
|
|
@app.route("/", defaults={"path": ""})
|
|
@app.route("/<path:path>")
|
|
async def serve_react_app(path):
|
|
if path and os.path.exists(os.path.join(app.template_folder, path)):
|
|
return await send_from_directory(app.template_folder, path)
|
|
return await render_template("index.html")
|
|
|
|
|
|
@app.route("/api/query", methods=["POST"])
|
|
async def query():
|
|
data = await request.get_json()
|
|
query = data.get("query")
|
|
# add message to database
|
|
conversation = await get_the_only_conversation()
|
|
print(conversation)
|
|
await add_message_to_conversation(
|
|
conversation=conversation, message=query, speaker="user"
|
|
)
|
|
|
|
response = consult_simba_oracle(query)
|
|
await add_message_to_conversation(
|
|
conversation=conversation, message=response, speaker="simba"
|
|
)
|
|
return jsonify({"response": response})
|
|
|
|
|
|
@app.route("/api/messages", methods=["GET"])
|
|
async def get_messages():
|
|
conversation = await get_the_only_conversation()
|
|
# Prefetch related messages
|
|
await conversation.fetch_related("messages")
|
|
|
|
# Manually serialize the conversation with messages
|
|
messages = []
|
|
for msg in conversation.messages:
|
|
messages.append(
|
|
{
|
|
"id": str(msg.id),
|
|
"text": msg.text,
|
|
"speaker": msg.speaker.value,
|
|
"created_at": msg.created_at.isoformat(),
|
|
}
|
|
)
|
|
|
|
return jsonify(
|
|
{
|
|
"id": str(conversation.id),
|
|
"name": conversation.name,
|
|
"messages": messages,
|
|
"created_at": conversation.created_at.isoformat(),
|
|
"updated_at": conversation.updated_at.isoformat(),
|
|
}
|
|
)
|
|
|
|
|
|
# @app.route("/api/ingest", methods=["POST"])
|
|
# def webhook():
|
|
# data = request.get_json()
|
|
# print(data)
|
|
# return jsonify({"status": "received"})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=8080, debug=True)
|