from flask import Flask
from src.models import db
from src.config import config
from sqlalchemy import text

app = Flask(__name__)
app.config.from_object(config['default'])
db.init_app(app)

if __name__ == '__main__':
    with app.app_context():
        try:
            db.session.execute(text('ALTER TABLE chat_messages ADD COLUMN response_time FLOAT NULL'))
            db.session.commit()
            print("Coluna response_time adicionada com sucesso!")
        except Exception as e:
            print(f"[Aviso] response_time: {str(e)}")
            db.session.rollback()
        try:
            db.session.execute(text('ALTER TABLE chat_messages ADD COLUMN tokens INTEGER NULL'))
            db.session.commit()
            print("Coluna tokens adicionada com sucesso!")
        except Exception as e:
            print(f"[Aviso] tokens: {str(e)}")
            db.session.rollback()
        try:
            db.session.execute(text('ALTER TABLE chats ADD COLUMN model VARCHAR(120) NULL'))
            db.session.commit()
            print("Coluna model adicionada com sucesso!")
        except Exception as e:
            print(f"[Aviso] model: {str(e)}")
            db.session.rollback()