app.py

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'인 경우
        
    keywords = []
    
    for name in product_names:
        if pd.notna(name):  # NaN 값 체크
            words = name.split(" ")  # 공백 기준으로 단어 분리
            keywords.extend(words)
    
    # 키워드 빈도수 계산
    keyword_count = Counter(keywords)
    
    # 데이터프레임으로 변환
    result_df = pd.DataFrame(keyword_count.items(), columns=['키워드', '빈도수'])
    
    # 빈도수 기준으로 내림차순 정렬
    result_df = result_df.sort_values(by='빈도수', ascending=False).reset_index(drop=True)
    
    # 새로운 엑셀 파일 생성 및 이미지 삽입
    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