from flask import Flask, request
from bot.handlers import handle_start, handle_button

app = Flask(__name__)


@app.route("/webhook", methods=["POST"])
def telegram_webhook():
    data = request.get_json()

    if "message" in data:
        chat_id = data["message"]["chat"]["id"]
        text = data["message"].get("text")
        if text == "/start":
            handle_start(chat_id)

    elif "callback_query" in data:
        chat_id = data["callback_query"]["message"]["chat"]["id"]
        callback_data = data["callback_query"]["data"]
        handle_button(chat_id, callback_data)

    return "OK", 200


if __name__ == "__main__":
    app.run(debug=True, port=5000)