Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import time
 import uuid
 import psutil
-import mysql.connector
+import pymysql
 from flask import Flask, jsonify, request, abort, render_template_string
 
 app = Flask(__name__)
@@ -135,7 +135,15 @@ def get_mysql_connection(database=None):
     cfg = MYSQL_CONFIG.copy()
     if database:
         cfg["database"] = database
-    return mysql.connector.connect(**cfg)
+    return pymysql.connect(
+        host=cfg["host"],
+        port=cfg["port"],
+        user=cfg["user"],
+        password=cfg["password"],
+        database=cfg.get("database"),
+        autocommit=False,
+        cursorclass=pymysql.cursors.DictCursor,
+    )
 
 def get_system_stats():
     cpu_percent = psutil.cpu_percent(interval=0.1)
@@ -224,26 +232,22 @@ def create_db():
         return jsonify({"error": f"size_mb must be between {MIN_DB_MB} and {MAX_DB_MB} MB"}), 400
     if total_allocated_mb() + size_mb > TOTAL_MAX_MB:
         return jsonify({"error": "Total allocated size would exceed 25GB limit"}), 400
-
     db_uuid = str(uuid.uuid4())
     db_name = f"db_{db_uuid.replace('-', '')[:16]}"
-
     try:
         conn = get_mysql_connection()
-        conn.autocommit = True
+        conn.autocommit(True)
         cur = conn.cursor()
         cur.execute(f"CREATE DATABASE `{db_name}`")
         cur.close()
         conn.close()
     except Exception as e:
         return jsonify({"error": f"Failed to create database: {e}"}), 500
-
     db_registry[db_uuid] = {
         "db_name": db_name,
         "size_mb": size_mb,
         "created_at": time.time(),
     }
-
     return jsonify({
         "uuid": db_uuid,
         "db_name": db_name,
@@ -255,28 +259,22 @@ def create_db():
 def db_query(db_uuid):
     if db_uuid not in db_registry:
         abort(404, description="Unknown database UUID")
-
     data = request.get_json(silent=True) or {}
     query = data.get("query")
     if not query:
         return jsonify({"status": "error", "error": "Missing 'query' in JSON body"}), 400
-
     db_name = db_registry[db_uuid]["db_name"]
     try:
         conn = get_mysql_connection(database=db_name)
         cur = conn.cursor()
         cur.execute(query)
-
         rows = None
         if cur.description is not None:
-            cols = [d[0] for d in cur.description]
-            rows = [dict(zip(cols, r)) for r in cur.fetchall()]
-
+            rows = cur.fetchall()
         try:
             conn.commit()
         except Exception:
             conn.rollback()
-
         cur.close()
         conn.close()
         return jsonify({