Update app.py

app.py

@@ -1,14 +1,19 @@
+import pandas as pd
+from collections import Counter
+import gradio as gr
+import openpyxl
+from openpyxl.drawing.image import Image  # 이미지 삽입을 위한 모듈
+
 def extract_keywords(file):
     # 엑셀 파일 읽기
     df = pd.read_excel(file, engine='openpyxl')
     
-    # 엑셀 열 이름 확인
+    # 열 이름 확인
     print("엑셀 열 이름:", df.columns)
     
-    # 상품명 열이 없는 경우 처리 (예: 첫 번째 행을 열 이름으로 사용할 때 발생)
-    if "Unnamed: 3" in df.columns:
-        product_names = df["Unnamed: 3"][3:]  # 열 이름이 'Unnamed: 3'인 경우
-        
+    # 상품명 열에서 D4부터 끝까지 텍스트 추출
+    product_names = df['Product Name'][3:]  # 올바른 열 이름으로 변경
+    
     keywords = []
     
     for name in product_names:
@@ -40,3 +45,19 @@ def extract_keywords(file):
         worksheet.add_image(img, "A1")
     
     return result_df, output_path
+
+def keyword_analysis_interface(file):
+    result_df, output_path = extract_keywords(file)
+    return result_df, output_path
+
+# Gradio 인터페이스 설정
+interface = gr.Interface(
+    fn=keyword_analysis_interface, 
+    inputs=gr.File(label="엑셀 파일 업로드"), 
+    outputs=[gr.Dataframe(headers=["키워드", "빈도수"]), gr.File(label="결과 엑셀 파일 다운로드")],
+    title="키워드 빈도수 분석기",
+    description="업로드된 엑셀 파일에서 상품명에 대한 키워드를 분석하여 빈도수를 계산하고 엑셀 파일로 결과를 제공합니다."
+)
+
+if __name__ == "__main__":
+    interface.launch()