Update app.py

app.py

@@ -2,60 +2,56 @@ import gradio as gr
 import pandas as pd
 import logging
 from datetime import datetime
-import os
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-def analyze_reviews(file_path):
+def analyze_reviews(file):
     try:
-        logging.info("파일 업로드 시작: %s", file_path)
+        logging.info("파일 업로드 시작.")
         # 엑셀 파일 읽기
-        df = pd.read_excel(file_path)
-        logging.info("엑셀 파일 읽기 완료")
-
-        # 현재 연도 기준 최근 3년 설정 (2025년 기준)
+        df = pd.read_excel(file.name, engine='openpyxl')
+        logging.info("엑셀 파일 읽기 완료.")
+        
+        # 현재 연도 설정 (2025년 기준)
         current_year = 2025
-        start_year = current_year - 3
-        logging.info("분석할 기간: %d년부터 %d년까지", start_year, current_year)
-
-        # B열이 리뷰 날짜라고 가정
-        df['B'] = pd.to_datetime(df.iloc[:, 1], errors='coerce')
-        df = df.dropna(subset=['B'])
-        df['Year'] = df['B'].dt.year
-        df['Month'] = df['B'].dt.month
-
+        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년 데이터 필터링
-        df_filtered = df[(df['Year'] > start_year) & (df['Year'] <= current_year)]
-        logging.info("최근 3년 데이터 필터링 완료: %d개의 데이터", len(df_filtered))
-
+        mask = (df['B'] >= start_date) & (df['B'] <= end_date)
+        filtered_df = df.loc[mask]
+        logging.info(f"필터링된 데이터 행 수: {len(filtered_df)}.")
+        
         # 년월별 리뷰 건수 집계
-        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_path, engine='openpyxl', mode='a') as writer:
-            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=False, startrow=0, startcol=0)
-        logging.info("새로운 시트 '월별 리뷰건수'에 저장 완료")
-
-        return file_path
+        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("분석 중 오류 발생: %s", e)
+        logging.error(f"오류 발생: {e}")
         return None
 
-# 그라디오 인터페이스 정의
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("# 리뷰 분석 스페이스")
-        with gr.Row():
-            file_input = gr.File(label="원본 엑셀 파일 업로드", file_types=[".xlsx"])
-        analyze_button = gr.Button("분석")
-        file_output = gr.File(label="분석된 엑셀 파일 다운로드", file_types=[".xlsx"], type="filepath")
-        
-        analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=file_output)
+# 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")
     
-    demo.launch()
+    analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=file_output)
 
+# 스페이스 실행
 if __name__ == "__main__":
-    main()
+    demo.launch()