app.py

from flask import Flask, render_template, request, jsonify
from LLM_SQL import ask_llm
from db import *
import json

app = Flask(__name__)

# Initialize database on startup
init_db()
fill_db()


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

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

    # Simple placeholder logic
    #reply = f"You said: {user_message}"
    try:
        llm_query = ask_llm(user_message)
    
        if 'restricted' in llm_query:
            answer = llm_query
        else:
            results = query_db(llm_query)
            answer = json.dumps(results, indent=2)
    except:
            answer = "There was an error. Please try again."

    return jsonify({"reply": answer})

if __name__ == "__main__":
    app.run(debug=False)