fix processor & UI

app/services/processor.py

@@ -195,21 +195,28 @@ class Processor:
 
         speakers = list(speaker_map.values())
 
-
-        # 6. Infer role ONLY if diarization did not provide
+        # 6. NORMALIZE ROLES
         speaker_duration = defaultdict(float)
         for seg in refined_segments:
-            speaker_duration[speaker_map[seg.speaker]] += seg.end - seg.start
+            speaker_duration[seg.speaker] += seg.end - seg.start
 
+        logger.info(f"speaker_duration(raw) = {speaker_duration}")
 
         if speaker_duration:
-            agent = max(speaker_duration, key=speaker_duration.get)
+            agent_raw = max(speaker_duration, key=speaker_duration.get)
+
             roles = {
-                speaker: ("NV" if speaker == agent else "KH")
-                for speaker in speaker_duration
+                speaker_map[spk]: ("NV" if spk == agent_raw else "KH")
+                for spk in speaker_duration
             }
         else:
-            roles = {} 
+            roles = {}
+
+        # Default fallback
+        for label in speakers:
+            roles.setdefault(label, "KH")
+
+        logger.info(f"roles(mapped) = {roles}")
 
 
             
@@ -261,7 +268,7 @@ class Processor:
                     start=seg.start,
                     end=seg.end,
                     speaker=label,
-                    role=roles.get(label, "KH"),
+                    role=roles[label],
                     text=text.strip(),
                 )
             )
@@ -272,28 +279,26 @@ class Processor:
                     start=0.0,
                     end=duration,
                     speaker=speakers[0],
-                    role=roles.get(speakers[0], "UNKNOWN"),
+                    role=roles[speakers[0]],
                     text="(No speech detected)"
                 )
             ]
 
         processing_time = time.time() - t0
-        speaker_count=len(speakers)
         
         txt_content = cls._generate_txt(
             processed_segments,
-            speaker_count,
+            len(speakers),
             processing_time,
             duration,
             roles
         )
 
-
         csv_content = cls._generate_csv(processed_segments)
 
         return ProcessingResult(
             segments=processed_segments,
-            speaker_count=speaker_count,
+            speaker_count=len(speakers),
             duration=duration,
             processing_time=processing_time,
             speakers=speakers,

app/static/js/app.js

@@ -207,7 +207,8 @@ document.addEventListener('DOMContentLoaded', () => {
 
     function displayResults(result) {
         // Update metadata
-        elements.speakerCount.textContent = `${result.num_speakers} speaker${result.num_speakers !== 1 ? 's' : ''}`;
+        elements.speakerCount.textContent =`${result.speaker_count} speaker${result.speaker_count !== 1 ? 's' : ''}`;
+
         elements.durationInfo.textContent = formatDuration(result.duration);
         elements.processingTime.textContent = `${result.processing_time}s`;
 
@@ -229,19 +230,27 @@ document.addEventListener('DOMContentLoaded', () => {
         let colorIndex = 0;
 
         segments.forEach((segment) => {
-            // Assign color to speaker
-            if (!(segment.speaker in speakerColors)) {
+
+            const displayName = segment.role
+                ? segment.role
+                : segment.speaker;
+
+            if (!(displayName in speakerColors)) {
                 colorIndex++;
-                speakerColors[segment.speaker] = `speaker-${Math.min(colorIndex, 5)}`;
+                speakerColors[displayName] = `speaker-${Math.min(colorIndex, 5)}`;
             }
 
             const segmentEl = document.createElement('div');
-            segmentEl.className = `segment ${speakerColors[segment.speaker]}`;
+            segmentEl.className = `segment ${speakerColors[displayName]}`;
 
             segmentEl.innerHTML = `
                 <div class="segment-header">
-                    <span class="segment-speaker">${escapeHtml(segment.speaker)}</span>
-                    <span class="segment-time">${formatTime(segment.start)} - ${formatTime(segment.end)}</span>
+                    <span class="segment-speaker">
+                        ${escapeHtml(displayName)}
+                    </span>
+                    <span class="segment-time">
+                        ${formatTime(segment.start)} - ${formatTime(segment.end)}
+                    </span>
                 </div>
                 <p class="segment-text">${escapeHtml(segment.text)}</p>
             `;
@@ -250,6 +259,7 @@ document.addEventListener('DOMContentLoaded', () => {
         });
     }
 
+
     function formatTime(seconds) {
         const h = Math.floor(seconds / 3600);
         const m = Math.floor((seconds % 3600) / 60);