Update app.py

app.py

@@ -437,7 +437,7 @@ async def process_batch_dynamic(wav_files: List[str], start_batch_index: int, ba
             # Download the WAV file
             wav_path = await download_wav_file_by_index(file_index + 1, wav_file)
             if not wav_path:
-                state["file_states"][wav_filename] = "failed"
+                state["file_states"][wav_file] = "failed"
                 await upload_hf_state(state)
                 print(f"[{FLOW_ID}] ❌ Failed to download {wav_filename}")
                 continue
@@ -445,7 +445,7 @@ async def process_batch_dynamic(wav_files: List[str], start_batch_index: int, ba
             # Assign to server and create task
             await assign_file_to_server(file_index, server)
             task = asyncio.create_task(send_audio_to_whisper(wav_path, server))
-            pending_tasks[task] = (file_index, wav_path, server, wav_filename)
+            pending_tasks[task] = (file_index, wav_path, server, wav_file)
             print(f"[{FLOW_ID}] Assigned {wav_filename} to server {servers.index(server) + 1}")
         
         # --- Wait for all tasks in this batch to complete and process results
@@ -456,31 +456,32 @@ async def process_batch_dynamic(wav_files: List[str], start_batch_index: int, ba
             )
             
             for task in done:
-                file_index, wav_path, server, wav_filename = pending_tasks.pop(task)
+                file_index, wav_path, server, wav_file = pending_tasks.pop(task)
+                wav_filename = Path(wav_file).name
                 
                 try:
                     transcription_result = task.result()
                     
                     if transcription_result:
-                        # Upload transcription immediately
-                        uploaded_ok = await upload_transcription_to_hf(wav_filename, transcription_result)
+                        # Upload transcription immediately with full path
+                        uploaded_ok = await upload_transcription_to_hf(wav_file, transcription_result)
                         if uploaded_ok:
                             # Update state locally but do NOT upload to HF yet
-                            state["file_states"][wav_filename] = "processed"
+                            state["file_states"][wav_file] = "processed"
                             uploaded_count += 1
                             progress['uploaded_count'] = uploaded_count
                             save_progress(progress)
                             print(f"[{FLOW_ID}] ✅ {wav_filename} uploaded (#{uploaded_count})")
                         else:
-                            state["file_states"][wav_filename] = "failed"
+                            state["file_states"][wav_file] = "failed"
                             print(f"[{FLOW_ID}] ❌ Failed to upload {wav_filename}")
                     else:
-                        state["file_states"][wav_filename] = "failed"
+                        state["file_states"][wav_file] = "failed"
                         print(f"[{FLOW_ID}] ❌ Transcription failed for {wav_filename}")
                         
                 except Exception as e:
                     print(f"[{FLOW_ID}] Error processing result for {wav_filename}: {e}")
-                    state["file_states"][wav_filename] = "failed"
+                    state["file_states"][wav_file] = "failed"
                 finally:
                     # Release the server
                     await release_server(server)