Update main.py

main.py

@@ -57,13 +57,14 @@ CORS(app)
 _CONN = None
 
 def _connect():
-    """(Re)connect to TiDB with TLS + autocommit; DictCursor for JSON friendliness."""
+    """Enhanced connection with better error handling"""
     global _CONN
     try:
         if _CONN:
             _CONN.close()
     except Exception:
         pass
+    
     _CONN = pymysql.connect(
         host=TIDB_HOST,
         port=TIDB_PORT,
@@ -73,21 +74,32 @@ def _connect():
         ssl={"ca": certifi.where()},
         ssl_verify_cert=True,
         ssl_verify_identity=True,
-        autocommit=True,  # ensure object rows are durable before child rows (FKs)
+        autocommit=True,
         charset="utf8mb4",
         cursorclass=pymysql.cursors.DictCursor,
+        # Add these timeouts:
+        connect_timeout=10,
+        read_timeout=30,
+        write_timeout=30,
     )
 
 def _ensure_conn():
     global _CONN
-    if _CONN is None:
-        _connect()
-    else:
+    max_retries = 3
+    for attempt in range(max_retries):
         try:
-            _CONN.ping(reconnect=True)
-        except Exception:
-            _connect()
-    return _CONN
+            if _CONN is None:
+                _connect()
+            else:
+                _CONN.ping(reconnect=False)  # Test without auto-reconnect first
+            return _CONN
+        except Exception as e:
+            log.warning(f"Connection attempt {attempt + 1} failed: {e}")
+            _CONN = None
+            if attempt < max_retries - 1:
+                time.sleep(0.5 * (attempt + 1))  # Backoff
+            else:
+                raise
 
 @contextmanager
 def cursor():