from flask import Flask
from db import get_db_connection

app = Flask(__name__)

@app.route("/")
def index():
    return "OTB Billing v0.0.3 running"

@app.route("/dbtest")
def dbtest():
    try:
        conn = get_db_connection()
        cursor = conn.cursor()
        cursor.execute("SELECT NOW()")
        result = cursor.fetchone()
        conn.close()

        return f"""
        <h1>Database Connection OK</h1>
        <p>Server time: {result[0]}</p>
        """

    except Exception as e:
        return f"""
        <h1>Database Connection FAILED</h1>
        <pre>{e}</pre>
        """
        
if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5050)