app.py

import gradio as gr
import pandas as pd
import logging
from datetime import datetime

# 로깅 설정
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

def analyze_reviews(file):
    try:
        logging.info("파일 업로드 시작.")
        # 엑셀 파일 읽기
        df = pd.read_excel(file.name, engine='openpyxl')
        logging.info("엑셀 파일 읽기 완료.")
        
        # 현재 연도 설정 (2025년 기준)
        current_year = 2025
        start_date = datetime(current_year - 3, 1, 1)
        end_date = datetime(current_year, 12, 31)
        
        logging.info(f"데이터 필터링: {start_date}부터 {end_date}까지.")
        # B열을 날짜로 변환
        df['B'] = pd.to_datetime(df.iloc[:, 1])
        
        # 최근 3년 데이터 필터링
        mask = (df['B'] >= start_date) & (df['B'] <= end_date)
        filtered_df = df.loc[mask]
        logging.info(f"필터링된 데이터 행 수: {len(filtered_df)}.")
        
        # 년월별 리뷰 건수 집계
        filtered_df['Year-Month'] = filtered_df['B'].dt.strftime('%Y-%m')
        review_counts = filtered_df.groupby('Year-Month').size().reset_index(name='리뷰건수')
        logging.info("년월별 리뷰 건수 집계 완료.")
        
        # 새로운 시트에 작성
        with pd.ExcelWriter(file.name, engine='openpyxl', mode='a') as writer:
            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['A열', 'B열'])
        logging.info("새로운 시트 '월별 리뷰건수' 작성 완료.")
        
        return file.name
    except Exception as e:
        logging.error(f"오류 발생: {e}")
        return None

# Gradio 인터페이스 구성
with gr.Blocks() as demo:
    gr.Markdown("# 리뷰 분석 스페이스")
    with gr.Row():
        file_input = gr.File(label="원본 엑셀 파일 업로드")
    analyze_button = gr.Button("분석")
    file_output = gr.File(label="분석된 엑셀 파일 다운로드", type="filepath")
    
    analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=file_output)

# 스페이스 실행
if __name__ == "__main__":
    demo.launch()