Update app.py

app.py

@@ -8,58 +8,68 @@ from io import BytesIO
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 def analyze_reviews(file):
+    logging.info("파일 업로드 시작")
     try:
-        logging.info("파일 업로드 시작")
         # 엑셀 파일 읽기
         df = pd.read_excel(file, engine='openpyxl')
         logging.info("엑셀 파일 읽기 완료")
-
-        # 현재 연도 설정 (2025년 기준)
+    except Exception as e:
+        logging.error(f"엑셀 파일 읽기 오류: {e}")
+        return f"엑셀 파일을 읽는 중 오류가 발생했습니다: {e}"
+    
+    try:
+        # 현재 기준 연도 설정 (2025년 기준)
         current_year = 2025
-        start_year = current_year - 3  # 2022년부터
+        start_year = current_year - 3  # 최근 3년
 
-        logging.info("리뷰 날짜 데이터 처리 시작")
+        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['리뷰날짜'] = pd.to_datetime(df.iloc[:, 1], errors='coerce')
+        # 연도 및 월 필터링
+        df_filtered = df[(df['리뷰날짜'].dt.year >= start_year) & (df['리뷰날짜'].dt.year < current_year)]
+        logging.info("리뷰 날짜 필터링 완료")
 
-        # 년월별 리뷰 건수 계산
-        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("월별 리뷰 건수 계산 완료")
+        # 년월별 리뷰 건수 집계
+        logging.info("년월별 리뷰 건수 집계 시작")
+        df_filtered['년월'] = df_filtered['리뷰날짜'].dt.strftime('%Y-%m')
+        review_counts = df_filtered.groupby('년월').size().reset_index(name='리뷰건수')
+        logging.info("년월별 리뷰 건수 집계 완료")
 
-        # 새로운 시트에 작성
+        # 새로운 시트에 데이터 작성
+        logging.info("새로운 시트 생성 및 데이터 작성 시작")
         with pd.ExcelWriter(file, engine='openpyxl', mode='a') as writer:
-            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
-        logging.info("새로운 시트 '월별 리뷰건수' 추가 완료")
+            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['A열', 'B열'])
+        logging.info("새로운 시트에 데이터 작성 완료")
 
-        # 결과 파일을 바이트 형태로 변환
+        # 결과 파일 저장
+        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)
+            writer.book = pd.ExcelFile(file).book
+            for sheet in pd.ExcelFile(file).sheet_names:
+                df_sheet = pd.read_excel(file, sheet_name=sheet, engine='openpyxl')
+                df_sheet.to_excel(writer, sheet_name=sheet, index=False)
+            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['년월', '리뷰건수'])
         output.seek(0)
-        logging.info("분석된 엑셀 파일 생성 완료")
+        logging.info("결과 파일 저장 완료")
 
         return output
-
     except Exception as e:
-        logging.error(f"에러 발생: {e}")
-        return None
+        logging.error(f"분석 과정 중 오류: {e}")
+        return f"분석 과정에서 오류가 발생했습니다: {e}"
 
 # Gradio 인터페이스 구성
 with gr.Blocks() as demo:
-    gr.Markdown("## 리뷰 분석기")
+    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)
+        file_input = gr.File(label="원본 엑셀 파일 업로드", type="file")
+    
+    analyze_button = gr.Button("분석")
+    output_download = gr.File(label="분석된 엑셀 파일 다운로드")
+    
+    analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=output_download)
 
-# 애플리케이션 실행
+# 스페이스 실행
 if __name__ == "__main__":
     demo.launch()