Update app.py

app.py

@@ -5,10 +5,8 @@ from collections import defaultdict
 import datetime
 import re
 import requests
-import time
-
-# 페이지 구성을 'wide'로 설정하여 여백 최소화
-st.set_page_config(layout="wide")
+import yt_dlp
+import os
 
 # Streamlit 제목 및 설명
 st.title("VOD 채팅 크롤러")
@@ -17,6 +15,10 @@ st.write("VOD URL을 입력하고 채팅 데이터를 크롤링합니다.")
 # URL 입력 받기
 vod_url = st.text_input("VOD URL 입력")
 
+# 선택된 시간대를 저장할 공간
+if 'selected_times' not in st.session_state:
+    st.session_state['selected_times'] = []
+
 # 채팅 크롤링 함수
 def crawl_chats(vod_url):
     # URL 설정
@@ -41,7 +43,6 @@ def crawl_chats(vod_url):
     chat_counts = defaultdict(int)
     laugh_counts = defaultdict(int)
     total_chats_collected = 0  # 총 수집된 채팅 개수
-    start_time = time.time()  # 크롤링 시작 시간 기록
 
     # 채팅 데이터를 순차적으로 요청하여 가져오기
     status_text = st.empty()  # 상태 메시지 출력용
@@ -98,14 +99,7 @@ def crawl_chats(vod_url):
                     laugh_counts[minute_key] += 1
 
         total_chats_collected += len(chats)
-
-        # 크롤링 경과 시간 계산
-        elapsed_time = time.time() - start_time
-        elapsed_minutes = int(elapsed_time // 60)
-        elapsed_seconds = int(elapsed_time % 60)
-
-        # 상태 메시지 업데이트 (총 채팅 개수와 경과 시간 표시)
-        status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected} | 경과 시간: {elapsed_minutes}분 {elapsed_seconds}초")
+        status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected}")
 
         # 다음 요청을 위해 playerMessageTime 파라미터 업데이트
         next_time = data["content"].get("nextPlayerMessageTime")
@@ -115,6 +109,46 @@ def crawl_chats(vod_url):
 
     return "\n".join(chat_logs), chat_counts, laugh_counts
 
+# 클릭한 시간대를 선택 및 출력
+def handle_time_selection(click_data):
+    if click_data:
+        selected_time = click_data['points'][0]['x']
+        if selected_time not in st.session_state['selected_times']:
+            st.session_state['selected_times'].append(selected_time)
+
+# 선택된 시간대를 표시 및 삭제 기능
+def display_selected_times():
+    if st.session_state['selected_times']:
+        st.write("### 선택된 시간대")
+        for time in st.session_state['selected_times']:
+            col1, col2 = st.columns([9, 1])
+            col1.write(f"{time}")
+            if col2.button("X", key=f"remove_{time}"):
+                st.session_state['selected_times'].remove(time)
+
+# yt-dlp를 사용하여 선택된 시간대의 영상을 다운로드
+def download_clips():
+    if st.session_state['selected_times']:
+        st.write("### 영상 다운로드")
+        for idx, start_time in enumerate(st.session_state['selected_times']):
+            start_time_obj = datetime.datetime.strptime(start_time, '%H:%M:%S')
+            end_time_obj = start_time_obj + datetime.timedelta(minutes=1)
+
+            # 시작 시간과 끝 시간을 HH:MM:SS 형식으로 변환
+            start_time_str = start_time_obj.strftime('%H:%M:%S')
+            end_time_str = end_time_obj.strftime('%H:%M:%S')
+
+            # yt-dlp 다운로드 명령어 실행
+            output_filename = f"clip_{idx + 1}.mp4"
+            ydl_opts = {
+                'outtmpl': output_filename,
+                'download_sections': [f"*{start_time_str}-{end_time_str}"]
+            }
+
+            st.write(f"다운로드 중: {start_time_str} ~ {end_time_str} | 파일명: {output_filename}")
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                ydl.download([vod_url])
+
 # 버튼을 눌렀을 때 채팅 크롤링 시작
 if 'chat_logs' not in st.session_state:
     st.session_state['chat_logs'] = None
@@ -162,7 +196,7 @@ if st.session_state['chat_logs']:
     fig.add_trace(go.Scatter(
         x=df['시간'],
         y=df['전체 채팅 개수'],
-        mode='lines',  # 마커 없이 선만 표시
+        mode='lines',
         name='전체 채팅 개수',
         line=dict(color='blue'),
         hovertemplate='%{x} - 전체 채팅 개수: %{y}<extra></extra>'
@@ -172,7 +206,7 @@ if st.session_state['chat_logs']:
     fig.add_trace(go.Scatter(
         x=df['시간'],
         y=df['ㅋㅋㅋㅋ 채팅 개수'],
-        mode='lines',  # 마커 없이 선만 표시
+        mode='lines',
         name='ㅋㅋㅋㅋ 채팅 개수',
         line=dict(color='red'),
         hovertemplate='%{x} - ㅋㅋㅋㅋ 채팅 개수: %{y}<extra></extra>'
@@ -181,16 +215,22 @@ if st.session_state['chat_logs']:
     # 그래프 레이아웃 설정
     fig.update_layout(
         title="분당 채팅 및 ㅋㅋㅋㅋ 채팅 개수",
-        height=600,  # 그래프의 높이를 키움
         xaxis_title="시간",
         yaxis_title="채팅 개수",
-        xaxis=dict(
-            showticklabels=False  # x축 시간 레이블 숨김
-        ),
+        xaxis=dict(showticklabels=False),  # x축 시간 레이블 숨김
         hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
-        showlegend=True,  # 범례 표시
-        margin=dict(l=50, r=50, t=100, b=100)  # 그래프 아래쪽 여백을 조정
+        showlegend=True,
+        margin=dict(l=50, r=50, t=100, b=100)
     )
 
-    # 그래프 출력
-    st.plotly_chart(fig, use_container_width=True)  # 화면
+    # 그래프 출력 및 클릭 이벤트 처리
+    st.plotly_chart(fig, use_container_width=True)
+    click_data = st.plotly_chart(fig).get_current_click()
+    handle_time_selection(click_data)
+
+    # 선택된 시간대 표시
+    display_selected_times()
+
+    # 다운로드 버튼 표시
+    if st.button("선택된 시간대의 영상 다운로드"):
+        download_clips()