Update app.py

app.py

@@ -115,12 +115,10 @@ 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 add_selected_time(time):
+    if time not in st.session_state['selected_times']:
+        st.session_state['selected_times'].append(time)
 
 # 선택된 시간대를 표시 및 삭제 기능
 def display_selected_times():
@@ -198,7 +196,7 @@ if st.session_state['chat_logs']:
     # Plotly 선 그래프 그리기
     fig = go.Figure()
 
-    # 전체 채팅 개수와 ㅋㅋㅋㅋ 채팅 개수를 한 그래프에 추가
+    # 전체 채팅 개수 선 그래프 추가
     fig.add_trace(go.Scatter(
         x=df['시간'],
         y=df['전체 채팅 개수'],
@@ -208,6 +206,7 @@ if st.session_state['chat_logs']:
         hovertemplate='%{x} - 전체 채팅 개수: %{y}<extra></extra>'
     ))
 
+    # ㅋㅋㅋㅋ 채팅 개수 선 그래프 추가
     fig.add_trace(go.Scatter(
         x=df['시간'],
         y=df['ㅋㅋㅋㅋ 채팅 개수'],
@@ -228,9 +227,12 @@ if st.session_state['chat_logs']:
         margin=dict(l=50, r=50, t=100, b=100)
     )
 
-    # 그래프 출력 및 클릭 이벤트 처리
-    click_data = st.plotly_chart(fig, use_container_width=True).get_current_click()
-    handle_time_selection(click_data)
+    # 그래프 출력
+    click_data = st.plotly_chart(fig, use_container_width=True)
+
+    # 클릭한 시간대가 있으면 처리
+    if click_data:
+        add_selected_time(click_data['points'][0]['x'])
 
     # 선택된 시간대 표시
     display_selected_times()