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

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

def analyze_reviews(file):
    try:
        logging.info("파일 업로드 시작")
        # 엑셀 파일 읽기
        df = pd.read_excel(file, engine='openpyxl')
        logging.info("엑셀 파일 읽기 완료")

        # 현재 연도 설정 (2025년 기준)
        current_year = 2025
        start_year = current_year - 3  # 2022년부터

        logging.info("리뷰 날짜 데이터 처리 시작")
        # B열에 있는 리뷰 날짜를 datetime 형식으로 변환
        df['B'] = pd.to_datetime(df.iloc[:, 1], errors='coerce')
        # 최근 3년 데이터 필터링
        df_filtered = df[(df['B'].dt.year >= start_year) & (df['B'].dt.year < current_year)]
        logging.info(f"필터링된 데이터 개수: {len(df_filtered)}")

        # 년월별 리뷰 건수 계산
        df_filtered['Year-Month'] = df_filtered['B'].dt.strftime('%Y-%m')
        review_counts = df_filtered.groupby('Year-Month').size().reset_index(name='Review Count')
        logging.info("월별 리뷰 건수 계산 완료")

        # 새로운 시트에 작성
        with pd.ExcelWriter(file, engine='openpyxl', mode='a') as writer:
            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
        logging.info("새로운 시트 '월별 리뷰건수' 추가 완료")

        # 결과 파일을 바이트 형태로 변환
        output = BytesIO()
        with pd.ExcelWriter(output, engine='openpyxl') as writer:
            df.to_excel(writer, index=False)
            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
        output.seek(0)
        logging.info("분석된 엑셀 파일 생성 완료")

        return output

    except Exception as e:
        logging.error(f"에러 발생: {e}")
        return None

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

# 애플리케이션 실행
if __name__ == "__main__":
    demo.launch()