Update app.py

app.py

@@ -210,35 +210,29 @@ def analyze_audio(audio_file: str,
                 diarize_output = None
         else:
             warn(results, "DIAR_SKIP", "HF_TOKEN not set. Skipping speaker diarization.")
-        print("Assigning speakers to words...")
-        
+        print("Assigning speakers to words...")                  
         try:
             diarize_segments_for_assignment = []
-            if diarize_output is not None:
-                if hasattr(diarize_output, "itertracks"):
-                    for segment, _, label in diarize_output.itertracks(yield_label=True):
-                        diarize_segments_for_assignment.append({
-                            "start": float(segment.start),
-                            "end": float(segment.end),
-                            "speaker": normalize_speaker(label)
-                        })
-                else:
-                    diarize_segments_for_assignment = diarize_output
-            else:
-                diarize_segments_for_assignment = []
-                for seg in aligned.get("segments", []):
+            if diarize_output is not None and hasattr(diarize_output, "itertracks"):
+                for segment, _, label in diarize_output.itertracks(yield_label=True):
                     diarize_segments_for_assignment.append({
-                        "start": seg.get("start", 0),
-                        "end": seg.get("end", seg.get("start", 0)),
-                        "speaker": "Speaker_1"
+                        "start": float(segment.start),
+                        "end": float(segment.end),
+                        "speaker": normalize_speaker(label)
                     })
-            
+                print(f"DEBUG: Converted {len(diarize_segments_for_assignment)} diarization segments.")
             if diarize_segments_for_assignment:
                 final = whisperx.assign_word_speakers(diarize_segments_for_assignment, aligned)
             else:
+                warn(results, "ASSIGN_FAIL", "Diarization segments were empty or unavailable. Defaulting all to Speaker_1.")
                 final = aligned
                 for seg in final.get("segments", []):
-                    seg["speaker"] = "Speaker_1"
+                    seg["speaker"] = "Speaker_1"          
+        except Exception as e:
+            warn(results, "ASSIGN_SPEAKERS_ERROR", f"Error assigning speakers: {e}. Falling back to unassigned segments.")
+            final = aligned
+            for seg in final.get("segments", []):
+                seg["speaker"] = "Speaker_1"
         except Exception as e:
             warn(results, "ASSIGN_SPEAKERS_ERROR", f"Error assigning speakers: {e}. Falling back to unassigned segments.")
             final = aligned