Upload 17 files

activities/instructions.json

@@ -0,0 +1,130 @@
+{
+    "obl": [
+        [
+            [
+                "obl",
+                "amod",
+                0
+            ],
+            [
+                "obl",
+                "case",
+                0
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "obl",
+                1
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "advmod",
+                0
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "nsubj",
+                0
+            ]
+        ]
+    ],
+    "root": [
+        [
+            [
+                "root",
+                "case",
+                0
+            ],
+            [
+                "root",
+                "cop",
+                0
+            ],
+            [
+                "root",
+                "advmod",
+                0
+            ]
+        ]
+    ],
+    "nmod": [
+        [
+            [
+                "nmod",
+                "case",
+                0
+            ],
+            [
+                [
+                    "obj"
+                ],
+                "nmod",
+                1
+            ],
+            [
+                "obj",
+                "amod",
+                0
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "obj",
+                1
+            ]
+        ],
+        [
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "nmod",
+                1
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "case",
+                0
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "cop",
+                0
+            ],
+            [
+                [
+                    "root",
+                    "conj",
+                    "ccomp"
+                ],
+                "advmod",
+                0
+            ]
+        ]
+    ]
+}

activities/words.txt

@@ -0,0 +1,231 @@
+мяч
+промысел
+рамс
+фараон
+гарпастон
+матанализ
+Зарница
+пятнашки
+утеха
+попрошайничество
+лабораторка
+кегли
+регби
+марьяж
+мацзян
+козёл
+индюшатина
+расшибец
+мигалка
+пато
+кикер
+соккер
+гандбол
+представление
+садоводство
+маджонг
+стрелялка
+такен
+пересечка
+сифа
+кулер
+мастер-класс
+волейбол
+договорняк
+футбол
+безик
+встреча
+купкари
+шарокат 
+теннис
+ролевая игра
+колечко
+кастплей
+бескрылка
+картёж
+гарпастум
+расшибалка
+упражнение
+гали
+эйрсофт
+крокет
+развлечение
+скорняжничество
+тренинг
+суета
+ужение
+шалость
+баскетбол
+рапид
+семинар
+бадминтон
+пряталки
+пионербол
+тренировка
+меледа
+сифак
+юй-мао-цю
+дартс
+тетрис
+расшибалочка
+расшибай
+штосс
+мобильная игра
+пинг-понг
+бейсбол
+стратегия
+иняз
+чеканка
+настолка
+прятки
+развлекаловка
+мастерство
+фант
+уженье
+стирки
+хобби
+меккано
+бочче
+бонки
+юй-мао-чю
+нуда
+экзерсис
+шляпа
+сиф
+коллекционерство
+шахматы
+поддавки
+игра
+кок-бору
+инди-игра
+компьютерная игра
+кроссворд
+лицедейство
+бильярд
+замеряшки
+биллиард
+ремесло
+рукомесло
+ловля
+индюшка
+квачик
+практикум
+ляпки
+пулька
+выезд
+пасьянс
+кошки-мышки
+начальствование
+пиньята
+шарада
+экзерцис
+вист
+дротики
+шашки
+рутина
+биллиард форма
+домино
+козлодрание
+охота
+штос
+свайка
+гулючки
+бузкаши
+шитьё
+лакросс
+ролевушка
+суперигр
+расшибаловка
+арпеджирование
+зарядка
+пазл
+соображалка
+козны
+управление
+беспокойство
+виселица
+горелки
+развлекуха
+симулирование
+фрисби
+пошив
+головоломка
+орлянка
+жмурки
+перебор
+чика
+комикование
+шарады
+мозаика
+лякросс
+хоккей
+видеоигра
+логогриф
+прятушки
+коллоквиум
+лапта
+гусарик
+рассеяние
+карамболь
+крестословица
+косплей
+катуха
+канитель
+матан
+картёжничание
+хлопоты
+рэндзю
+матч
+стукалка
+догоняшки
+развлекалово
+бильбоке
+ролевуха
+расшиши
+классики
+страйкбол
+пара
+городки
+рыбалка
+пристенок
+триктрак
+салки
+потеха
+суетня
+кокпар
+преферанс
+салочки
+куличик
+бабки
+догонялки
+лаун-теннис
+спектакль
+лов
+профессия
+кости
+морской бой
+квач
+нарды
+бинго
+пеструха
+буриме
+баловство
+пушбол
+прессинг
+чехарда
+забава
+нищенство
+бутылочка
+шутер
+очко
+ералаш
+серсо
+гляделки
+радиоигра
+бег
+ролёвка
+фусбол
+кёрлинг
+мушкетеры
+мушкетёры
+квн

