Update app.py

app.py

@@ -1,10 +1,22 @@
 from flask import Flask, jsonify
 import psycopg2
 import os
+import socket
 
 app = Flask(__name__)
 
+def check_port(host: str, port: int):
+    """Check if a specific port on a host is open."""
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
+        sock.settimeout(1)  # 1 second timeout
+        try:
+            sock.connect((host, port))
+            return True
+        except socket.error:
+            return False
+
 def connect_db():
+    """Attempt to connect to the PostgreSQL database and retrieve version info."""
     try:
         connection = psycopg2.connect(
             dbname=os.getenv('POSTGRES_DB'),
@@ -28,8 +40,20 @@ def connect_db():
 
 @app.route('/')
 def home():
+    # Check if PostgreSQL is running and port is open
+    db_running = check_port('localhost', 5432)
+    
+    # Perform database connection test
     db_info = connect_db()
-    return jsonify(db_info)
+    
+    result = {
+        "message": "Database Connection Test Results",
+        "database_running": db_running,
+        "port_open": db_running,  # The same check, for clarity
+        "connection_result": db_info
+    }
+    
+    return jsonify(result)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)