from flask import Flask, render_template, request
import ollama

app = Flask(__name__)

MODEL_NAME = "bts"

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

@app.route("/get", methods=["POST"])
def chat():
    user_message = request.form["msg"]

    try:
        response = ollama.chat(
            model=MODEL_NAME,
            messages=[{"role": "user", "content": user_message}]
        )
        reply = response['message']['content']
    except Exception as e:
        reply = f"Error: {str(e)}"

    return reply

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000)