Update app.py

app.py

@@ -1,64 +1,60 @@
 import gradio as gr
 import pandas as pd
 from datetime import datetime
-import logging
+import os
 
-# 로그 설정
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+# 로그 메시지를 출력하는 함수
+def log_message(step):
+    print(f"[LOG] {step} 단계가 실행되었습니다.")
 
+# 엑셀 파일 처리 및 분석 함수
 def analyze_reviews(file_path):
-    try:
-        logger.info("엑셀 파일 업로드 시작")
-        # 엑셀 파일 읽기
-        df = pd.read_excel(file_path)
-        logger.info("엑셀 파일 읽기 완료")
+    log_message("1. 파일 업로드 처리 중")
 
-        # 현재 연도
-        current_year = 2025
-        start_year = current_year - 3  # 최근 3년
+    # 엑셀 파일 읽기
+    df = pd.read_excel(file_path)
 
-        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')
-        # 최근 3년 데이터 필터링
-        df_recent = df[df['리뷰날짜'].dt.year >= start_year]
-        logger.info("최근 3년 데이터 필터링 완료")
+    if "B" not in df.columns:
+        return "오류: 원본 데이터에 B열(리뷰 날짜)이 없습니다."
 
-        # 년월별 리뷰 건수 계산
-        logger.info("월별 리뷰 건수 집계 시작")
-        df_recent['년월'] = df_recent['리뷰날짜'].dt.strftime('%Y-%m')
-        review_counts = df_recent.groupby('년월').size().reset_index(name='리뷰건수')
-        logger.info("월별 리뷰 건수 집계 완료")
+    # "B" 열에서 년도와 월을 추출하여 새로운 열 생성
+    df["년도-월"] = pd.to_datetime(df["B"]).dt.to_period("M").astype(str)
 
-        # 새로운 시트에 저장
-        logger.info("새로운 시트 '월별 리뷰건수' 생성 시작")
-        with pd.ExcelWriter(file_path, engine='openpyxl', mode='a') as writer:
-            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
-        logger.info("새로운 시트 '월별 리뷰건수' 생성 완료")
+    # 최근 3년 데이터를 필터링 (기준: 2025년)
+    current_year = 2025
+    recent_years = [str(year) for year in range(current_year - 2, current_year + 1)]
+    filtered_df = df[df["년도-월"].str[:4].isin(recent_years)]
 
-        return file_path
-    except Exception as e:
-        logger.error(f"분석 중 오류 발생: {e}")
-        return None
+    # 년월별 리뷰 건수 계산
+    review_counts = filtered_df["년도-월"].value_counts().sort_index().reset_index()
+    review_counts.columns = ["년도-월", "리뷰 건수"]
 
-# 그라디오 인터페이스 구성
-def main():
-    logger.info("그라디오 인터페이스 시작")
-    with gr.Blocks() as demo:
-        gr.Markdown("### 리뷰 분석 스페이스")
-        with gr.Row():
-            file_input = gr.File(label="원본 엑셀 파일 업로드", file_types=[".xlsx"])
-            file_output = gr.File(label="분석된 엑셀 파일 다운로드", type="filepath")
-        analyze_button = gr.Button("분석")
-        
-        analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=file_output)
+    # 새 엑셀 파일 작성
+    output_path = "review_analysis.xlsx"
+    with pd.ExcelWriter(output_path) as writer:
+        review_counts.to_excel(writer, sheet_name="월별 리뷰건수", index=False)
+        writer.sheets["월별 리뷰건수"].cell(1, 1, "년도-월")
+        writer.sheets["월별 리뷰건수"].cell(1, 2, "리뷰 건수")
+
+    log_message("2. 리뷰 분석 완료 및 파일 생성됨")
     
-    demo.launch()
+    return output_path
 
-if __name__ == "__main__":
-    main()
+# 그라디오 인터페이스 구성
+def interface(file):
+    log_message("3. 분석 버튼 클릭됨")
+    analyzed_file_path = analyze_reviews(file.name)
+    log_message("4. 엑셀 파일 다운로드 준비 완료")
+    return analyzed_file_path
+
+# 그라디오 UI 구성
+with gr.Blocks() as demo:
+    gr.Markdown("# 리뷰 분석 시스템")
+    file_input = gr.File(label="엑셀 파일 업로드", type="file", file_types=[".xlsx"])
+    analyze_button = gr.Button("분석하기")
+    download_file = gr.File(label="분석된 엑셀 파일 다운로드", type="filepath")
+
+    analyze_button.click(fn=interface, inputs=file_input, outputs=download_file)
+
+# 애플리케이션 실행
+demo.launch()