Update app.py

app.py

@@ -3,6 +3,8 @@ import pandas as pd
 import re
 from collections import Counter
 import os
+from openpyxl import load_workbook
+from openpyxl.drawing.image import Image
 
 def process_excel(file):
     # 엑셀 파일 읽기
@@ -39,6 +41,17 @@ def process_excel(file):
     with pd.ExcelWriter(output_file, engine='openpyxl') as writer:
         result_df.to_excel(writer, index=False, startrow=3)  # startrow=3으로 설정하여 4번째 행(A4, B4)부터 시작
     
+    # 이미지를 엑셀 파일의 A1 셀에 삽입
+    wb = load_workbook(output_file)
+    ws = wb.active
+    
+    # ssboost-logo.png 파일을 A1 셀에 삽입
+    img = Image("ssboost-logo.png")
+    ws.add_image(img, "A1")
+    
+    # 수정된 엑셀 파일 저장
+    wb.save(output_file)
+    
     return output_file
 
 # Gradio 인터페이스 정의
@@ -46,8 +59,8 @@ iface = gr.Interface(
     fn=process_excel, 
     inputs=gr.File(file_types=[".xlsx"]),  # 엑셀 파일만 업로드할 수 있게 설정
     outputs="file",
-    title="Excel Keyword Extractor",
-    description="엑셀 파일의 D열에서 키워드를 추출하고 빈도를 계산하여 새로운 엑셀 파일로 출력합니다."
+    title="Excel Keyword Extractor with Image",
+    description="엑셀 파일의 D열에서 키워드를 추출하고 빈도를 계산한 후, A1 셀에 이미지를 삽입하여 새로운 엑셀 파일로 출력합니다."
 )
 
 if __name__ == "__main__":