diary.py

@@ -21,6 +21,7 @@ def analyze(text):
     
     # # Выделение фактов из текста
     diary['locations'] = diary['tokens'].apply(lambda tokens: cp.get_facts(tokens, 'locations'))
+    diary['activities'] = diary['tokens'].apply(lambda tokens: cp.get_facts(tokens, 'activities'))
 
     # Определение сентимента по записям
     diary['sent'] = diary['tokens'].apply(lambda tokens: sp.get_overall_sentiment(tokens))

network_builder.py

@@ -17,11 +17,18 @@ def build_graph(df):
     # Добавление связей дат и фактов, фактов и слов
     for index, row in df[['date_start', 'locations']].iterrows():
         for fact in row['locations']:
-            G.add_node(fact[0], group="Category_word", color = "green")
+            G.add_node(fact[0], group="Location", color = "green")
             G.add_node(fact[1], group="Fact", color = "red")
             G.add_edge(row['date_start'], fact[1])
             G.add_edge(fact[0], fact[1])
 
+    for index, row in df[['date_start', 'activities']].iterrows():
+        for fact in row['activities']:
+            G.add_node(fact[0], group="Activity", color = "purple")
+            # G.add_node(fact[1], group="Fact", color = "red")
+            G.add_edge(row['date_start'], fact[0])
+            # G.add_edge(fact[0], fact[1])
+
     # Добавление связей даты записи с сентиментом
     for index, row in df[['date_start', 'sent_index']].iterrows():
         G.add_node(sp.get_most_sentiment([row['sent_index']]), group="Sentiment", color = "pink")
@@ -44,10 +51,17 @@ def dates_of_Diary_writing(G):
 def most_visited_places(G):
     res = Counter()
     for key, value in G.nodes.data():
-        if value.get('group') == 'Category_word':
+        if value.get('group') == 'Location':
             res[key] = len(G[key])
     return [i[0] for i in Counter(res).most_common(3)]
 
+def most_activity(G):
+    res = Counter()
+    for key, value in G.nodes.data():
+        if value.get('group') == 'Activity':
+            res[key] = len(G[key])
+    return [i[0] for i in Counter(res).most_common(1)]
+
 
 def facts_for_annotation(G, gender, most_places):
     '''
@@ -55,7 +69,7 @@ def facts_for_annotation(G, gender, most_places):
     '''
     res = Counter()
     for key, value in G.nodes.data():
-            if value.get('group') == 'Category_word':
+            if value.get('group') == 'Location':
                 res[key] = len(G[key])
     facts = [s for i in Counter(res).keys() for s in G.successors(i)]
     
@@ -113,10 +127,15 @@ def annotation(G, gender, locations):
     most_places = most_visited_places(G)
     facts = facts_for_annotation(G, gender, most_places)
     sentiment = sentiment_of_date(G)
+    activities = most_activity(G)
 
     # facts = ', '.join([f"{fact[1].lower()} ({fact[0]})" for fact in facts])
     # facts = ''
+    if activities:
+        act = f'В дневнике также упоминаниется игра в {activities[0]}.'
+    else:
+        act = ''
 
-    annotation = f'{wt.get_noun(gender).title()} этого дневника {wt.gender_transformer("вести", gender)} его с {dates[0]} по {dates[1]}. Наиболее часто {wt.get_pronoun(gender)} {wt.gender_transformer("описывал", gender)} {wt.inflector(most_places[0], "accs")}, {wt.inflector(most_places[1], "accs")} и {wt.inflector(most_places[2], "accs")}.\n\n{constuct_fact_for_annotation(facts, sentiment, gender, locations)}'
+    annotation = f'{wt.get_noun(gender).title()} этого дневника {wt.gender_transformer("вести", gender)} его с {dates[0]} по {dates[1]}. Наиболее часто {wt.get_pronoun(gender)} {wt.gender_transformer("описывал", gender)} {wt.inflector(most_places[0], "accs")}, {wt.inflector(most_places[1], "accs")} и {wt.inflector(most_places[2], "accs")}. {act}\n\n{constuct_fact_for_annotation(facts, sentiment, gender, locations)}'
 
     return annotation

test_1.txt

@@ -1,3 +1,7 @@
+1940
+
+26 июня. Очень долго не писал. Это правда. Каждый день почти одинаковый. Сегодня был в лесу. Играю в шахматы. У меня 3 ½ очка, а у моего противника 4 ½ оч[ка]. Сейчас я отдыхаю и встал чтобы написать.
+
 1943
 
 24 марта. Челябинск