Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pandas as pd
 from datetime import datetime
 import logging
+import matplotlib.pyplot as plt
 
 # 로그 설정
 logging.basicConfig(level=logging.INFO)
@@ -29,19 +30,33 @@ def analyze_reviews(file_path):
         df_recent = df[df['리뷰날짜'].dt.year >= start_year]
         logger.info("최근 3년 데이터 필터링 완료")
 
-        # 년도별 리뷰 건수 계산
-        logger.info("년도별 리뷰 건수 집계 시작")
-        df_recent['년도'] = df_recent['리뷰날짜'].dt.year
-        yearly_review_counts = df_recent.groupby('년도').size().reset_index(name='리뷰건수')
-        logger.info("년도별 리뷰 건수 집계 완료")
+        # 월별 리뷰 건수 계산
+        logger.info("월별 리뷰 건수 집계 시작")
+        df_recent['년월'] = df_recent['리뷰날짜'].dt.to_period('M').astype(str)
+        monthly_review_counts = df_recent.groupby('년월').size().reset_index(name='리뷰건수')
+        logger.info("월별 리뷰 건수 집계 완료")
 
-        # 새로운 시트에 저장
+        # 그래프 생성
+        logger.info("그래프 생성 시작")
+        plt.figure(figsize=(10, 6))
+        plt.bar(monthly_review_counts['년월'], monthly_review_counts['리뷰건수'], color='skyblue')
+        plt.xlabel('년월')
+        plt.ylabel('리뷰 건수')
+        plt.title('최근 3년 월별 리뷰 건수')
+        plt.xticks(rotation=45)
+        graph_path = file_path.replace('.xlsx', '_graph.png')
+        plt.tight_layout()
+        plt.savefig(graph_path)
+        plt.close()
+        logger.info("그래프 생성 완료")
+
+        # 새로운 시트와 그래프 경로 저장
         logger.info("새로운 시트 생성 시작")
         with pd.ExcelWriter(file_path, engine='openpyxl', mode='a') as writer:
-            yearly_review_counts.to_excel(writer, sheet_name='년도별 리뷰건수', index=False)
+            monthly_review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
         logger.info("새로운 시트 생성 완료")
 
-        return file_path
+        return graph_path
     except Exception as e:
         logger.error(f"분석 중 오류 발생: {e}")
         return None
@@ -53,10 +68,10 @@ def main():
         gr.Markdown("### 리뷰 분석 스페이스")
         with gr.Row():
             file_input = gr.File(label="원본 엑셀 파일 업로드", file_types=[".xlsx"])
-            file_output = gr.File(label="분석된 엑셀 파일 다운로드", type="filepath")
+            graph_output = gr.File(label="생성된 그래프 파일 다운로드", type="filepath")
         analyze_button = gr.Button("분석")
         
-        analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=file_output)
+        analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=graph_output)
     
     demo.launch()