Update app.py

app.py

@@ -16,7 +16,7 @@ st.set_page_config(layout="wide")
 st.title("VOD 채팅 크롤러")
 st.write("VOD URL을 입력하고 채팅 데이터를 크롤링합니다.")
 
-# URL 입력 받기 (기본값 없이)
+# URL 입력 받기
 vod_url = st.text_input("VOD URL 입력")
 
 # 선택된 시간대를 저장할 공간
@@ -25,7 +25,7 @@ if 'selected_times' not in st.session_state:
 
 # 채팅 크롤링 함수
 def crawl_chats(vod_url):
-    # URL 설정 (여기서 /chats를 붙임)
+    # URL 설정
     url = vod_url + "/chats"
 
     # 요청 헤더 (필요시 쿠키나 기타 헤더 정보 추가)
@@ -47,11 +47,10 @@ 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()  # 상태 메시지 출력용
-    start_time = time.time()  # 크롤링 시작 시간
-
     while True:
         # API 요청 보내기
         response = requests.get(url, params=params, headers=headers)
@@ -105,9 +104,7 @@ def crawl_chats(vod_url):
                     laugh_counts[minute_key] += 1
 
         total_chats_collected += len(chats)
-
-        # 현재 시간과 경과 시간 표시
-        elapsed_time = time.time() - start_time
+        elapsed_time = time.time() - start_time  # 경과 시간 계산
         status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected} | 경과 시간: {int(elapsed_time // 60)}분 {int(elapsed_time % 60)}초")
 
         # 다음 요청을 위해 playerMessageTime 파라미터 업데이트
@@ -118,6 +115,39 @@ def crawl_chats(vod_url):
 
     return "\n".join(chat_logs), chat_counts, laugh_counts
 
+# 선택된 시간대를 표시 및 삭제 기능
+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': os.path.join(os.path.expanduser("~/Downloads"), output_filename),  # Downloads 폴더에 저장
+                '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])  # 원래 URL 사용
+
 # 버튼을 눌렀을 때 채팅 크롤링 시작
 if 'chat_logs' not in st.session_state:
     st.session_state['chat_logs'] = None
@@ -186,6 +216,7 @@ if st.session_state['chat_logs']:
         title="분당 채팅 및 ㅋㅋㅋㅋ 채팅 개수",
         xaxis_title="시간",
         yaxis_title="채팅 개수",
+        height=600,  # 그래프 높이 설정
         xaxis=dict(showticklabels=False),  # x축 시간 레이블 숨김
         hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
         showlegend=True,
@@ -213,35 +244,7 @@ if st.session_state['chat_logs']:
         minute = 0
 
     # 선택된 시간대를 표시 및 삭제 기능
-    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']:
-            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': os.path.join(os.path.expanduser("~/Downloads"), 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])  # 원래 URL 사용
+    display_selected_times()
 
     # 다운로드 버튼 표시
     if st.button("선택된 시간대의 영상 다운로드"):