Update app.py

app.py

@@ -1,115 +1,68 @@
-import gradio as gr
-from moviepy.editor import VideoFileClip
-from PIL import Image
-import tempfile
+dimport gradio as gr
+import pandas as pd
+from datetime import datetime
 import logging
 
-# Configure logging
+# 로그 설정
 logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# Function to get video info
-def get_video_info(video):
-    logging.info("Getting video info")
-    if not video:
-        return "No video uploaded."
-    clip = VideoFileClip(video)
-    duration = clip.duration
-    clip.close()
-    return f"Duration: {duration:.2f} seconds"
+def analyze_reviews(file_path):
+    try:
+        logger.info("엑셀 파일 업로드 시작")
+        # 엑셀 파일 읽기
+        df = pd.read_excel(file_path)
+        logger.info("엑셀 파일 읽기 완료")
 
-# Function to generate thumbnails
-def generate_thumbnails(video, start_time, end_time):
-    logging.info("Generating thumbnails")
-    if not video:
-        return None, None
-    clip = VideoFileClip(video)
-    if start_time > clip.duration:
-        start_img = None
-    else:
-        start_frame = clip.get_frame(start_time)
-        start_img = Image.fromarray(start_frame)
-    if end_time > clip.duration:
-        end_img = None
-    else:
-        end_frame = clip.get_frame(end_time)
-        end_img = Image.fromarray(end_frame)
-    clip.close()
-    return start_img, end_img
+        # 현재 연도
+        current_year = 2025
+        start_year = current_year - 3  # 최근 3년
 
-# Function to generate GIF
-def generate_gif(video, start_time, end_time, resolution, frame_rate, playback_speed, loop_count):
-    logging.info("Generating GIF")
-    if not video:
-        return None, None
-    clip = VideoFileClip(video)
-    # Apply start and end times
-    if start_time > clip.duration:
-        start_time = 0
-    if end_time > clip.duration:
-        end_time = clip.duration
-    subclip = clip.subclip(start_time, end_time)
-    # Apply playback speed
-    new_fps = subclip.fps * playback_speed
-    if new_fps <= 0:
-        new_fps = subclip.fps
-    subclip = subclip.set_fps(new_fps)
-    # Resize if resolution is changed
-    if resolution != subclip.size[0]:
-        ratio = resolution / subclip.size[0]
-        new_height = int(subclip.size[1] * ratio)
-        subclip = subclip.resize((resolution, new_height))
-    # Write GIF
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".gif") as temp:
-        gif_path = temp.name
-        subclip.write_gif(gif_path, fps=frame_rate, loop=int(loop_count))
-    # Read GIF for preview
-    with Image.open(gif_path) as img:
-        preview_frame = img.copy()
-    return preview_frame, gif_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년 데이터 필터링 완료")
 
-# Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("<h1>Video to GIF Converter</h1>")
-    
-    # Upload video
-    video_input = gr.Video(label="Upload Video")
-    
-    # Button to get video info
-    with gr.Row():
-        get_info_button = gr.Button("Get Video Info")
-        video_info = gr.Textbox(label="Video Information")
-    
-    # Display uploaded video
-    video_output = gr.Video(label="Uploaded Video")
-    
-    # Inputs for start and end times
-    start_time = gr.Number(label="Start Time (seconds)", value=0)
-    end_time = gr.Number(label="End Time (seconds)", value=0)
-    
-    # Thumbnails
-    with gr.Row():
-        start_thumbnail = gr.Image(label="Start Thumbnail")
-        end_thumbnail = gr.Image(label="End Thumbnail")
-    
-    # Sliders
-    resolution = gr.Slider(100, 1920, step=100, value=1920, label="Resolution")
-    frame_rate = gr.Slider(1, 30, step=1, value=10, label="Frame Rate")
-    playback_speed = gr.Slider(0.1, 2.0, step=0.1, value=1.0, label="Playback Speed")
-    loop_count = gr.Slider(0, 10, step=1, value=0, label="GIF Loop Count (0 for infinite)")
-    
-    # GIF Generation button
-    generate_button = gr.Button("GIF 생성")
-    
-    # GIF Preview and Download
-    gif_preview = gr.Image(label="GIF Preview")
-    gif_download = gr.File(label="Download GIF")
+        # 년월별 리뷰 건수 계산
+        logger.info("월별 리뷰 건수 집계 시작")
+        df_recent['년월'] = df_recent['리뷰날짜'].dt.strftime('%Y-%m')
+        review_counts = df_recent.groupby('년월').size().reset_index(name='리뷰건수')
+        logger.info("월별 리뷰 건수 집계 완료")
+
+        # 새로운 시트에 저장
+        logger.info("새로운 시��� '월별 리뷰건수' 생성 시작")
+        with pd.ExcelWriter(file_path, engine='openpyxl', mode='a') as writer:
+            review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False)
+        logger.info("새로운 시트 '월별 리뷰건수' 생성 완료")
+
+        return file_path
+    except Exception as e:
+        logger.error(f"분석 중 오류 발생: {e}")
+        return None
+
+# 그라디오 인터페이스 구성
+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)
     
-    # Button functions
-    get_info_button.click(get_video_info, inputs=video_input, outputs=video_info)
-    video_input.change(lambda x: x, inputs=video_input, outputs=video_output)
-    video_input.change(generate_thumbnails, inputs=[video_input, start_time, end_time], outputs=[start_thumbnail, end_thumbnail])
-    generate_button.click(generate_gif, 
-                         inputs=[video_input, start_time, end_time, resolution, frame_rate, playback_speed, loop_count], 
-                         outputs=[gif_preview, gif_download])
+    demo.launch()
+
+if __name__ == "__main__":
+    main()
 
-demo.launch()
+gradio==4.44.1
+pandas
+openpyxl