from flask import Flask
import psycopg2

app = Flask(__name__)

# Database connection details
DB_USER = 'user'
DB_PASSWORD = 'password'
DB_NAME = 'mydatabase'
DB_HOST = 'db'
DB_PORT = '5432'

@app.route('/')
def index():
    # Connect to PostgreSQL database
    conn = psycopg2.connect(
        database=DB_NAME,
        user=DB_USER,
        password=DB_PASSWORD,
        host=DB_HOST,
        port=DB_PORT
    )
    
    cursor = conn.cursor()
    cursor.execute("SELECT version();")
    db_version = cursor.fetchone()

    return f"PostgreSQL database version: {db_version}"

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