Upload index.html

static/index.html

@@ -1444,28 +1444,48 @@ async function uploadDatabase() {
     const data = await res.json();
     
     if (res.ok && data.status === 'ok') {
-      mode = 'custom';
       customDbLoaded = true;
       customDbSchema = data.tables;
       customDbFilename = data.filename;
       firstCustomTable = data.tables[0]?.table_name || 'table';
       
-      // Update UI
-      document.getElementById('upload-status').innerHTML = '✓ ' + data.filename + ' loaded (' + data.tables.length + ' tables)';
-      document.getElementById('upload-status').style.color = 'var(--success)';
-      document.getElementById('back-to-tasks-btn').style.display = 'inline-flex';
-      
-      // Update schema panel
-      renderSchema(data.tables);
-      
-      // Clear and set placeholder
-      document.getElementById('query-sql').value = '';
-      document.getElementById('query-sql').placeholder = 'SELECT * FROM ' + firstCustomTable + ' LIMIT 10';
-      
-      // Show custom mode UI
-      updateUIForMode();
-      
-      toast('Database uploaded successfully', 'success');
+      // Store compatible tasks
+      if (data.compatible_tasks && data.compatible_tasks.length > 0) {
+        window.uploadedCompatibleTasks = data.compatible_tasks;
+        window.usingUploadedDbForTasks = true;
+        
+        // Enable task selection for uploaded DB
+        mode = 'task';  // Stay in task mode, not custom mode
+        
+        // Update UI
+        document.getElementById('upload-status').innerHTML = '✓ ' + data.filename + ' loaded (' + data.tables.length + ' tables) - <strong>' + data.compatible_tasks.length + ' tasks available</strong>';
+        document.getElementById('upload-status').style.color = 'var(--success)';
+        document.getElementById('back-to-tasks-btn').style.display = 'inline-flex';
+        
+        // Update schema panel
+        renderSchema(data.tables);
+        
+        // Show task selection for uploaded DB
+        showUploadedDbTaskSelector(data.compatible_tasks);
+        
+        toast('Database uploaded! ' + data.compatible_tasks.length + ' tasks available', 'success');
+      } else {
+        // No compatible tasks - go to custom exploration mode
+        mode = 'custom';
+        window.uploadedCompatibleTasks = [];
+        window.usingUploadedDbForTasks = false;
+        
+        document.getElementById('upload-status').innerHTML = '✓ ' + data.filename + ' loaded (' + data.tables.length + ' tables) - <em>No RL tasks compatible</em>';
+        document.getElementById('upload-status').style.color = 'var(--text-secondary)';
+        document.getElementById('back-to-tasks-btn').style.display = 'inline-flex';
+        
+        renderSchema(data.tables);
+        document.getElementById('query-sql').value = '';
+        document.getElementById('query-sql').placeholder = 'SELECT * FROM ' + firstCustomTable + ' LIMIT 10';
+        updateUIForMode();
+        
+        toast('Database uploaded for exploration only (no compatible RL tasks)', 'success');
+      }
     } else {
       document.getElementById('upload-error').style.display = 'block';
       document.getElementById('upload-error').innerHTML = escapeHtml(data.detail || 'Upload failed');
@@ -1480,6 +1500,72 @@ async function uploadDatabase() {
   setLoading('upload', false);
 }
 
+function showUploadedDbTaskSelector(compatibleTasks) {
+  // Create or update task selector for uploaded DB
+  let selector = document.getElementById('uploaded-db-task-selector');
+  if (!selector) {
+    selector = document.createElement('div');
+    selector.id = 'uploaded-db-task-selector';
+    selector.className = 'help-section';
+    selector.style.cssText = 'background: #E8F5E9; border-color: #81C784; margin-top: 12px;';
+    document.getElementById('upload-section').appendChild(selector);
+  }
+  
+  const taskLabels = {
+    'null_audit': 'Null Audit',
+    'type_repair': 'Type Repair',
+    'dedup_bug': 'Duplicate Bug',
+    'fk_integrity': 'FK Integrity',
+    'schema_drift': 'Schema Drift'
+  };
+  
+  let html = '<div style="font-size: 14px; color: #2E7D32; margin-bottom: 8px;"><strong>📊 Select a Task to Run on Uploaded Database:</strong></div>';
+  html += '<div style="display: flex; gap: 8px; flex-wrap: wrap;">';
+  
+  compatibleTasks.forEach(taskId => {
+    html += '<button class="btn btn-primary" onclick="selectUploadedDbTask(\'' + taskId + '\')" style="font-size: 12px; padding: 6px 12px;">' + taskLabels[taskId] + '</button>';
+  });
+  
+  html += '</div>';
+  html += '<div style="margin-top: 8px; font-size: 11px; color: #558B2F;">These tasks will analyze your uploaded database</div>';
+  
+  selector.innerHTML = html;
+  selector.style.display = 'block';
+}
+
+function selectUploadedDbTask(taskId) {
+  // Update task selector to match
+  document.getElementById('task-select').value = taskId;
+  onTaskChange();
+  
+  // Add indicator that we're using uploaded DB
+  let indicator = document.getElementById('uploaded-db-indicator');
+  if (!indicator) {
+    indicator = document.createElement('div');
+    indicator.id = 'uploaded-db-indicator';
+    indicator.className = 'help-section';
+    indicator.style.cssText = 'background: #FFF3E0; border-color: #FFB74D; margin-bottom: 12px;';
+    document.querySelector('.main').insertBefore(indicator, document.querySelector('.main').firstChild);
+  }
+  
+  const taskLabels = {
+    'null_audit': 'Null Audit',
+    'type_repair': 'Type Repair',
+    'dedup_bug': 'Duplicate Bug',
+    'fk_integrity': 'FK Integrity',
+    'schema_drift': 'Schema Drift'
+  };
+  
+  indicator.innerHTML = '<div style="font-size: 13px; color: #E65100;"><strong>🗄️ Running on Uploaded Database:</strong> ' + taskLabels[taskId] + ' task using your uploaded data</div>';
+  indicator.style.display = 'block';
+  
+  // Hide the task selector after selection
+  const selector = document.getElementById('uploaded-db-task-selector');
+  if (selector) selector.style.display = 'none';
+  
+  toast('Task selected: ' + taskLabels[taskId] + ' (using uploaded DB)', 'success');
+}
+
 function backToTasks() {
   mode = 'task';
   customDbLoaded = false;