Emil25 commited on
Commit
9352c11
·
1 Parent(s): 57a1384

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -6,6 +6,7 @@ from transformers import pipeline
6
  import matplotlib.pyplot as plt
7
  import seaborn as sns
8
  import os
 
9
 
10
  st.title('Анализатор комментариев :red[YouTube] :sunglasses:')
11
 
@@ -22,9 +23,17 @@ st.sidebar.markdown('# Меню')
22
  API_key = os.getenv("api_key_youtube") #st.sidebar.text_input('YouTube API KEY')
23
  st.sidebar.markdown('***')
24
 
25
- # Получаем id видеоролика для отправки запроса
26
- vidID = st.sidebar.text_input('Video Id')
27
- st.sidebar.write('Вы ввели... ', vidID )
 
 
 
 
 
 
 
 
28
  st.sidebar.markdown('***')
29
 
30
  btn_start = st.sidebar.button('Загрузить')
 
6
  import matplotlib.pyplot as plt
7
  import seaborn as sns
8
  import os
9
+ import re
10
 
11
  st.title('Анализатор комментариев :red[YouTube] :sunglasses:')
12
 
 
23
  API_key = os.getenv("api_key_youtube") #st.sidebar.text_input('YouTube API KEY')
24
  st.sidebar.markdown('***')
25
 
26
+ # Получаем id видеоролика из URL для отправки запроса
27
+ pattern = r"(?<=v=)[\w-]+(?=&|\b)"
28
+ url = st.sidebar.text_input('URL YouTube')
29
+ match = re.search(pattern, url)
30
+ if match:
31
+ vidID = match.group()
32
+ st.success('URL', icon="✅")
33
+ else:
34
+ st.warning('URL', icon="⚠️")
35
+ vidID = 'KJA9A1q9l7E'
36
+
37
  st.sidebar.markdown('***')
38
 
39
  btn_start = st.sidebar.button('Загрузить')