Merge pull request #112 from AAA-Intelligence/feature/mood_analyzer

updates pattern recognizer with more categories
AAA-Intelligence · Apr 18, 2018 · cbec670 · cbec670
2 parents 47defb6 + 25603e7
commit cbec670
Show file tree

Hide file tree

Showing 10 changed files with 52 additions and 7 deletions.
diff --git a/bot/model_definitions.py b/bot/model_definitions.py
@@ -42,10 +42,8 @@ class PatternCategory(IntEnum):
     WEATHER = auto()
     COMPLIMENTS = auto()
     PICKUP_LINES = auto()
-    """
-    DATE = auto()
     BOT_HOBBIES = auto()
-    """
+    DATE = auto()
 
 
 Category = Union[MoodCategory, AffectionCategory, PatternCategory]

diff --git a/bot/patterns/BOT_HOBBIES.txt b/bot/patterns/BOT_HOBBIES.txt
@@ -0,0 +1,8 @@
+Was hast du so für Hobbies?
+Was treibst du so in deiner Freizeit?
+Was machst du gerne außerhalb von der Arbeit?
+Was machst du gerne außerhalb von der Studium?
+Was machst du gerne außerhalb von der Schule?
+Wie vertreibst du dir so deine Zeit?
+Was tust du gerne?
+Was machst du gerne?
diff --git a/bot/patterns/BOT_RELIGION.txt b/bot/patterns/BOT_RELIGION.txt
@@ -0,0 +1,10 @@
+Wie stehst du zu Gott?
+Gehst du in die Kirche?
+Bist du gläubig?
+Glaubst du an Allah?
+Glaubst du an Buddha?
+Glaubst du an Gott?
+Was ist für dich der Sinn im Leben?
+Gehst du in den Gottesdienst?
+Betest du?
+Warst du schonmal in einer Moschee?
diff --git a/bot/patterns/DATE.txt b/bot/patterns/DATE.txt
@@ -0,0 +1,9 @@
+Willst du mal mit mir zum Pizza essen?
+Wollen wir mal Essen gehen?
+Wollen wir mal ins Kino gehen?
+Hast du Lust einen gemeinsamen Spaziergang zu machen?
+Wollen wir ein Eis essen gehen?
+Gehen wir mal zusammen ins Schwimmbad?
+Hast du Lust gemeinsam zu Kochen?
+Wollen wir mal zusammen ein Konzert anhören?
+Hast du Lust auf Netflix and Chill?
diff --git a/bot/predefined_answers/BOT_HOBBIES.txt b/bot/predefined_answers/BOT_HOBBIES.txt
@@ -0,0 +1,11 @@
+Ich schaue gerne Fußball.
+In meiner Freizeit spiele ich gerne Fußball.
+Ich bin sehr sportlich.
+Ich mag Musik.
+Ich singe gerne.
+Ich spiele Klavier.
+Manchmal begleite ich mich selber am Klavier.
+Am liebsten spiele ich Stücke von Mozart und Beethoven.
+Mein Lieblingsverein ist der FC Bayern München.
+Malen kann ich gar nicht.
+Ich bin künstlerisch totaal unbegabt.
diff --git a/bot/predefined_answers/BOT_RELIGION.txt b/bot/predefined_answers/BOT_RELIGION.txt
@@ -1,2 +1,3 @@
-Ja Gott ist der beste.
-Ne Gott ist doof..
+Glaube ist nichts für mich.
+Jeder soll glauben, was er möchte.
+Ich bin Atheist, Agnostiker um genau zu sein.
diff --git a/bot/predefined_answers/DATE.txt b/bot/predefined_answers/DATE.txt
diff --git a/bot/predefined_answers/DATE_NEG.txt b/bot/predefined_answers/DATE_NEG.txt
@@ -0,0 +1,3 @@
+Ne sorry..ich mag dich gar nicht.
+Vielleicht ein ander mal!
+Nicht falsch verstehen, aber im Moment möchte ich das nicht.
diff --git a/bot/predefined_answers/DATE_POS.txt b/bot/predefined_answers/DATE_POS.txt
@@ -0,0 +1,3 @@
+Klar sehr gerne sogar!
+Ich dachte schon du fragst gar nicht mehr!! :D
+Oh ja!
diff --git a/bot/predefined_answers/__init__.py b/bot/predefined_answers/__init__.py
@@ -33,9 +33,11 @@ def answers_for_category(category: PatternCategory, request: Request) -> List[
     if category in cache:
         return cache[category]
     if category.name in 'MOOD PICKUP_LINES':
-        direction = "_POS" if request.mood >= 0 else  "_NEG"
-    elif category.name in 'AFFECTION':  #DATES':
+        direction = "_POS" if request.mood >= 0 else "_NEG"
+    elif category.name in 'AFFECTION':
         direction = "_POS" if request.affection >= 0 else "_NEG"
+    elif category.name in 'DATES':
+        direction = "_POS" if request.affection >= 0.5 else "_NEG"
     p = Path(dir, '%s%s.txt' % (category.name, direction))
 
     if not p.is_file():