Fix real-time progress tracking - clear stale progress, filter by request timestamp, improve status formatting

FRRONTEEEND/components/ChatInterface.tsx

@@ -75,10 +75,12 @@ export const ChatInterface: React.FC<{ onBack: () => void }> = ({ onBack }) => {
     updateSession(activeSessionId, newMessages);
     setInput('');
     setIsTyping(true);
+    setCurrentStep('');  // Clear previous step immediately when new request starts
     
-    // Start polling for progress updates
+    // Start polling for progress updates - use consistent session key
     const sessionKey = activeSessionId || 'default';
     let progressInterval: NodeJS.Timeout | null = null;
+    const requestStartTime = Date.now();
     
     const pollProgress = async () => {
       try {
@@ -88,14 +90,28 @@ export const ChatInterface: React.FC<{ onBack: () => void }> = ({ onBack }) => {
           const progressData = await progressResponse.json();
           const steps = progressData.steps || [];
           
-          // Find the most recent running step, or last completed step
-          const runningSteps = steps.filter((s: any) => s.status === 'running');
+          // Only show progress from this request (filter by timestamp)
+          const thisRequestSteps = steps.filter((s: any) => {
+            if (!s.timestamp) return true;
+            const stepTime = s.timestamp * 1000;  // Convert to ms
+            return stepTime >= requestStartTime - 1000;  // Allow 1s margin
+          });
+          
+          // Find the most recent running step
+          const runningSteps = thisRequestSteps.filter((s: any) => s.status === 'running');
           if (runningSteps.length > 0) {
             const lastStep = runningSteps[runningSteps.length - 1];
-            setCurrentStep(lastStep.tool);
-          } else if (steps.length > 0) {
+            // Format tool name nicely: "generate_interactive_scatter" -> "Generating Interactive Scatter"
+            const toolName = lastStep.tool
+              .replace(/^(generate|train|clean|encode|create|analyze)_/, '')
+              .replace(/_/g, ' ')
+              .split(' ')
+              .map((w: string) => w.charAt(0).toUpperCase() + w.slice(1))
+              .join(' ');
+            setCurrentStep(`${['Generating', 'Training', 'Cleaning', 'Encoding', 'Creating', 'Analyzing'][0]} ${toolName}`);
+          } else if (thisRequestSteps.length > 0) {
             // No running steps - check if there are completed steps
-            const completedSteps = steps.filter((s: any) => s.status === 'completed');
+            const completedSteps = thisRequestSteps.filter((s: any) => s.status === 'completed');
             if (completedSteps.length > 0) {
               // Only show if very recent (workflow still in progress)
               const lastCompleted = completedSteps[completedSteps.length - 1];

src/api/app.py

@@ -96,9 +96,15 @@ async def root():
 @app.get("/api/progress/{session_id}")
 async def get_progress(session_id: str):
     """Get progress updates for a specific session."""
+    steps = progress_store.get(session_id, [])
+    
+    # Filter out stale progress (older than 5 minutes)
+    current_time = time.time()
+    fresh_steps = [s for s in steps if current_time - s.get('timestamp', 0) < 300]
+    
     return {
         "session_id": session_id,
-        "steps": progress_store.get(session_id, [])
+        "steps": fresh_steps
     }