from flask import Flask, jsonify
import psycopg2
import os

app = Flask(__name__)

def connect_db():
    try:
        connection = psycopg2.connect(
            dbname=os.getenv('POSTGRES_DB'),
            user=os.getenv('POSTGRES_USER'),
            password=os.getenv('POSTGRES_PASSWORD'),
            host='localhost'
        )
        cursor = connection.cursor()
        cursor.execute("SELECT version();")
        db_version = cursor.fetchone()

        return {
            "status": "success",
            "db_version": db_version,
            "user": os.getenv('POSTGRES_USER'),
            "database": os.getenv('POSTGRES_DB')
        }

    except Exception as e:
        return {"status": "failure", "error": str(e)}

@app.route('/')
def home():
    db_info = connect_db()
    return jsonify(db_info)

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