Update app.py

app.py

@@ -2,6 +2,7 @@ import pandas as pd
 from collections import Counter
 import gradio as gr
 import openpyxl
+from openpyxl.drawing.image import Image  # 이미지 삽입을 위한 모듈
 
 def extract_keywords(file):
     # 엑셀 파일 읽기
@@ -26,9 +27,19 @@ def extract_keywords(file):
     # 빈도수 기준으로 내림차순 정렬
     result_df = result_df.sort_values(by='빈도수', ascending=False).reset_index(drop=True)
     
-    # 결과를 새로운 엑셀 파일로 저장
-    output_path = "/mnt/data/키워드_분석_결과.xlsx"
-    result_df.to_excel(output_path, index=False)
+    # 새로운 엑셀 파일 생성 및 이미지 삽입
+    output_path = "/mnt/data/키워드_분석_결과_with_logo.xlsx"
+    with pd.ExcelWriter(output_path, engine='openpyxl') as writer:
+        result_df.to_excel(writer, index=False)
+        workbook = writer.book
+        worksheet = workbook.active
+        
+        # 이미지 파일 불러오기
+        logo_path = "/path/to/ssboost-logo.png"  # ssboost-logo.png 경로
+        img = Image(logo_path)
+        
+        # A1 셀에 이미지 삽입
+        worksheet.add_image(img, "A1")
     
     return result_df, output_path
 
@@ -42,7 +53,7 @@ interface = gr.Interface(
     inputs=gr.File(label="엑셀 파일 업로드"), 
     outputs=[gr.Dataframe(headers=["키워드", "빈도수"]), gr.File(label="결과 엑셀 파일 다운로드")],
     title="키워드 빈도수 분석기",
-    description="업로드된 엑셀 파일에서 상품명에 대한 키워드를 분석하여 빈도수를 계산합니다."
+    description="업로드된 엑셀 파일에서 상품명에 대한 키워드를 분석하여 빈도수를 계산하고 엑셀 파일로 결과를 제공합니다."
 )
 
 if __name__ == "__main__":