alchemist-nrel 0.3.1__tar.gz → 0.3.2__tar.gz

{alchemist_nrel-0.3.1/alchemist_nrel.egg-info → alchemist_nrel-0.3.2}/PKG-INFO

@@ -1,14 +1,14 @@
 Metadata-Version: 2.4
 Name: alchemist-nrel
-Version: 0.3.1
+Version: 0.3.2
 Summary: Active learning and optimization toolkit for chemical and materials research
 Author-email: Caleb Coatney <caleb.coatney@nrel.gov>
 License: BSD-3-Clause
-Project-URL: Homepage, https://github.com/NREL/ALchemist
-Project-URL: Documentation, https://nrel.github.io/ALchemist/
-Project-URL: Source, https://github.com/NREL/ALchemist
-Project-URL: Bug Tracker, https://github.com/NREL/ALchemist/issues
-Project-URL: Changelog, https://github.com/NREL/ALchemist/releases
+Project-URL: Homepage, https://github.com/NatLabRockies/ALchemist
+Project-URL: Documentation, https://natlabrockies.github.io/ALchemist/
+Project-URL: Source, https://github.com/NatLabRockies/ALchemist
+Project-URL: Bug Tracker, https://github.com/NatLabRockies/ALchemist/issues
+Project-URL: Changelog, https://github.com/NatLabRockies/ALchemist/releases
 Keywords: active learning,bayesian optimization,gaussian processes,materials science,chemistry
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Science/Research
@@ -70,7 +70,7 @@ ALchemist is a modular Python toolkit that brings active learning and Bayesian o
 ## Documentation
 
 Full user guide and documentation:  
