from flask import Flask
import os
import psycopg2

app = Flask(__name__)

DATABASE_URL = os.getenv("DATABASE_URL", "postgresql://user:password@db:5432/mydatabase")

@app.route('/')
def home():
    return "This is my application!"

@app.route('/db-test')
def db_test():
    try:
        conn = psycopg2.connect(DATABASE_URL)
        cur = conn.cursor()
        cur.execute("SELECT 'Database connection successful!'")
        message = cur.fetchone()[0]
        cur.close()
        conn.close()
        return message
    except Exception as e:
        return f"Database connection failed: {e}"

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)