Update app.py

app.py

@@ -49,9 +49,6 @@ def crawl_chats(vod_url):
 
     # 채팅 데이터를 순차적으로 요청하여 가져오기
     status_text = st.empty()  # 상태 메시지 출력용
-    timer_text = st.empty()  # 경과 시간 표시용
-    start_time = datetime.datetime.now()  # 시작 시간 기록
-
     while True:
         # API 요청 보내기
         response = requests.get(url, params=params, headers=headers)
@@ -105,9 +102,7 @@ def crawl_chats(vod_url):
                     laugh_counts[minute_key] += 1
 
         total_chats_collected += len(chats)
-        elapsed_time = datetime.datetime.now() - start_time  # 경과 시간 계산
         status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected}")
-        timer_text.text(f"경과 시간: {elapsed_time.seconds // 60}분 {elapsed_time.seconds % 60}초")  # 경과 시간 표시
 
         # 다음 요청을 위해 playerMessageTime 파라미터 업데이트
         next_time = data["content"].get("nextPlayerMessageTime")
@@ -117,6 +112,12 @@ 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']:
@@ -148,7 +149,7 @@ def download_clips():
 
             st.write(f"다운로드 중: {start_time_str} ~ {end_time_str} | 파일명: {output_filename}")
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                ydl.download([vod_url])
+                ydl.download([vod_url[:-6]])  # Remove the last 6 characters ('/chats')
 
 # 버튼을 눌렀을 때 채팅 크롤링 시작
 if 'chat_logs' not in st.session_state:
@@ -218,28 +219,25 @@ if st.session_state['chat_logs']:
         title="분당 채팅 및 ㅋㅋㅋㅋ 채팅 개수",
         xaxis_title="시간",
         yaxis_title="채팅 개수",
-        height=600,
         xaxis=dict(showticklabels=False),  # x축 시간 레이블 숨김
         hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
         showlegend=True,
-        margin=dict(l=50, r=50, t=100, b=100)
+        margin=dict(l=50, r=50, t=100, b=100),
+        height=600  # 그래프 높이 설정
     )
 
     # 그래프 출력
     st.plotly_chart(fig, use_container_width=True)
 
-    # 시간과 분 입력란 추가
-    col1, col2 = st.columns(2)
-    hour = col1.number_input("시간", min_value=0, max_value=23, value=0)
-    minute = col2.number_input("분", min_value=0, max_value=59, value=0)
+    # 시간대 입력 필드
+    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)
 
-    # 시간대 선택 버튼
     if st.button("시간대 선택"):
-        # 입력한 시간과 분을 기반으로 선택된 시간대 생성
-        selected_time = f"{hour:02}:{minute:02}:00"  # "HH:MM:SS" 형식
-        st.session_state['selected_times'].append(selected_time)
+        add_selected_time(hour, minute)
 
-    # 선택된 시간대 표시 및 삭제 기능
+    # 선택된 시간대 표시
     display_selected_times()
 
     # 다운로드 버튼 표시