Update app.py

app.py

@@ -11,12 +11,12 @@ app = Flask(__name__)
 app.secret_key = os.urandom(24)  # Required for session management
 load_dotenv()
 
-# Default database configuration from .env (no localhost default)
+# Default database configuration from .env
 default_db_config = {
-    'host': os.getenv('DB_HOST', ''),
-    'user': os.getenv('DB_USER', 'root'),
-    'password': os.getenv('DB_PASSWORD', ''),
-    'port': int(os.getenv('DB_PORT', 3306))
+    'host': os.getenv('DB_HOST'),
+    'user': os.getenv('DB_USER'),
+    'password': os.getenv('DB_PASSWORD'),
+    'port': int(os.getenv('DB_PORT', 4000))  # Default to TiDB port
 }
 
 # Groq API configuration with error handling
@@ -34,15 +34,15 @@ current_summary = {}
 def get_db_connection(db_name=None):
     """Establish a database connection using session or default config."""
     config = session.get('db_config', default_db_config).copy()
-    if not config['host'] or not config['user']:
-        return None, "Please configure a valid MySQL connection using the 'Configure MySQL Connection' modal."
+    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 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 and accessible, and check your credentials."
+        return None, f"Database connection failed: {str(e)}. Ensure the MySQL server is running, accessible, and credentials are correct."
 
 def parse_sql_file(file_content):
     """Parse SQL file to extract database name and clean statements."""
@@ -308,10 +308,10 @@ def configure_db():
     host = request.form.get('host', '').strip()
     user = request.form.get('user', '').strip()
     password = request.form.get('password', '')
-    port = request.form.get('port', '3306').strip()
+    port = request.form.get('port', '4000').strip()
 
     if not host or not user:
-        return render_template('index.html', error="Host and user are required.", schema=current_schema, summary=current_summary)
+        return render_template('index.html', error="Host and user are required for custom MySQL configuration.", schema=current_schema, summary=current_summary)
 
     try:
         port = int(port)
@@ -327,7 +327,7 @@ def configure_db():
     # Store in session
     session['db_config'] = test_config
     conn.close()
-    return render_template('index.html', error=None, schema=current_schema, summary=current_summary, success="MySQL connection configured successfully.")
+    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.")
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.getenv('PORT', 7860)), debug=False)