Update app.py

app.py

@@ -11,7 +11,7 @@ from flask_cors import CORS
 app = Flask(__name__)
 CORS(app)
 
-# --- STANDARD SAMPLE DATA (Same for everyone) ---
+# --- STANDARD SAMPLE DATA ---
 SAMPLE_SQL = """
 -- Standard Sample Data
 CREATE TABLE Departments (
@@ -31,7 +31,6 @@ CREATE TABLE Employees (
     dept_id INT,
     role VARCHAR(50),
     salary DECIMAL(10, 2)
-    -- Note: Foreign keys might be loose depending on DB strictness settings
 );
 
 INSERT INTO Employees VALUES (101, 'Alice Smith', 1, 'Engineer', 90000);
@@ -74,7 +73,7 @@ def file_to_sql(file, filename):
         else:
             df = pd.read_excel(file)
         
-        # 2. Sanitize Table Name (filename without extension, alphanumeric only)
+        # 2. Sanitize Table Name
         table_name = os.path.splitext(filename)[0]
         table_name = "".join([c if c.isalnum() else "_" for c in table_name])
         if not table_name: table_name = "uploaded_table"
@@ -100,24 +99,25 @@ def file_to_sql(file, filename):
                 if pd.isna(v): 
                     vals.append("NULL")
                 elif isinstance(v, str): 
-                    # FIX: Handle nested quotes for Python < 3.12 by extracting variable
                     clean_v = str(v).replace("'", "''")
                     vals.append(f"'{clean_v}'") 
                 else: 
                     vals.append(str(v))
             inserts.append(f"INSERT INTO {table_name} VALUES ({', '.join(vals)});")
         
-        return create_stmt + "\n".join(inserts) + "\n", table_name
+        return create_stmt + "\n".join(inserts) + "\n"
     
     except Exception as e:
-        return f"-- Error converting file: {str(e)}\n", None
+        return f"-- Error converting file {filename}: {str(e)}\n"
 
 @app.route('/api/run-sql', methods=['POST'])
 def run_sql():
     sql_code = request.form.get('code', '')
     db_type = request.form.get('type', 'sqlite').lower()
     use_sample = request.form.get('use_sample') == 'true'
-    uploaded_file = request.files.get('database')
+    
+    # SUPPORT MULTIPLE FILES
+    uploaded_files = request.files.getlist('database')
     
     if not sql_code: 
         return jsonify({'output': 'No SQL code provided'}), 400
@@ -126,81 +126,66 @@ def run_sql():
     session_id = uuid.uuid4().hex[:8]
     temp_file_path = None
     
-    # Prepare Initialization SQL (Sample Data + Uploaded CSV/Excel SQL)
+    # Prepare Initialization SQL
     init_sql = ""
     
     if use_sample:
         init_sql += SAMPLE_SQL + "\n"
+
+    # Process all uploaded files (Convert CSV/Excel to SQL, Append SQL Dumps)
+    binary_db_file = None
+    
+    for file in uploaded_files:
+        fname = file.filename.lower()
         
-    if uploaded_file:
-        fname = uploaded_file.filename.lower()
+        # Data Files (CSV/Excel) -> Convert to SQL
         if fname.endswith('.csv') or fname.endswith('.xlsx') or fname.endswith('.xls'):
-            # Convert Data file to SQL
-            converted_sql, table_name = file_to_sql(uploaded_file, uploaded_file.filename)
-            init_sql += f"-- Imported from {uploaded_file.filename}\n" + converted_sql + "\n"
-            # Reset uploaded_file because we consumed it; we don't need to pass it as a binary DB
-            uploaded_file = None 
+            converted_sql = file_to_sql(file, file.filename)
+            init_sql += f"-- Imported from {file.filename}\n" + converted_sql + "\n"
+            
+        # SQL Dumps -> Append to Script
+        elif fname.endswith('.sql'):
+            content = file.read().decode('utf-8', errors='ignore')
+            init_sql += f"\n-- Dump from {file.filename}\n{content}\n"
+            
+        # SQLite Binary -> Save for later (Only 1 supported per run for SQLite)
+        elif (fname.endswith('.db') or fname.endswith('.sqlite')) and db_type == 'sqlite':
+            binary_db_file = file
 
     try:
         # --- 1. SQLITE LOGIC ---
         if db_type == 'sqlite':
             db_target = ":memory:"
             
-            # If user uploaded a binary .db file (and didn't convert it above)
-            if uploaded_file and (uploaded_file.filename.endswith('.db') or uploaded_file.filename.endswith('.sqlite')):
+            # Handle Binary DB File if present
+            if binary_db_file:
                 fd, temp_file_path = tempfile.mkstemp(suffix='.db')
                 os.close(fd)
-                uploaded_file.save(temp_file_path)
+                binary_db_file.save(temp_file_path)
                 db_target = temp_file_path
             
             cmd = ["sqlite3", db_target]
             header = ".mode column\n.headers on\n" 
-            
-            # Combine: Header + Init SQL (Sample/CSV) + User SQL
             full_script = header + init_sql + sql_code
             output = run_command(cmd, input_text=full_script)
 
         # --- 2. POSTGRESQL LOGIC ---
         elif db_type in ['postgres', 'postgresql']:
             schema_name = f"sess_{session_id}"
-            
-            # A. Create Schema
             run_command(["psql", "-h", "localhost", "-U", "playground", "-d", "playground", "-c", f"CREATE SCHEMA {schema_name};"])
-
             try:
-                # B. Handle Binary Dump Upload (if any - distinct from CSV/Excel handled in init_sql)
-                if uploaded_file: # If it's a .sql dump
-                    dump_content = uploaded_file.read().decode('utf-8', errors='ignore')
-                    init_sql += f"\n{dump_content}\n"
-
-                # C. Run Everything (Init + User Code) inside Schema
-                # We wrap everything in one transaction block pointing to the schema
                 full_script = f"SET search_path TO {schema_name};\n{init_sql}\n{sql_code}"
-                
                 output = run_command(["psql", "-h", "localhost", "-U", "playground", "-d", "playground"], input_text=full_script)
-
             finally:
                 run_command(["psql", "-h", "localhost", "-U", "playground", "-d", "playground", "-c", f"DROP SCHEMA {schema_name} CASCADE;"])
 
         # --- 3. MYSQL LOGIC ---
         elif db_type == 'mysql':
             db_name = f"sess_{session_id}"
-
-            # A. Create DB
             run_command(["mysql", "-h", "localhost", "-u", "playground", "-e", f"CREATE DATABASE {db_name};"])
-
             try:
-                 # B. Handle Binary Dump Upload
-                if uploaded_file:
-                    dump_content = uploaded_file.read().decode('utf-8', errors='ignore')
-                    init_sql += f"\n{dump_content}\n"
-
-                # C. Run Everything
-                # MySQL accepts multiple statements if we just pipe them
                 full_script = f"USE {db_name};\n{init_sql}\n{sql_code}"
-                
                 output = run_command(["mysql", "-h", "localhost", "-u", "playground", "-t"], input_text=full_script)
-
             finally:
                 run_command(["mysql", "-h", "localhost", "-u", "playground", "-e", f"DROP DATABASE {db_name};"])