Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import pandas as pd
+from transformers import pipeline
 from huggingface_hub import InferenceClient
 import os
 import random
@@ -78,6 +80,31 @@ def generate_response(prompt, reference_text, max_tokens, temperature, top_p, mo
     """
     return response_html
 
+# 리뷰 파일 처리 함수
+def process_reviews(file):
+    df = pd.read_excel(file.name)
+    
+    if 'review' not in df.columns:
+        return "리뷰 파일에 'review' 열이 없습니다. 올바른 파일을 업로드하세요."
+
+    sentiment_analyzer = pipeline("sentiment-analysis")
+    reviews = df['review'].tolist()
+
+    # 감성 분석 수행
+    sentiments = sentiment_analyzer(reviews)
+
+    # 긍정 및 부정 리뷰 필터링
+    positive_reviews = [r['review'] for r, s in zip(reviews, sentiments) if s['label'] == 'POSITIVE'][:10]
+    negative_reviews = [r['review'] for r, s in zip(reviews, sentiments) if s['label'] == 'NEGATIVE'][:10]
+
+    # 분석 결과 요약
+    total_reviews = len(reviews)
+    positive_count = len([s for s in sentiments if s['label'] == 'POSITIVE'])
+    negative_count = len([s for s in sentiments if s['label'] == 'NEGATIVE'])
+    analysis_summary = f"총 리뷰 수: {total_reviews}, 긍정 리뷰 수: {positive_count}, 부정 리뷰 수: {negative_count}"
+
+    return "\n".join(positive_reviews), "\n".join(negative_reviews), analysis_summary
+
 # Gradio 인터페이스 설정
 with gr.Blocks() as demo:
     gr.Markdown("## 언어 모델 프롬프트 플레이그라운드")
@@ -95,6 +122,19 @@ with gr.Blocks() as demo:
         generate_button = gr.Button("응답 생성")
         response_output = gr.HTML(label="생성된 응답")
 
+    # 리뷰 파일 업로드 메뉴 추가
+    file_input = gr.File(label="리뷰 엑셀 파일 업로드")
+    positive_reviews_output = gr.Textbox(label="대표 긍정 리뷰 10개", lines=10, interactive=False)
+    negative_reviews_output = gr.Textbox(label="대표 부정 리뷰 10개", lines=10, interactive=False)
+    analysis_output = gr.Textbox(label="분석 결과", interactive=False)
+
+    # 리뷰 파일 처리 버튼
+    file_input.change(
+        process_reviews,
+        inputs=file_input,
+        outputs=[positive_reviews_output, negative_reviews_output, analysis_output]
+    )
+
     # 버튼 클릭 시 응답 생성
     generate_button.click(
         generate_response,