main.py

import os
import logging
from app import create_app
from app.data_utils import load_data_from_db
from app.layout import create_layout
from app.callbacks import register_callbacks

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Initialize the Flask app
app = create_app()
server = app.server  # Expose Flask server for Gunicorn

def main():
    try:
        # Load data from the database
        data = load_data_from_db()
        if data.empty:
            logger.error("Failed to load data from the database.")
            return

        # Set up the app layout and callbacks
        app.layout = create_layout(data)
        register_callbacks(app, data)

    except Exception as e:
        logger.error(f"Error initializing app: {e}")
        raise

if __name__ == "__main__":
    # Get port from environment variable (Heroku sets this automatically)
    port = int(os.getenv("PORT", 8000))

    # Initialize the app
    main()

    # Run the server
    app.run_server(
        host='0.0.0.0',  # Necessary for Heroku
        port=port,
        debug=False  # Set to False for production
    )