webapp.py

from flask import Flask
import os
import logging
import pymongo
from datetime import datetime, timezone
from config import MONGODB_URL
from typing import Dict, Any

logger = logging.getLogger(__name__)

app = Flask(__name__)

def check_mongodb_connection() -> bool:
    """Check MongoDB connection and return True if connected."""
    try:
        client = pymongo.MongoClient(MONGODB_URL)
        client.server_info()  # This will raise an exception if not connected
        logger.info("MongoDB connection successful")
        return True
    except Exception as e:
        logger.error(f"MongoDB connection failed: {str(e)}")
        return False

def get_system_status() -> Dict[str, Any]:
    """Get system status information."""
    return {
        "mongodb_connected": check_mongodb_connection(),
        "timestamp": datetime.now(timezone.utc).isoformat(),
    }

@app.route('/')
def home():
    logger.info("Home endpoint accessed")
    return "Bot is running!", 200

@app.route('/health')
def health():
    logger.info("Health check endpoint accessed")
    status = get_system_status()
    if status["mongodb_connected"]:
        return {"status": "healthy", **status}, 200
    return {"status": "unhealthy", **status}, 503

@app.route('/kaithheathcheck')
def kaithheathcheck():
    """Leapcell.io specific health check endpoint"""
    logger.info("Leapcell.io health check endpoint accessed")
    status = get_system_status()
    if status["mongodb_connected"]:
        return "OK", 200
    return "Service Unavailable", 503

def run_flask():
    """Start the Flask app in development mode"""
    port = int(os.environ.get('PORT', 8080))
    # Only run Flask directly in development
    if os.environ.get('FLASK_ENV') == 'development':
        app.run(host='0.0.0.0', port=port, debug=True)
    else:
        # In production, we use gunicorn
        pass