Update app.py

app.py

@@ -17,7 +17,7 @@ st.title("VOD 채팅 크롤러")
 st.write("VOD URL을 입력하고 채팅 데이터를 크롤링합니다.")
 
 # URL 입력 받기
-vod_url = st.text_input("VOD URL 입력")
+vod_url = st.text_input("VOD URL 입력", "https://chzzk.naver.com/video/3890359")
 
 # 선택된 시간대를 저장할 공간
 if 'selected_times' not in st.session_state:
@@ -105,9 +105,10 @@ def crawl_chats(vod_url):
                     laugh_counts[minute_key] += 1
 
         total_chats_collected += len(chats)
-        elapsed_time = time.time() - start_time  # 경과 시간 계산
-        elapsed_minutes, elapsed_seconds = divmod(int(elapsed_time), 60)
-        status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected} | 경과 시간: {elapsed_minutes}분 {elapsed_seconds}초")
+
+        # 현재 시간과 경과 시간 표시
+        elapsed_time = time.time() - start_time
+        status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected} | 경과 시간: {int(elapsed_time // 60)}분 {int(elapsed_time % 60)}초")
 
         # 다음 요청을 위해 playerMessageTime 파라미터 업데이트
         next_time = data["content"].get("nextPlayerMessageTime")
@@ -117,45 +118,6 @@ def crawl_chats(vod_url):
 
     return "\n".join(chat_logs), chat_counts, laugh_counts
 
-# 선택한 시간대 추가 및 출력
-def add_selected_time(hour, minute):
-    time_str = f"{int(hour):02}:{int(minute):02}:00"  # HH:MM:00 형식
-    if time_str not in st.session_state['selected_times']:
-        st.session_state['selected_times'].append(time_str)
-
-# 선택된 시간대를 표시 및 삭제 기능
-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[:-6]])  # Remove the last 6 characters ('/chats')
-
 # 버튼을 눌렀을 때 채팅 크롤링 시작
 if 'chat_logs' not in st.session_state:
     st.session_state['chat_logs'] = None
@@ -227,23 +189,59 @@ if st.session_state['chat_logs']:
         xaxis=dict(showticklabels=False),  # x축 시간 레이블 숨김
         hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
         showlegend=True,
-        margin=dict(l=50, r=50, t=100, b=100),
-        height=600  # 그래프 높이 설정
+        margin=dict(l=50, r=50, t=100, b=100)
     )
 
     # 그래프 출력
     st.plotly_chart(fig, use_container_width=True)
 
-    # 시간대 입력 필드
+    # 시간 입력 필드
     col1, col2 = st.columns([1, 1])
-    hour = col1.number_input("시간", min_value=0, max_value=23, format="%d", value=0)
-    minute = col2.number_input("분", min_value=0, max_value=59, format="%d", value=0)
+    with col1:
+        hour = st.number_input("시간", min_value=0, max_value=23, value=0)
+    with col2:
+        minute = st.number_input("분", min_value=0, max_value=59, value=0)
 
+    # 선택된 시간대 추가 버튼
     if st.button("시간대 선택"):
-        add_selected_time(hour, minute)
+        selected_time = f"{hour:02}:{minute:02}:00"  # "HH:MM:SS" 형식으로 변환
+        if selected_time not in st.session_state['selected_times']:
+            st.session_state['selected_times'].append(selected_time)
+
+        # 입력 칸 초기화
+        hour = 0
+        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)
 
-    # ��택된 시간대 표시
-    display_selected_times()
+    # 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[:-6]])  # Remove the last 6 characters ('/chats')
 
     # 다운로드 버튼 표시
     if st.button("선택된 시간대의 영상 다운로드"):