Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 import pandas as pd
 from datetime import datetime
@@ -15,32 +14,50 @@ def analyze_reviews(file_path, selected_year):
         df = pd.read_excel(file_path)
         logger.info("엑셀 파일 읽기 완료")
 
-        # 선택된 연도
+        # 현재 연도
         current_year = 2025
-        selected_year = int(selected_year)
+        start_year = current_year - 3  # 최근 3년
 
-        logger.info(f"데이터 필터링: {selected_year}년")
-        # B열이 리뷰 날짜라고 가정하고, '리뷰날짜' 열의 이름을 확인 및 변경
+        logger.info(f"데이터 필터링: {start_year}년부터 {current_year}년까지")
+        # B열이 리뷰 날짜라고 가정하고, 'B' 열의 이름을 '리뷰날짜'로 변경
         if '리뷰날짜' not in df.columns:
             df.rename(columns={df.columns[1]: '리뷰날짜'}, inplace=True)
         
         # 리뷰 날짜를 datetime으로 변환
         df['리뷰날짜'] = pd.to_datetime(df['리뷰날짜'], errors='coerce')
-        # 선택된 연도 데이터 필터링
-        df_selected = df[df['리뷰날짜'].dt.year == selected_year]
-        logger.info(f"{selected_year}년 데이터 필터링 완료")
+        # 최근 3년 데이터 필터링
+        df_recent = df[df['리뷰날짜'].dt.year >= start_year]
+        logger.info("최근 3년 데이터 필터링 완료")
 
         # 년월별 리뷰 건수 계산
         logger.info("월별 리뷰 건수 집계 시작")
-        df_selected['년월'] = df_selected['리뷰날짜'].dt.strftime('%Y-%m')
-        review_counts = df_selected.groupby('년월').size().reset_index(name='리뷰건수')
+        df_recent['년월'] = df_recent['리뷰날짜'].dt.strftime('%Y-%m')
+        review_counts = df_recent.groupby('년월').size().reset_index(name='리뷰건수')
         logger.info("월별 리뷰 건수 집계 완료")
 
+        # 새로운 시트에 저장
+        logger.info("새로운 시트 '월별 리뷰건수' 생성 시작")
+        with pd.ExcelWriter(file_path, engine='openpyxl', mode='a') as writer:
+            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
+        logger.info("새로운 시트 '월별 리뷰건수' 생성 완료")
+
+        # 요청사항 추가 시작
+        logger.info("선택년 데이터 필터링 및 분석 시작")
+        # 선택한 연도로 데이터 필터링
+        df_selected = df[df['리뷰날짜'].dt.year == selected_year]
+        logger.info(f"선택한 연도 {selected_year} 데이터 필터링 완료")
+
+        # 년월별 리뷰 건수 계산
+        df_selected['년월'] = df_selected['리뷰날짜'].dt.strftime('%Y-%m')
+        selected_review_counts = df_selected.groupby('년월').size().reset_index(name='리뷰건수')
+        logger.info("선택년 월별 리뷰 건수 집계 완료")
+
         # 새로운 시트에 저장
         logger.info("새로운 시트 '선택월별 리뷰건수' 생성 시작")
         with pd.ExcelWriter(file_path, engine='openpyxl', mode='a') as writer:
-            review_counts.to_excel(writer, sheet_name='선택월별 리뷰건수', index=False)
+            selected_review_counts.to_excel(writer, sheet_name='선택월별 리뷰건수', index=False)
         logger.info("새로운 시트 '선택월별 리뷰건수' 생성 완료")
+        # 요청사항 추가 끝
 
         return file_path
     except Exception as e:
@@ -54,14 +71,15 @@ def main():
         gr.Markdown("### 리뷰 분석 스페이스")
         with gr.Row():
             file_input = gr.File(label="원본 엑셀 파일 업로드", file_types=[".xlsx"])
-            with gr.Column():
-                current_year = 2025
-                years = [str(current_year - i) for i in range(5)]
-                year_radio = gr.Radio(label="분석할 연도 선택", choices=years, value=str(current_year))
             file_output = gr.File(label="분석된 엑셀 파일 다운로드", type="filepath")
+        with gr.Row():
+            # 최근 5년을 라디오 버튼으로 선택
+            year_options = [str(year) for year in range(2021, 2026)]  # 2021년부터 2025년까지
+            year_selection = gr.Radio(label="분석할 연도 선택", choices=year_options, value="2025")
         analyze_button = gr.Button("분석")
         
-        analyze_button.click(fn=analyze_reviews, inputs=[file_input, year_radio], outputs=file_output)
+        # 버튼 클릭 시 analyze_reviews 함수 호출, 파일과 선택한 연도 전달
+        analyze_button.click(fn=analyze_reviews, inputs=[file_input, year_selection], outputs=file_output)
     
     demo.launch()