aliceblue11 commited on
Commit
3b1dcc0
·
verified ·
1 Parent(s): e9edbd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -29
app.py CHANGED
@@ -1,18 +1,9 @@
1
- import pandas as pd
2
- from collections import Counter
3
- import gradio as gr
4
- import openpyxl
5
- from openpyxl.drawing.image import Image # 이미지 삽입을 위한 모듈
6
-
7
  def extract_keywords(file):
8
- # 엑셀 파일 읽기
9
- df = pd.read_excel(file, engine='openpyxl')
10
 
11
- # 열 이름 확인
12
- print("엑셀 이름:", df.columns)
13
-
14
- # 상품명 열에서 D4부터 끝까지 텍스트 추출
15
- product_names = df['Product Name'][3:] # 올바른 열 이름으로 변경
16
 
17
  keywords = []
18
 
@@ -45,19 +36,3 @@ def extract_keywords(file):
45
  worksheet.add_image(img, "A1")
46
 
47
  return result_df, output_path
48
-
49
- def keyword_analysis_interface(file):
50
- result_df, output_path = extract_keywords(file)
51
- return result_df, output_path
52
-
53
- # Gradio 인터페이스 설정
54
- interface = gr.Interface(
55
- fn=keyword_analysis_interface,
56
- inputs=gr.File(label="엑셀 파일 업로드"),
57
- outputs=[gr.Dataframe(headers=["키워드", "빈도수"]), gr.File(label="결과 엑셀 파일 다운로드")],
58
- title="키워드 빈도수 분석기",
59
- description="업로드된 엑셀 파일에서 상품명에 대한 키워드를 분석하여 빈도수를 계산하고 엑셀 파일로 결과를 제공합니다."
60
- )
61
-
62
- if __name__ == "__main__":
63
- interface.launch()
 
 
 
 
 
 
 
1
  def extract_keywords(file):
2
+ # 엑셀 파일 읽기 (첫 번째 행을 무시)
3
+ df = pd.read_excel(file, header=None, engine='openpyxl') # header=None으로 열 이름 없이 로드
4
 
5
+ # 상품명 열을 직접 추출 (필요한 번호를 사용)
6
+ product_names = df[3][3:] # D4 셀부터 시작하는 데이터를 추출
 
 
 
7
 
8
  keywords = []
9
 
 
36
  worksheet.add_image(img, "A1")
37
 
38
  return result_df, output_path