Spaces:
Running
Running
File size: 3,191 Bytes
876403c 7d61140 eea4f3c 876403c eea4f3c a8dc9d8 eea4f3c a8dc9d8 eea4f3c 876403c 7d61140 876403c eea4f3c a8dc9d8 7d61140 a8dc9d8 7d61140 a8dc9d8 7d61140 a8dc9d8 876403c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import streamlit as st
import diary as d
import sentiment_parser as sp
import network_builder as nb
from pyvis.network import Network
import streamlit.components.v1 as components
import word_transformations as wt
st.title('Автоматический аннотатор')
# st.markdown("Скопируйте текст дневика в это поле или выберите для теста один из подготовленных отрывков.")
txt = st.text_area('Скопируйте текст дневника в это поле', height=100)
option = st.selectbox(
'Или выберите один из пробных текстов дневников:',
('Выбрать...', 'Анатолий Василивицкий', 'Мария Германова'))
if option == 'Анатолий Василивицкий':
with open('test_1.txt', 'r') as f:
txt = f.read()
elif option == 'Мария Германова':
with open('test.txt', 'r') as f:
txt = f.read()
# diary = st.text_area('Текст дневника')
if st.button('Обработать') and txt != '':
try:
df = d.analyze(txt)
graph = nb.build_graph(df)
GENDER = wt.get_gender(df['tokens'])
locations = df['locations']
st.markdown(f'**Аннотация этого дневника:** {nb.annotation(graph, GENDER, locations)}')
textnet = Network( height='400px',
width='100%',
bgcolor='white',
font_color='black'
)
textnet.from_nx(graph)
textnet.repulsion(
node_distance=420,
central_gravity=0.33,
spring_length=110,
spring_strength=0.10,
damping=0.95
)
try:
path = '/tmp'
textnet.save_graph(f'{path}/pyvis_graph.html')
HtmlFile = open(f'{path}/pyvis_graph.html', 'r', encoding='utf-8')
except:
path = '/html_files'
textnet.save_graph(f'{path}/pyvis_graph.html')
HtmlFile = open(f'{path}/pyvis_graph.html', 'r', encoding='utf-8')
st.markdown('### «Граф знания» этого дневника')
st.caption(':blue[Синим цветом] выделены узлы, связанные с одной датированной дневниковой записью, розовым — общий сентимент записи, :red[красным] — найденное утверждение, :green[зелёным] — места и локации, фиолетовым — занятия.')
st.caption('Чтобы увеличить граф и посмотреть лейблы узлов, установите курсор в нужном месте и проскорольте вниз.')
components.html(HtmlFile.read(), height=435)
except:
st.warning('Вставьте текст дневника, который начинается с даты!', icon="⚠️")
|