Update emotion_predictor.py

emotion_predictor.py

@@ -1,5 +1,6 @@
 import re
 import math
+import json
 import matplotlib.pyplot as plt
 import matplotlib.font_manager as fm
 import torch
@@ -10,6 +11,7 @@ from sklearn.linear_model import LinearRegression
 from collections import defaultdict
 import base64
 from io import BytesIO
+from IPython.display import HTML, display  # Jupyter/Colab HTML 표시용
 
 # ======================
 # ⚙️ 폰트 설정
@@ -159,7 +161,19 @@ def predict_and_plot(raw_text):
         else:
             html_output += "<p>⚠️ 시각화할 수 있는 감정이 없습니다.</p><hr/>"
 
-    return {
+    # === 결과 정리 ===
+    result = {
         "html": html_output,
         "utterance_emotion_log": utterance_emotion_log
-    }
+    }
+
+    # HTML 시각화 환경 (예: Jupyter, Colab)에서는 바로 렌더링
+    try:
+        display(HTML(result["html"]))
+    except:
+        pass  # 웹서버나 비렌더 환경에서는 무시
+
+    # 구조화된 감정 로그를 보기 좋게 출력 (or API에서는 JSON 직렬화 가능)
+    print(json.dumps(result["utterance_emotion_log"], ensure_ascii=False, indent=2))
+
+    return result