Update app.py

app.py

@@ -465,6 +465,7 @@ async def process_batch_dynamic(wav_files: List[str], start_batch_index: int, ba
                         # Upload transcription immediately
                         uploaded_ok = await upload_transcription_to_hf(wav_filename, transcription_result)
                         if uploaded_ok:
+                            # Update state locally but do NOT upload to HF yet
                             state["file_states"][wav_filename] = "processed"
                             uploaded_count += 1
                             progress['uploaded_count'] = uploaded_count
@@ -476,20 +477,22 @@ async def process_batch_dynamic(wav_files: List[str], start_batch_index: int, ba
                     else:
                         state["file_states"][wav_filename] = "failed"
                         print(f"[{FLOW_ID}] ❌ Transcription failed for {wav_filename}")
-                    
-                    # Persist state change for this file to HF
-                    await upload_hf_state(state)
                         
                 except Exception as e:
                     print(f"[{FLOW_ID}] Error processing result for {wav_filename}: {e}")
                     state["file_states"][wav_filename] = "failed"
-                    await upload_hf_state(state)
                 finally:
                     # Release the server
                     await release_server(server)
                     # Clean up the WAV file
                     if wav_path.exists():
                         wav_path.unlink()
+        
+        # --- After all files in this batch are uploaded, update HF state once
+        if await upload_hf_state(state):
+            print(f"[{FLOW_ID}] ✅ Batch state updated on HF: files {start_batch_index}-{batch_end - 1} marked processed")
+        else:
+            print(f"[{FLOW_ID}] ❌ Failed to update batch state on HF")
     except Exception as e:
         print(f"[{FLOW_ID}] Error in process_batch_dynamic: {e}")