Update app.py

app.py

@@ -2,24 +2,49 @@ import json
 import gradio as gr
 import spaces
 import wbgtopic
-from topic_translator import translate_topics  # 새로 만들 파일입니다
+import plotly.graph_objects as go
+from topic_translator import translate_topics
 
 SAMPLE_TEXT = """A growing literature attributes gender inequality in labor market outcomes in part to the reduction in female labor supply after childbirth, the child penalty..."""
 
 clf = wbgtopic.WBGDocTopic()
 
+def create_chart(topics):
+    # 데이터 준비
+    labels = [t['label'] for t in topics]
+    scores = [t['score'] for t in topics]
+    confidence = [t['confidence'] for t in topics]
+    
+    # 막대 차트 생성
+    fig = go.Figure()
+    
+    # 주요 막대 추가
+    fig.add_trace(go.Bar(
+        x=labels,
+        y=scores,
+        name='관련도',
+        marker_color='rgb(55, 83, 109)'
+    ))
+    
+    # 차트 레이아웃 설정
+    fig.update_layout(
+        title='문서 주제 분석 결과',
+        xaxis_title='주제',
+        yaxis_title='관련도 (%)',
+        yaxis_range=[0, 100],
+        height=500,
+        font=dict(size=14)
+    )
+    
+    return fig
+
 def process_results(results):
-    # 결과를 보기 좋게 가공
     if not results or not results[0]:
-        return []
+        return [], None
     
-    # 첫 번째 문서의 결과만 사용 (단일 문서 기준)
     topics = results[0]
-    
-    # 상위 5개 주제만 선택
     top_topics = sorted(topics, key=lambda x: x['score_mean'], reverse=True)[:5]
     
-    # 각 주제의 점수를 퍼센트로 변환하고 한글로 번역
     formatted_topics = []
     for topic in top_topics:
         formatted_topic = {
@@ -29,7 +54,10 @@ def process_results(results):
         }
         formatted_topics.append(formatted_topic)
     
-    return formatted_topics
+    # 차트 생성
+    chart = create_chart(formatted_topics)
+    
+    return formatted_topics, chart
 
 @spaces.GPU(enable_queue=True, duration=50)
 def fn(inputs):
@@ -53,13 +81,17 @@ with gr.Blocks(title="문서 주제 분석기") as demo:
         submit_btn = gr.Button("분석 시작", variant="primary")
     
     with gr.Row():
-        output = gr.JSON(label="분석 결과")
+        # 차트를 보여줄 Plot 컴포넌트 추가
+        plot = gr.Plot(label="주제 분석 차트")
+        
+    with gr.Row():
+        output = gr.JSON(label="상세 분석 결과")
     
     # 이벤트 연결
     submit_btn.click(
         fn=fn,
         inputs=[text],
-        outputs=output
+        outputs=[output, plot]
     )
 
 demo.launch(debug=True)