Update app.py

app.py

@@ -93,10 +93,21 @@ def crawl_chats(vod_url):
     return "\n".join(chat_logs), chat_counts, laugh_counts
 
 # 버튼을 눌렀을 때 채팅 크롤링 시작
+if 'chat_logs' not in st.session_state:
+    st.session_state['chat_logs'] = None
+if 'chat_counts' not in st.session_state:
+    st.session_state['chat_counts'] = None
+if 'laugh_counts' not in st.session_state:
+    st.session_state['laugh_counts'] = None
+
 if st.button("크롤링 시작"):
     if vod_url:
         chat_logs, chat_counts, laugh_counts = crawl_chats(vod_url)
 
+        st.session_state['chat_logs'] = chat_logs
+        st.session_state['chat_counts'] = chat_counts
+        st.session_state['laugh_counts'] = laugh_counts
+
         # 파일로 저장
         file_name = "chat_logs.txt"
         with open(file_name, "w") as file:
@@ -110,53 +121,52 @@ if st.button("크롤링 시작"):
                 file_name=file_name,
                 mime="text/plain"
             )
-
-        # 그래프 출력
-        if chat_counts and laugh_counts:
-            # 데이터프레임 생성
-            times = [time.strftime('%H:%M:%S') for time in chat_counts.keys()]
-            chat_numbers = list(chat_counts.values())
-            laugh_numbers = [laugh_counts.get(time, 0) for time in chat_counts.keys()]
-            df = pd.DataFrame({'시간': times, '전체 채팅 개수': chat_numbers, 'ㅋㅋㅋㅋ 채팅 개수': laugh_numbers})
-
-            # Plotly 선 그래프 그리기
-            fig = go.Figure()
-
-            # 전체 채팅 개수 선 그래프 추가
-            fig.add_trace(go.Scatter(
-                x=df['시간'],
-                y=df['전체 채팅 개수'],
-                mode='lines',  # 마커 없이 선만 표시
-                name='전체 채팅 개수',
-                line=dict(color='blue'),
-                hovertemplate='%{x} - 전체 채팅 개수: %{y}<extra></extra>'
-            ))
-
-            # ㅋㅋㅋㅋ 채팅 개수 선 그래프 추가
-            fig.add_trace(go.Scatter(
-                x=df['시간'],
-                y=df['ㅋㅋㅋㅋ 채팅 개수'],
-                mode='lines',  # 마커 없이 선만 표시
-                name='ㅋㅋㅋㅋ 채팅 개수',
-                line=dict(color='red'),
-                hovertemplate='%{x} - ㅋㅋㅋㅋ 채팅 개수: %{y}<extra></extra>'
-            ))
-
-            # 그래프 레이아웃 설정
-            fig.update_layout(
-                title="분당 채팅 및 ㅋㅋㅋㅋ 채팅 개수",
-                xaxis_title="시간",
-                yaxis_title="채팅 개수",
-                xaxis=dict(
-                    showticklabels=True  # x축 시간 레이블 표시
-                ),
-                hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
-                showlegend=True,  # 범례 표시
-                margin=dict(l=50, r=50, t=100, b=100)  # 그래프 아래쪽 여백을 조정
-            )
-
-            # 그래프 출력
-            st.plotly_chart(fig)
-
     else:
-        st.warning("URL을 입력하세요.")
+        st.warning("URL을 입력하세요.")
+
+# 이전에 크롤링한 결과가 있으면 그래프와 다운로드 버튼 표시
+if st.session_state['chat_logs']:
+    # 데이터프레임 생성
+    times = [time.strftime('%H:%M:%S') for time in st.session_state['chat_counts'].keys()]
+    chat_numbers = list(st.session_state['chat_counts'].values())
+    laugh_numbers = [st.session_state['laugh_counts'].get(time, 0) for time in st.session_state['chat_counts'].keys()]
+    df = pd.DataFrame({'시간': times, '전체 채팅 개수': chat_numbers, 'ㅋㅋㅋㅋ 채팅 개수': laugh_numbers})
+
+    # Plotly 선 그래프 그리기
+    fig = go.Figure()
+
+    # 전체 채팅 개수 선 그래프 추가
+    fig.add_trace(go.Scatter(
+        x=df['시간'],
+        y=df['전체 채팅 개수'],
+        mode='lines',  # 마커 없이 선만 표시
+        name='전체 채팅 개수',
+        line=dict(color='blue'),
+        hovertemplate='%{x} - 전체 채팅 개수: %{y}<extra></extra>'
+    ))
+
+    # ㅋㅋㅋㅋ 채팅 개수 선 그래프 추가
+    fig.add_trace(go.Scatter(
+        x=df['시간'],
+        y=df['ㅋㅋㅋㅋ 채팅 개수'],
+        mode='lines',  # 마커 없이 선만 표시
+        name='ㅋㅋㅋㅋ 채팅 개수',
+        line=dict(color='red'),
+        hovertemplate='%{x} - ㅋㅋㅋㅋ 채팅 개수: %{y}<extra></extra>'
+    ))
+
+    # 그래프 레이아웃 설정
+    fig.update_layout(
+        title="분당 채팅 및 ㅋㅋㅋㅋ 채팅 개수",
+        xaxis_title="시간",
+        yaxis_title="채팅 개수",
+        xaxis=dict(
+            showticklabels=True  # x축 시간 레이블 표시
+        ),
+        hovermode="x unified",  # 마우스를 올렸을 때 해당 x축에서 툴팁 표시
+        showlegend=True,  # 범례 표시
+        margin=dict(l=50, r=50, t=100, b=100)  # 그래프 아래쪽 여백을 조정
+    )
+
+    # 그래프 출력
+    st.plotly_chart(fig)