UI 텍스트 수정 (숫자 -> 단어)

app.py

@@ -1,26 +1,43 @@
-# -*- coding: utf-8 -*-
-"""Untitled0.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1wnyeCNxzRVxoae3tCcMuf3s9Adx503U7
-"""
-
 # app.py
 import gradio as gr
 from news_analyzer import run_once, title_attention_index
+import math  # math 모듈 추가
+
+# [NEW] ✨ 점수를 텍스트로 변환하는 헬퍼 함수
+# (news_analyzer.py에 있던 함수를 app.py로 가져옵니다)
+def convert_sim_to_text(score: float, threshold: float = 0.5) -> str:
+    """
+    유사도 점수를 기준으로 텍스트 등급을 반환합니다.
+    threshold (기본값 0.5) 이상이면 '유사', 미만이면 '유사하지 않음'으로 판단합니다.
+    """
+    if score is None or (isinstance(score, float) and math.isnan(score)):
+        return "판단 불가"  # nan 값 처리
+
+    if score >= threshold + 0.15: # 예: 0.65 이상
+        return "높음 (제목이 본문을 잘 반영합니다)"
+    elif score >= threshold: # 예: 0.5 ~ 0.65
+        return "보통 (제목이 본문과 어느 정도 일치합니다)"
+    else: # 예: 0.5 미만
+        return "낮음 (제목과 본문 내용에 차이가 있습니다)"
+
 
 def predict(title, body):
     r = run_once(title, body)
     final_score = r["최종 기사 점수"]
     grade = title_attention_index(final_score)
+
+    # [NEW] 텍스트 변환 로직을 predict 함수에 직접 추가
+    summary_sim_text = convert_sim_to_text(r["요약유사도"], threshold=0.45)
+    body_sim_text = convert_sim_to_text(r["본문 일치도(Top5 평균)"], threshold=0.5)
+
+    # [MODIFIED] 💡 return 순서 변경
+    # 'grade' (제목 주의 지수)를 가장 첫 번째로 보냅니다.
     return (
-        r["요약유사도"],
-        r["본문 일치도(Top5 평균)"],
-        r["과장점수"],
-        final_score,
-        grade,
+        grade,                 # <-- [0]
+        summary_sim_text,      # <-- [1]
+        body_sim_text,         # <-- [2]
+        r["과장점수"],           # <-- [3]
+        final_score,           # <-- [4]
     )
 
 demo = gr.Interface(
@@ -29,12 +46,14 @@ demo = gr.Interface(
         gr.Textbox(label="제목", lines=2),
         gr.Textbox(label="본문", lines=18, placeholder="여기에 기사 본문을 붙여넣으세요"),
     ],
+    # [MODIFIED] 💡 outputs 리스트 순서 변경
+    # '제목 주의 지수'를 가장 위로 올리고, 텍스트 출력을 위해 Textbox로 변경
     outputs=[
-        gr.Number(label="요약유사도"),
-        gr.Number(label="본문 일치도(Top5 평균)"),
-        gr.Number(label="과장점수"),
-        gr.Number(label="최종 기사 점수"),
-        gr.Textbox(label="제목 주의 지수", interactive=False),
+        gr.Textbox(label="제목 주의 지수", interactive=False), # <-- [0]
+        gr.Textbox(label="요약유사도"),                      # <-- [1]
+        gr.Textbox(label="본문 일치도(Top5 평균)"),          # <-- [2]
+        gr.Number(label="과장점수"),                      # <-- [3]
+        gr.Number(label="최종 기사 점수"),                # <-- [4]
     ],
     title="제목 주의 지수",
     description=(
@@ -45,4 +64,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()  # Spaces에서는 이 라인이 없어도 자동 실행됩니다.
+    demo.launch()