Update app.py

app.py

@@ -7,7 +7,6 @@ import re
 import requests
 import yt_dlp
 import os
-import time
 
 # 페이지 구성을 'wide'로 설정하여 여백 최소화
 st.set_page_config(layout="wide")
@@ -47,7 +46,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()  # 상태 메시지 출력용
@@ -104,8 +102,7 @@ def crawl_chats(vod_url):
                     laugh_counts[minute_key] += 1
 
         total_chats_collected += len(chats)
-        elapsed_time = time.time() - start_time  # 경과 시간 계산
-        status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected} | 경과 시간: {int(elapsed_time // 60)}분 {int(elapsed_time % 60)}초")
+        status_text.text(f"현재까지 수집된 채팅 메시지 개수: {total_chats_collected}")
 
         # 다음 요청을 위해 playerMessageTime 파라미터 업데이트
         next_time = data["content"].get("nextPlayerMessageTime")
@@ -216,7 +213,7 @@ if st.session_state['chat_logs']:
         title="분당 채팅 및 ㅋㅋㅋㅋ 채팅 개수",
         xaxis_title="시간",
         yaxis_title="채팅 개수",
-        height=600,  # 그래프 높이 설정
+        height=600,
         xaxis=dict(showticklabels=False),  # x축 시간 레이블 숨김
         hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
         showlegend=True,
@@ -226,20 +223,18 @@ if st.session_state['chat_logs']:
     # 그래프 출력
     st.plotly_chart(fig, use_container_width=True)
 
-    # 선택한 시간대를 출력할 공간
-    if 'selected_time' not in st.session_state:
-        st.session_state['selected_time'] = None
+    # 시간과 분 입력란 추가
+    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)
 
-    # 클릭 이벤트 처리
-    clicked_time = st.button("선택된 시간대 출력")
-    if clicked_time:
-        st.session_state['selected_time'] = clicked_time  # 클릭된 시간으로 업데이트
+    # 시간대 선택 버튼
+    if st.button("시간대 선택"):
+        # 입력한 시간과 분을 기반으로 선택된 시간대 생성
+        selected_time = f"{hour:02}:{minute:02}:00"  # "HH:MM:SS" 형식
+        st.session_state['selected_times'].append(selected_time)
 
-    # 선택된 시간대 표시
-    if st.session_state['selected_time']:
-        st.write(f"선택된 시간대: {st.session_state['selected_time']}")
-
-    # 선택된 시간대 표시
+    # 선택된 시간대 표시 및 삭제 기능
     display_selected_times()
 
     # 다운로드 버튼 표시