Update app.py

app.py

@@ -1,6 +1,10 @@
 import streamlit as st
-import requests
+import pandas as pd
+import plotly.graph_objects as go
+from collections import defaultdict
+import datetime
 import re
+import requests
 
 # Streamlit 제목 및 설명
 st.title("VOD 채팅 크롤러")
@@ -29,10 +33,9 @@ def crawl_chats(vod_url):
         "previousVideoChatSize": 50  # 가져올 채팅 메시지 수
     }
 
-    # 마지막 수집한 채팅 메시지 저장 변수
-    last_collected_chats = None
-    total_collected_chats = 0
     chat_logs = []
+    chat_counts = defaultdict(int)
+    laugh_counts = defaultdict(int)
 
     # 채팅 데이터를 순차적으로 요청하여 가져오기
     while True:
@@ -41,7 +44,7 @@ def crawl_chats(vod_url):
 
         # 요청 결과가 성공적이지 않을 경우 종료
         if response.status_code != 200:
-            return f"API 요청 실패: {response.status_code}"
+            return f"API 요청 실패: {response.status_code}", None, None
 
         # JSON 데이터로 변환
         data = response.json()
@@ -53,10 +56,6 @@ def crawl_chats(vod_url):
         if not chats:
             break
 
-        # 현재 수집된 채팅이 이전에 수집된 채팅과 동일한지 확인
-        if last_collected_chats == chats:
-            break
-
         # 채팅 메시지(content)와 playerMessageTime(채팅 시간)만 수집
         for chat in chats:
             chat_content = chat.get("content")  # 채팅 메시지 내용
@@ -76,11 +75,14 @@ def crawl_chats(vod_url):
                 # 채팅 로그에 추가
                 chat_logs.append(f"{formatted_time} - {chat_content}")
 
-        # 마지막으로 수집한 채팅 메시지 업데이트
-        last_collected_chats = chats
+                # 시간대별로 채팅 개수 계산
+                time_obj = datetime.datetime.strptime(formatted_time, '%H:%M:%S')
+                minute_key = time_obj.replace(second=0)  # 분 단위로 변환하여 집계
+                chat_counts[minute_key] += 1
 
-        # 채팅 데이터 개수 업데이트 및 출력
-        total_collected_chats += len(chats)
+                # 'ㅋㅋㅋㅋ'가 포함된 채팅 개수 카운트
+                if len(re.findall(r'ㅋ', chat_content)) >= 4:
+                    laugh_counts[minute_key] += 1
 
         # 다음 요청을 위해 playerMessageTime 파라미터 업데이트
         next_time = data["content"].get("nextPlayerMessageTime")
@@ -88,15 +90,13 @@ def crawl_chats(vod_url):
             break
         params["playerMessageTime"] = next_time
 
-    # 결과 반환
-    return "\n".join(chat_logs)
+    return "\n".join(chat_logs), chat_counts, laugh_counts
 
 # 버튼을 눌렀을 때 채팅 크롤링 시작
 if st.button("크롤링 시작"):
     if vod_url:
-        chat_logs = crawl_chats(vod_url)
-        st.text_area("채팅 로그", value=chat_logs, height=400)
-        
+        chat_logs, chat_counts, laugh_counts = crawl_chats(vod_url)
+
         # 파일로 저장
         file_name = "chat_logs.txt"
         with open(file_name, "w") as file:
@@ -110,5 +110,53 @@ 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을 입력하세요.")