Update app.py

app.py

@@ -1,5 +1,3 @@
-import os
-import time
 import streamlit as st
 import pandas as pd
 import plotly.graph_objects as go
@@ -8,6 +6,8 @@ import datetime
 import re
 import requests
 import yt_dlp
+import os
+import time
 
 # 페이지 구성을 'wide'로 설정하여 여백 최소화
 st.set_page_config(layout="wide")
@@ -115,6 +115,11 @@ def crawl_chats(vod_url):
 
     return "\n".join(chat_logs), chat_counts, laugh_counts
 
+# 클릭한 시간대를 선택 및 출력
+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():
     if st.session_state['selected_times']:
@@ -140,13 +145,13 @@ def download_clips():
             # yt-dlp 다운로드 명령어 실행
             output_filename = f"clip_{idx + 1}.mp4"
             ydl_opts = {
-                'outtmpl': os.path.join(os.path.expanduser("~/Downloads"), output_filename),  # Downloads 폴더에 저장
+                '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])  # 원래 URL 사용
+                ydl.download([vod_url])
 
 # 버튼을 눌렀을 때 채팅 크롤링 시작
 if 'chat_logs' not in st.session_state:
@@ -226,24 +231,13 @@ if st.session_state['chat_logs']:
     # 그래프 출력
     st.plotly_chart(fig, use_container_width=True)
 
-    # 시간 입력 필드
-    col1, col2 = st.columns([1, 1])
-    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("시간대 선택"):
-        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
+    # 그래프 클릭 시 시간대 추가
+    click_data = st.session_state.get('click_data')
+    if click_data:
+        time_selected = click_data['points'][0]['x']
+        add_selected_time(time_selected)
 
-    # 선택된 시간대를 표시 및 삭제 기능
+    # 선택된 시간대 표시
     display_selected_times()
 
     # 다운로드 버튼 표시