-[https://nrel.github.io/ALchemist/](https://nrel.github.io/ALchemist/)
+[https://natlabrockies.github.io/ALchemist/](https://natlabrockies.github.io/ALchemist/)
 
 ---
 
@@ -108,7 +108,7 @@ Session files (JSON format) are fully interoperable between desktop and web appl
 - **Autonomous Optimization**: Use the REST API to integrate ALchemist with automated laboratory equipment for real-time process control
 - **Remote Monitoring**: Web dashboard provides read-only monitoring mode when ALchemist is being remote-controlled
 
-For detailed application examples, see [Use Cases](https://nrel.github.io/ALchemist/use_cases/) in the documentation.
+For detailed application examples, see [Use Cases](https://natlabrockies.github.io/ALchemist/use_cases/) in the documentation.
 
 ---
 
@@ -133,10 +133,10 @@ pip install alchemist-nrel
 > *Note: This installs the latest unreleased version. The web application is not pre-built with this method because static build files are not included in the repository.*
 
 ```bash
-pip install git+https://github.com/NREL/ALchemist.git
+pip install git+https://github.com/NatLabRockies/ALchemist.git
 ```
 
-For advanced installation options, Docker deployment, and development setup, see the [Advanced Installation Guide](https://nrel.github.io/ALchemist/#advanced-installation) in the documentation.
+For advanced installation options, Docker deployment, and development setup, see the [Advanced Installation Guide](https://natlabrockies.github.io/ALchemist/#advanced-installation) in the documentation.
 
 ---
 
@@ -153,7 +153,7 @@ Opens at [http://localhost:8000/app](http://localhost:8000/app)
 alchemist
 ```
 
-For detailed usage instructions, see [Getting Started](https://nrel.github.io/ALchemist/) in the documentation.
+For detailed usage instructions, see [Getting Started](https://natlabrockies.github.io/ALchemist/) in the documentation.
 
 ---
 
@@ -165,7 +165,7 @@ ALchemist is under active development at NLR as part of the DataHub project with
 
 ## Issues & Troubleshooting
 
-If you encounter any issues or have questions, please [open an issue on GitHub](https://github.com/NREL/ALchemist/issues) or contact ccoatney@nrel.gov.
+If you encounter any issues or have questions, please [open an issue on GitHub](https://github.com/NatLabRockies/ALchemist/issues) or contact ccoatney@nrel.gov.
 
 For the latest known issues and troubleshooting tips, see the [Issues & Troubleshooting Log](docs/ISSUES_LOG.md).
 
@@ -181,5 +181,5 @@ This project is licensed under the BSD 3-Clause License. See the [LICENSE](LICEN
 
 ## Repository
 
-[https://github.com/NREL/ALchemist](https://github.com/NREL/ALchemist)
+[https://github.com/NatLabRockies/ALchemist](https://github.com/NatLabRockies/ALchemist)
 

{alchemist_nrel-0.3.1 → alchemist_nrel-0.3.2}/README.md

@@ -11,7 +11,7 @@ ALchemist is a modular Python toolkit that brings active learning and Bayesian o
 ## Documentation
 
 Full user guide and documentation:  
-[https://nrel.github.io/ALchemist/](https://nrel.github.io/ALchemist/)
+[https://natlabrockies.github.io/ALchemist/](https://natlabrockies.github.io/ALchemist/)
 
 ---
 
@@ -49,7 +49,7 @@ Session files (JSON format) are fully interoperable between desktop and web appl
 - **Autonomous Optimization**: Use the REST API to integrate ALchemist with automated laboratory equipment for real-time process control
 - **Remote Monitoring**: Web dashboard provides read-only monitoring mode when ALchemist is being remote-controlled
 
-For detailed application examples, see [Use Cases](https://nrel.github.io/ALchemist/use_cases/) in the documentation.
+For detailed application examples, see [Use Cases](https://natlabrockies.github.io/ALchemist/use_cases/) in the documentation.
 
 ---
 
@@ -74,10 +74,10 @@ pip install alchemist-nrel
 > *Note: This installs the latest unreleased version. The web application is not pre-built with this method because static build files are not included in the repository.*
 
 ```bash
-pip install git+https://github.com/NREL/ALchemist.git
+pip install git+https://github.com/NatLabRockies/ALchemist.git
 ```
 
-For advanced installation options, Docker deployment, and development setup, see the [Advanced Installation Guide](https://nrel.github.io/ALchemist/#advanced-installation) in the documentation.
+For advanced installation options, Docker deployment, and development setup, see the [Advanced Installation Guide](https://natlabrockies.github.io/ALchemist/#advanced-installation) in the documentation.
 
 ---
 
@@ -94,7 +94,7 @@ Opens at [http://localhost:8000/app](http://localhost:8000/app)
 alchemist
 ```
 
-For detailed usage instructions, see [Getting Started](https://nrel.github.io/ALchemist/) in the documentation.
+For detailed usage instructions, see [Getting Started](https://natlabrockies.github.io/ALchemist/) in the documentation.
 
 ---
 
@@ -106,7 +106,7 @@ ALchemist is under active development at NLR as part of the DataHub project with
 
 ## Issues & Troubleshooting
 
-If you encounter any issues or have questions, please [open an issue on GitHub](https://github.com/NREL/ALchemist/issues) or contact ccoatney@nrel.gov.
+If you encounter any issues or have questions, please [open an issue on GitHub](https://github.com/NatLabRockies/ALchemist/issues) or contact ccoatney@nrel.gov.
 
 For the latest known issues and troubleshooting tips, see the [Issues & Troubleshooting Log](docs/ISSUES_LOG.md).
 
@@ -122,5 +122,5 @@ This project is licensed under the BSD 3-Clause License. See the [LICENSE](LICEN
 
 ## Repository
 
-[https://github.com/NREL/ALchemist](https://github.com/NREL/ALchemist)
+[https://github.com/NatLabRockies/ALchemist](https://github.com/NatLabRockies/ALchemist)
 

{alchemist_nrel-0.3.1 → alchemist_nrel-0.3.2}/alchemist-web/package.json

@@ -1,7 +1,7 @@
 {
   "name": "alchemist-web",
   "private": true,
-  "version": "0.3.1",
+  "version": "0.3.2",
   "type": "module",
   "scripts": {
     "dev": "vite",

{alchemist_nrel-0.3.1 → alchemist_nrel-0.3.2}/alchemist-web/src/App.tsx

@@ -3,7 +3,6 @@ import { Toaster, toast } from 'sonner';
 import { QueryProvider } from './providers/QueryProvider';
 import { VisualizationProvider, useVisualization } from './providers/VisualizationProvider';
 import { 
-  getStoredSessionId, 
   clearStoredSessionId, 
   useCreateSession, 
   useSession,
@@ -11,7 +10,7 @@ import {
   useImportSession,
   useUpdateSessionMetadata
 } from './hooks/api/useSessions';
-import { useLockStatus } from './hooks/useLockStatus';
+import { useSessionEvents } from './hooks/useSessionEvents';
 import { VariablesPanel } from './features/variables/VariablesPanel';
 import { ExperimentsPanel } from './features/experiments/ExperimentsPanel';
 import { InitialDesignPanel } from './features/experiments/InitialDesignPanel';
@@ -31,7 +30,11 @@ function AppContent() {
   const [showMetadataDialog, setShowMetadataDialog] = useState(false);
   const [copiedSessionId, setCopiedSessionId] = useState(false);
   const [sessionFromUrl, setSessionFromUrl] = useState<boolean>(false);
+  const [showRecoveryBanner, setShowRecoveryBanner] = useState(false);
+  const [recoverySession, setRecoverySession] = useState<any>(null);
   const fileInputRef = useRef<HTMLInputElement>(null);
+  const urlSessionRef = useRef<string | null>(null);
+  const [joinSessionId, setJoinSessionId] = useState('');
   
   const createSession = useCreateSession();
   const exportSession = useExportSession();
@@ -41,25 +44,43 @@ function AppContent() {
   const { theme, toggleTheme } = useTheme();
   const { isVisualizationOpen, closeVisualization, sessionId: vizSessionId } = useVisualization();
   
-  // Monitor lock status and auto-switch to monitoring mode
-  const { lockStatus } = useLockStatus(sessionId, 5000, (locked) => {
+  // Monitor session events (lock status, experiment updates, model training) via WebSocket
+  const { lockStatus } = useSessionEvents(sessionId, (locked) => {
     setIsMonitoringMode(locked);
   });
   
   // Global staged suggestions to mirror desktop main_app.pending_suggestions
   const [pendingSuggestions, setPendingSuggestions] = useState<any[]>([]);
-  const [loadedFromFile, setLoadedFromFile] = useState<boolean>(false);
 
-  // Restore pending suggestions from audit log on session load (desktop workflow)
+  // Restore pending suggestions from staged experiments API on session load
   useEffect(() => {
     if (!sessionId) return;
     
-    async function restorePendingSuggestions() {
+    async function restoreStagedExperiments() {
       try {
-        const response = await fetch(`/api/v1/sessions/${sessionId}/audit?entry_type=acquisition_locked`);
-        if (!response.ok) return;
+        // First try the staged experiments API (preferred)
+        const stagedResponse = await fetch(`/api/v1/sessions/${sessionId}/experiments/staged`);
+        if (stagedResponse.ok) {
+          const stagedData = await stagedResponse.json();
+          if (stagedData.experiments && stagedData.experiments.length > 0) {
+            // API now returns clean experiments + reason separately
+            // Tag each experiment with the reason for the Add Point dialog
+            const reason = stagedData.reason || 'Staged';
+            const taggedExperiments = stagedData.experiments.map((exp: any) => ({
+              ...exp,
+              _reason: reason  // UI-only metadata for dialog auto-fill
+            }));
+            setPendingSuggestions(taggedExperiments);
+            console.log(`✓ Restored ${stagedData.experiments.length} staged experiments from API (reason: ${reason})`);
+            return;
+          }
+        }
+        
+        // Fallback: check audit log for backward compatibility
+        const auditResponse = await fetch(`/api/v1/sessions/${sessionId}/audit?entry_type=acquisition_locked`);
+        if (!auditResponse.ok) return;
         
-        const data = await response.json();
+        const data = await auditResponse.json();
         if (data.entries && data.entries.length > 0) {
           // Get latest acquisition entry
           const latestAcq = data.entries[data.entries.length - 1];
@@ -74,52 +95,53 @@ function AppContent() {
               Iteration: latestAcq.parameters?.iteration
             }));
             setPendingSuggestions(taggedSuggestions);
-            console.log(`✓ Restored ${suggestions.length} pending suggestions from audit log`);
+            console.log(`✓ Restored ${suggestions.length} pending suggestions from audit log (fallback)`);
           }
         }
       } catch (e) {
-        console.error('Failed to restore pending suggestions:', e);
+        console.error('Failed to restore staged experiments:', e);
       }
     }
     
-    restorePendingSuggestions();
+    restoreStagedExperiments();
   }, [sessionId]);
 
-  // Check for URL parameters (session ID and monitoring mode)
+  // Check URL parameters on mount (always takes priority over recovery)
   useEffect(() => {
     const urlParams = new URLSearchParams(window.location.search);
-    console.log('URL search params:', window.location.search);
-    
-    // Check for session ID in URL
     const urlSessionId = urlParams.get('session');
-    console.log('Session ID from URL:', urlSessionId);
-    
     if (urlSessionId) {
+      urlSessionRef.current = urlSessionId;
       setSessionId(urlSessionId);
-      setSessionFromUrl(true);  // Mark that this session came from URL
+      setSessionFromUrl(true);
       console.log(`✓ Loaded session from URL: ${urlSessionId}`);
-    } else {
-      // Fallback to localStorage if no URL session
-      const storedId = getStoredSessionId();
-      if (storedId) {
-        setSessionId(storedId);
-        console.log(`✓ Loaded session from localStorage: ${storedId}`);
-      }
     }
-    
-    // Check for monitoring mode
     const monitorParam = urlParams.get('mode');
-    console.log('Monitoring mode from URL:', monitorParam);
     if (monitorParam === 'monitor') {
       setIsMonitoringMode(true);
       console.log('✓ Monitoring mode enabled');
     }
   }, []);
 
+  // Check for recovery sessions on startup — skip if URL session is present
+  useEffect(() => {
+    if (urlSessionRef.current) return; // URL intent takes priority
+    fetch('/api/v1/recovery/list')
+      .then(res => res.json())
+      .then(data => {
+        if (data.recoveries && data.recoveries.length > 0) {
+          setRecoverySession(data.recoveries[0]);
+          setShowRecoveryBanner(true);
+          console.log('✓ Found recovery session:', data.recoveries[0]);
+        }
+      })
+      .catch(err => console.warn('Failed to check for recovery sessions:', err));
+  }, []);
+
   // Auto-clear invalid session (but not if it came from URL - let user see the error)
   useEffect(() => {
     if (sessionError && sessionId && !sessionFromUrl) {
-      toast.error('Session expired or not found. Please create a new session.');
+      toast.error('Session not found. Please create a new session.');
       handleClearSession();
     } else if (sessionError && sessionFromUrl) {
       // Show error but don't clear - might be loading or user wants to see the issue
@@ -127,24 +149,25 @@ function AppContent() {
     }
   }, [sessionError, sessionId, sessionFromUrl]);
 
-  // Prompt to save on page close/reload (desktop _quit behavior)
+  // Silent auto-backup every 30 seconds for crash recovery
   useEffect(() => {
     if (!sessionId) return;
     
-    const handleBeforeUnload = (e: BeforeUnloadEvent) => {
-      // Only prompt if session has data
-      if (session && (session.variable_count > 0 || session.experiment_count > 0)) {
-        e.preventDefault();
-        e.returnValue = 'You have unsaved changes. Are you sure you want to leave?';
-        return e.returnValue;
-      }
-    };
+    // Only backup if session has data
+    if (!session || (session.variable_count === 0 && session.experiment_count === 0)) {
+      return;
+    }
     
-    window.addEventListener('beforeunload', handleBeforeUnload);
-    return () => window.removeEventListener('beforeunload', handleBeforeUnload);
+    const interval = setInterval(() => {
+      fetch(`/api/v1/sessions/${sessionId}/backup`, { method: 'POST' })
+        .then(() => console.debug('Recovery backup saved'))
+        .catch(err => console.warn('Recovery backup failed:', err));
+    }, 30000); // Every 30 seconds
+    
+    return () => clearInterval(interval);
   }, [sessionId, session]);
 
-  // Prompt to save on page close/reload (desktop _quit behavior)
+  // Prompt to save on page close/reload if session has data
   useEffect(() => {
     if (!sessionId) return;
     
@@ -166,7 +189,6 @@ function AppContent() {
     try {
       const newSession = await createSession.mutateAsync({ ttl_hours: 24 });
       setSessionId(newSession.session_id);
-      setLoadedFromFile(false);
       toast.success('Session created successfully!');
       // Show metadata dialog for new session
       setShowMetadataDialog(true);
@@ -180,7 +202,6 @@ function AppContent() {
   const handleClearSession = () => {
     clearStoredSessionId();
     setSessionId(null);
-    setLoadedFromFile(false);
     toast.info('Session cleared');
   };
 
@@ -188,15 +209,16 @@ function AppContent() {
   const handleExportSession = async () => {
     if (!sessionId) return;
     try {
-      // If this session was loaded via the file upload dialog, persist server-side
-      if (loadedFromFile) {
-        await exportSession.mutateAsync({ sessionId, serverSide: true });
-      } else {
-        await exportSession.mutateAsync({ sessionId, serverSide: false });
-      }
-      toast.success('Session exported successfully!');
+      await exportSession.mutateAsync({ sessionId, serverSide: false });
+      
+      // Clear recovery backup after successful save
+      fetch(`/api/v1/sessions/${sessionId}/backup`, { method: 'DELETE' })
+        .then(() => console.log('Recovery backup cleared after save'))
+        .catch(err => console.warn('Failed to clear recovery backup:', err));
+      
+      toast.success('Session saved to Downloads folder!');
     } catch (error: any) {
-      toast.error('Failed to export session');
+      toast.error('Failed to save session');
       console.error('Error exporting session:', error);
     }
   };
@@ -212,8 +234,6 @@ function AppContent() {
       try {
         const newSession = await importSession.mutateAsync(file);
         setSessionId(newSession.session_id);
-        // Mark that this session was created from a file upload/import
-        setLoadedFromFile(true);
         toast.success('Session imported successfully!');
         // Reset file input
         if (fileInputRef.current) {
@@ -225,6 +245,46 @@ function AppContent() {
       }
     }
   };
+
+  // Handle recovery session restore
+  const handleRestoreRecovery = async () => {
+    if (!recoverySession) return;
+    try {
+      const response = await fetch(`/api/v1/recovery/${recoverySession.session_id}/restore`, {
+        method: 'POST'
+      });
+      if (!response.ok) throw new Error('Failed to restore recovery session');
+      
+      const data = await response.json();
+      setSessionId(data.session_id);
+      setShowRecoveryBanner(false);
+      setRecoverySession(null);
+      toast.success('Session restored from recovery backup!');
+    } catch (error: any) {
+      toast.error('Failed to restore recovery session');
+      console.error('Error restoring recovery:', error);
+    }
+  };
+
+  // Handle recovery dismissal
+  const handleDismissRecovery = async () => {
+    if (!recoverySession) return;
+    try {
+      // Delete the recovery backup
+      await fetch(`/api/v1/sessions/${recoverySession.session_id}/backup`, {
+        method: 'DELETE'
+      });
+      setShowRecoveryBanner(false);
+      setRecoverySession(null);
+      toast.info('Recovery session discarded');
+    } catch (error: any) {
+      console.error('Error dismissing recovery:', error);
+      // Still hide banner even if delete fails
+      setShowRecoveryBanner(false);
+      setRecoverySession(null);
+    }
+  };
+
   // Handle audit log export (matches desktop File → Export Audit Log)
   const handleExportAuditLog = async () => {
     if (!sessionId) return;
@@ -263,6 +323,15 @@ function AppContent() {
     }
   };
 
+  // Handle joining an existing session by ID
+  const handleJoinSession = () => {
+    const trimmed = joinSessionId.trim();
+    if (!trimmed) return;
+    setSessionId(trimmed);
+    setSessionFromUrl(true); // Prevent auto-clear on 404
+    setJoinSessionId('');
+  };
+
   // Debug: Log state before render
   console.log('=== Render State ===');
   console.log('sessionId:', sessionId);
@@ -271,6 +340,51 @@ function AppContent() {
 
   return (
     <div className="h-screen bg-background flex flex-col overflow-hidden">
+      {/* Recovery Banner */}
+      {showRecoveryBanner && recoverySession && (
+        <div className="bg-yellow-50 dark:bg-yellow-900/20 border-b border-yellow-200 dark:border-yellow-800">
+          <div className="max-w-7xl mx-auto px-6 py-4">
+            <div className="flex items-start justify-between">
+              <div className="flex-1">
+                <div className="flex items-center gap-2 mb-2">
+                  <svg className="w-5 h-5 text-yellow-600 dark:text-yellow-500" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                    <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 9v2m0 4h.01m-6.938 4h13.856c1.54 0 2.502-1.667 1.732-3L13.732 4c-.77-1.333-2.694-1.333-3.464 0L3.34 16c-.77 1.333.192 3 1.732 3z" />
+                  </svg>
+                  <h3 className="text-sm font-semibold text-yellow-900 dark:text-yellow-100">
+                    Unsaved work detected
+                  </h3>
+                </div>
+                <p className="text-sm text-yellow-800 dark:text-yellow-200 mb-2">
+                  We found a session from {new Date(recoverySession.backup_time).toLocaleString()} that wasn't saved.
+                  Would you like to restore it?
+                </p>
+                <div className="text-xs text-yellow-700 dark:text-yellow-300 space-y-1">
+                  <div><strong>Session:</strong> {recoverySession.session_name || 'Untitled Session'}</div>
+                  <div>
+                    <strong>Data:</strong> {recoverySession.n_variables} variables, {recoverySession.n_experiments} experiments
+                    {recoverySession.model_trained && ', trained model'}
+                  </div>
+                </div>
+              </div>
+              <div className="flex gap-2 ml-4">
+                <button
+                  onClick={handleRestoreRecovery}
+                  className="px-4 py-2 bg-yellow-600 hover:bg-yellow-700 text-white text-sm font-medium rounded-md transition-colors"
+                >
+                  Restore Session
+                </button>
+                <button
+                  onClick={handleDismissRecovery}
+                  className="px-4 py-2 bg-gray-200 hover:bg-gray-300 dark:bg-gray-700 dark:hover:bg-gray-600 text-gray-800 dark:text-gray-200 text-sm font-medium rounded-md transition-colors"
+                >
+                  Discard
+                </button>
+              </div>
+            </div>
+          </div>
+        </div>
+      )}
+
       {/* Monitoring Mode - Show dedicated dashboard */}
       {isMonitoringMode && sessionId ? (
         <MonitoringDashboard sessionId={sessionId} pollingInterval={90000} />
@@ -330,13 +444,13 @@ function AppContent() {
                     </div>
                     <button
                       onClick={handleCopySessionId}
-                      className="p-1.5 rounded hover:bg-accent transition-colors"
+                      className="p-1.5 rounded hover:bg-accent transition-colors border border-transparent hover:border-border"
                       title="Copy full session ID to clipboard"
                     >
                       {copiedSessionId ? (
-                        <Check className="h-3.5 w-3.5 text-green-500" />
+                        <Check className="h-4 w-4 text-green-500" />
                       ) : (
-                        <Copy className="h-3.5 w-3.5 text-muted-foreground hover:text-foreground" />
+                        <Copy className="h-4 w-4 text-muted-foreground hover:text-foreground" />
                       )}
                     </button>
                   </div>
@@ -391,6 +505,23 @@ function AppContent() {
                     onChange={handleFileSelected}
                     className="hidden"
                   />
+                  <div className="flex gap-2 items-center ml-4 pl-4 border-l border-border">
+                    <input
+                      type="text"
+                      value={joinSessionId}
+                      onChange={(e) => setJoinSessionId(e.target.value)}
+                      onKeyDown={(e) => e.key === 'Enter' && handleJoinSession()}
+                      placeholder="Paste session ID..."
+                      className="text-sm px-3 py-2 rounded border border-input bg-background w-64"
+                    />
+                    <button
+                      onClick={handleJoinSession}
+                      disabled={!joinSessionId.trim()}
+                      className="text-sm bg-accent text-accent-foreground px-4 py-2 rounded hover:bg-accent/80 disabled:opacity-50"
+                    >
+                      Join
+                    </button>
+                  </div>
                 </div>
               )}
             </div>
@@ -407,7 +538,10 @@ function AppContent() {
                   pendingSuggestions={pendingSuggestions}
                   onStageSuggestions={(s:any[])=>setPendingSuggestions(s)}
                 />
-                <InitialDesignPanel sessionId={sessionId} />
+                <InitialDesignPanel 
+                  sessionId={sessionId}
+                  onStageSuggestions={(s:any[])=>setPendingSuggestions(s)}
+                />
               </div>
 
               {/* CENTER - Visualization Area (expandable) */}

{alchemist_nrel-0.3.1 → alchemist_nrel-0.3.2}/alchemist-web/src/api/endpoints/experiments.ts

@@ -42,13 +42,51 @@ export const createExperimentBatch = async (
  */
 export const uploadExperimentsCSV = async (
   sessionId: string,
-  file: File
+  file: File,
+  targetColumns?: string | string[]
 ): Promise<{ message: string; n_experiments: number }> => {
   const formData = new FormData();
   formData.append('file', file);
   
+  // Build URL with target_columns as query parameter
+  const targetParam = targetColumns 
+    ? (Array.isArray(targetColumns) ? targetColumns.join(',') : targetColumns)
+    : undefined;
+  
+  const url = targetParam
+    ? `/sessions/${sessionId}/experiments/upload?target_columns=${encodeURIComponent(targetParam)}`
+    : `/sessions/${sessionId}/experiments/upload`;
+  
+  const response = await apiClient.post(
+    url,
+    formData,
+    {
+      headers: {
+        'Content-Type': 'multipart/form-data',
+      },
+    }
+  );
+  return response.data;
+};
+
+/**
+ * Preview CSV columns before upload to check for target columns
+ */
+export const previewCSVColumns = async (
+  sessionId: string,
+  file: File
+): Promise<{
+  columns: string[];
+  available_targets: string[];
+  has_output: boolean;
+  recommended_target: string | null;
+  n_rows: number;
+}> => {
+  const formData = new FormData();
+  formData.append('file', file);
+  
   const response = await apiClient.post(
-    `/sessions/${sessionId}/experiments/upload`,
+    `/sessions/${sessionId}/experiments/preview`,
     formData,
     {
       headers: {