app.py

import json
import gradio as gr
import spaces
import wbgtopic
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 [], None
    
    topics = results[0]
    top_topics = sorted(topics, key=lambda x: x['score_mean'], reverse=True)[:5]
    
    formatted_topics = []
    for topic in top_topics:
        formatted_topic = {
            'label': translate_topics.get(topic['label'], topic['label']),
            'score': round(topic['score_mean'] * 100, 1),
            'confidence': round((1 - topic['score_std']) * 100, 1)
        }
        formatted_topics.append(formatted_topic)
    
    # 차트 생성
    chart = create_chart(formatted_topics)
    
    return formatted_topics, chart

@spaces.GPU(enable_queue=True, duration=50)
def fn(inputs):
    raw_results = clf.suggest_topics(inputs)
    return process_results(raw_results)

# Gradio 인터페이스 생성
with gr.Blocks(title="문서 주제 분석기") as demo:
    gr.Markdown("## 📚 문서 주제 분석기")
    gr.Markdown("문서를 입력하면 관련된 주제들을 분석하여 보여줍니다.")
    
    with gr.Row():
        text = gr.Textbox(
            value=SAMPLE_TEXT,
            label="분석할 텍스트",
            placeholder="여기에 분석할 텍스트를 입력하세요",
            lines=5
        )
    
    with gr.Row():
        submit_btn = gr.Button("분석 시작", variant="primary")
    
    with gr.Row():
        # 차트를 보여줄 Plot 컴포넌트 추가
        plot = gr.Plot(label="주제 분석 차트")
        
    with gr.Row():
        output = gr.JSON(label="상세 분석 결과")
    
    # 이벤트 연결
    submit_btn.click(
        fn=fn,
        inputs=[text],
        outputs=[output, plot]
    )

demo.launch(debug=True)