Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import random
 import os
+import pandas as pd  # 엑셀 파일 처리를 위한 pandas 라이브러리 추가
 from huggingface_hub import InferenceClient
 
 MODELS = {
@@ -28,6 +29,14 @@ def call_api(model, content, system_message, max_tokens, temperature, top_p):
 def generate_text(model, user_message, system_message, max_tokens, temperature, top_p):
     return call_api(model, user_message, system_message, max_tokens, temperature, top_p)
 
+# 엑셀 파일을 처리하는 함수 추가
+def process_excel(file):
+    if file is not None:
+        df = pd.read_excel(file.name)  # 엑셀 파일을 읽음
+        return df.head().to_string()  # 엑셀 파일의 처음 몇 줄을 출력
+    else:
+        return "엑셀 파일이 업로드되지 않았습니다."
+
 title = "AI 텍스트 생성기"
 
 with gr.Blocks() as demo:
@@ -42,11 +51,18 @@ with gr.Blocks() as demo:
         temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.75, step=0.05)
         top_p = gr.Slider(label="Top P", minimum=0.1, maximum=1.0, value=0.95, step=0.05)
     
+    # 엑셀 파일 업로드 컴포넌트 추가
+    excel_file = gr.File(label="엑셀 파일 업로드", type="file")
+    excel_output = gr.Textbox(label="엑셀 내용", lines=10)
+
     generate_btn = gr.Button("텍스트 생성하기")
     output = gr.Textbox(label="생성된 텍스트", lines=10)
     
     generate_btn.click(fn=generate_text, 
                        inputs=[model, user_message, system_message, max_tokens, temperature, top_p], 
                        outputs=[output])
+    
+    # 엑셀 파일 업로드 버튼과 처리 함수 연결
+    excel_file.upload(fn=process_excel, inputs=[excel_file], outputs=[excel_output])
 
-demo.launch()
+demo.launch()