Update app.py

app.py

@@ -68,7 +68,9 @@ class DatabaseConnection(BaseModel):
 @app.post("/api/connect")
 async def connect_to_database(connection: DatabaseConnection):
     try:
-        if connection.database_type == "postgres":
+        print(f"Attempting to connect to database: {connection.database_type}")
+        if connection.database_type == "Postgres":
+            print(f"PostgreSQL connection details - Host: {connection.server}, Port: {connection.port}, Database: {connection.databaseName}, User: {connection.username}")
             conn = psycopg2.connect(
                 host=connection.server,
                 port=connection.port,
@@ -78,7 +80,6 @@ async def connect_to_database(connection: DatabaseConnection):
             )
             query_schemas = "SELECT schema_name FROM information_schema.schemata"
             query_tables = "SELECT table_name FROM information_schema.tables WHERE table_schema = %s"
-            print(query_tables)
         elif connection.database_type == "mysql":
             print(f"inside mysql",connection.server,connection.port,connection.databaseName,connection.username,connection.password)
             conn = mysql.connector.connect(
@@ -115,6 +116,14 @@ async def connect_to_database(connection: DatabaseConnection):
         raise HTTPException(status_code=500, detail=str(e))
 
 # Function to create a new database connection for MySQL (Example)
+def get_db_connection():
+    try:
+        cnx = mysql.connector.connect(user=DB_USER, password=DB_PASSWORD, host=DB_HOST, database=DB_NAME)
+        return cnx
+    except mysql.connector.Error as err:
+        logging.error(f"Database connection error: {err}")
+        return None
+# Function to create a new database connection for MySQL (Example)
 def get_db_connection():
     try:
         cnx = mysql.connector.connect(user=DB_USER, password=DB_PASSWORD, host=DB_HOST, database=DB_NAME)