Update app.py

app.py

@@ -19,6 +19,10 @@ default_db_config = {
     'port': int(os.getenv('DB_PORT', 4000))  # Default to TiDB port
 }
 
+# Validate default config at startup
+if not all([default_db_config['host'], default_db_config['user'], default_db_config['password']]):
+    print("Warning: Incomplete default MySQL credentials in Secrets. Default connection may fail.")
+
 # Groq API configuration with error handling
 try:
     groq_client = Groq(api_key=os.getenv('GROQ_API_KEY'))
@@ -33,16 +37,21 @@ current_summary = {}
 
 def get_db_connection(db_name=None):
     """Establish a database connection using session or default config."""
+    # Try session config first, but only if complete
     config = session.get('db_config', default_db_config).copy()
-    if not config['host'] or not config['user'] or not config['password']:
-        return None, "No valid MySQL credentials provided. Use the 'Configure MySQL Connection' modal to set up your own database, or contact the app owner to ensure default credentials are set."
+    if not all([config.get('host'), config.get('user'), config.get('password')]):
+        # Fallback to default_db_config if session config is incomplete
+        config = default_db_config.copy()
+        if not all([config.get('host'), config.get('user'), config.get('password')]):
+            return None, "No valid MySQL credentials provided. Ensure Secrets (DB_HOST, DB_USER, DB_PASSWORD, DB_PORT) are set correctly in Hugging Face Space settings, or use the 'Configure MySQL Connection' modal."
+    
     if db_name:
         config['database'] = db_name
     try:
         conn = mysql.connector.connect(**config)
         return conn, None
     except Error as e:
-        return None, f"Database connection failed: {str(e)}. Ensure the MySQL server is running, accessible, and credentials are correct."
+        return None, f"Database connection failed: {str(e)}. Verify credentials, ensure the MySQL/TiDB server is running, and check network settings (e.g., IP whitelist, TLS)."
 
 def parse_sql_file(file_content):
     """Parse SQL file to extract database name and clean statements."""
@@ -329,5 +338,11 @@ def configure_db():
     conn.close()
     return render_template('index.html', error=None, schema=current_schema, summary=current_summary, success="Custom MySQL connection configured successfully. You can now upload .sql files and query your database.")
 
+@app.route('/reset_db', methods=['POST'])
+def reset_db():
+    """Reset to default TiDB Serverless connection."""
+    session.pop('db_config', None)
+    return render_template('index.html', error=None, schema=current_schema, summary=current_summary, success="Reverted to default TiDB Serverless connection. You can now upload .sql files using the default database.")
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.getenv('PORT', 7860)), debug=False)