Update app.py

app.py

@@ -10,13 +10,15 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 def analyze_reviews(file):
     logging.info("파일 업로드 시작")
     try:
+        # 바이너리 데이터를 BytesIO로 변환
+        file_bytes = BytesIO(file)
         # 엑셀 파일 읽기
-        df = pd.read_excel(file, engine='openpyxl')
+        df = pd.read_excel(file_bytes, engine='openpyxl')
         logging.info("엑셀 파일 읽기 완료")
     except Exception as e:
         logging.error(f"엑셀 파일 읽기 오류: {e}")
         return f"엑셀 파일을 읽는 중 오류가 발생했습니다: {e}"
-    
+
     try:
         # 현재 기준 연도 설정 (2025년 기준)
         current_year = 2025
@@ -37,18 +39,21 @@ def analyze_reviews(file):
 
         # 새로운 시트에 데이터 작성
         logging.info("새로운 시트 생성 및 데이터 작성 시작")
-        with pd.ExcelWriter(file, engine='openpyxl', mode='a') as writer:
-            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['A열', 'B열'])
+        with pd.ExcelWriter(file_bytes, engine='openpyxl', mode='a', if_sheet_exists='replace') as writer:
+            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['년월', '리뷰건수'])
         logging.info("새로운 시트에 데이터 작성 완료")
 
         # 결과 파일 저장
         logging.info("결과 파일 저장 시작")
         output = BytesIO()
+        file_bytes.seek(0)  # 원본 파일 포인터 초기화
         with pd.ExcelWriter(output, engine='openpyxl') as writer:
-            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)
+            # 모든 시트 복사
+            with pd.ExcelFile(file_bytes) as xls:
+                for sheet_name in xls.sheet_names:
+                    df_sheet = pd.read_excel(xls, sheet_name=sheet_name, engine='openpyxl')
+                    df_sheet.to_excel(writer, sheet_name=sheet_name, index=False)
+            # 추가된 '월별 리뷰건수' 시트 추가
             review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['년월', '리뷰건수'])
         output.seek(0)
         logging.info("결과 파일 저장 완료")
@@ -61,13 +66,13 @@ def analyze_reviews(file):
 # Gradio 인터페이스 구성
 with gr.Blocks() as demo:
     gr.Markdown("## 리뷰 분석 스페이스")
-    
+
     with gr.Row():
-        file_input = gr.File(label="원본 엑셀 파일 업로드", type="file")
-    
+        file_input = gr.File(label="원본 엑셀 파일 업로드", type="binary")
+
     analyze_button = gr.Button("분석")
     output_download = gr.File(label="분석된 엑셀 파일 다운로드")
-    
+
     analyze_button.click(fn=analyze_reviews, inputs=file_input, outputs=output_download)
 
 # 스페이스 실행