Update app/api/routes.py

app/api/routes.py

@@ -119,20 +119,31 @@ async def transcribe_audio(
         background_tasks.add_task(cleanup_files, upload_path)
         
         # Build response
-        return TranscriptionResponse(
-            success=True,
-            segments=[
+        segments = []
+
+        for seg in result.segments:
+            if seg.role == "KH":
+                emotion = seg.emotion
+                icon = seg.icon
+            else:
+                emotion = None
+                icon = None
+
+            segments.append(
                 {
                     "start": seg.start,
                     "end": seg.end,
                     "speaker": seg.speaker,
                     "role": seg.role,
                     "text": seg.text,
-                    "emotion": seg.emotion if seg.role == "KH" else None,
-                    "icon": seg.icon if seg.icon == "KH" else None
+                    "emotion": emotion,
+                    "icon": icon,
                 }
-                for seg in result.segments
-            ],
+            )
+            
+        return TranscriptionResponse(
+            success=True,
+            segments=segments,
             speaker_count=result.speaker_count,
             speakers=result.speakers,
             duration=result.duration,