[feat] implement homophone-based pun detector

pingpingy1 · Dec 4, 2023 · 8b7e9fe · 8b7e9fe
1 parent 3b1da67
commit 8b7e9fe
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 2 deletions.
diff --git a/hopre/core.py b/hopre/core.py
@@ -53,7 +53,22 @@ def hopre() -> None:
                         f'The two incompatible interpretations of "{context1}" and "{context2}" are resolved with the homonym "{homonym}"!'
                     )
                 else:
-                    print("I don't think this is a joke...")
+                    result = pthread.query(
+                        f"homophonePun({sentences},Homophone1,Homophone2,Context1,Context2)"
+                    )
+                    if result:
+                        assert isinstance(result, list)
+                        fst = result[0]
+                        context1 = " ".join(fst["Context1"])
+                        context2 = " ".join(fst["Context2"])
+                        homophone1 = " ".join(fst["Homophone1"])
+                        homophone2 = " ".join(fst["Homophone2"])
+                        print("I get it!")
+                        print(
+                            f'The two incompatible interpretations of "{context1}" and "{context2}" are resolved with the homophones "{homophone1}" and "{homophone2}"!'
+                        )
+                    else:
+                        print("I don't think that is a joke...")
 
                 print()
                 if input("Press enter to continue or q to exit...").endswith("q"):

diff --git a/hopre/englishAnalysis/englishLexicon.pl b/hopre/englishAnalysis/englishLexicon.pl
@@ -77,6 +77,10 @@
 
 lexEntry(noun,[symbol:tomato,syntax:[tomato]]).
 lexEntry(noun,[symbol:salad,syntax:[salad]]).
+lexEntry(noun,[symbol:computer,syntax:[computer]]).
+lexEntry(noun,[symbol:doctor,syntax:[doctor]]).
+lexEntry(noun,[symbol:bite,syntax:[bite]]).
+lexEntry(noun,[symbol:byte,syntax:[byte]]).
 
 
 /*========================================================================
@@ -196,7 +200,12 @@
 
 lexEntry(tv,[symbol:have,syntax:[have],inf:inf,num:sg]).
 lexEntry(tv,[symbol:have,syntax:[has],inf:fin,num:sg]).
+lexEntry(tv,[symbol:have,syntax:[have],inf:fin,num:sg]).
 lexEntry(tv,[symbol:have,syntax:[have],inf:fin,num:pl]).
+lexEntry(tv,[symbol:have,syntax:[had],inf:inf,num:sg]).
+lexEntry(tv,[symbol:have,syntax:[had],inf:inf,num:pl]).
+lexEntry(tv,[symbol:have,syntax:[had],inf:fin,num:sg]).
+lexEntry(tv,[symbol:have,syntax:[had],inf:fin,num:pl]).
 
 lexEntry(tv,[symbol:kill,syntax:[kill],inf:inf,num:sg]).
 lexEntry(tv,[symbol:kill,syntax:[kills],inf:fin,num:sg]).
@@ -232,6 +241,15 @@
 lexEntry(tv,[symbol:see,syntax:[saw],inf:fin,num:sg]).
 lexEntry(tv,[symbol:see,syntax:[saw],inf:fin,num:pl]).
 
+lexEntry(tv,[symbol:goto,syntax:[go,to],inf:inf,num:sg]).
+lexEntry(tv,[symbol:goto,syntax:[go,to],inf:fin,num:sg]).
+lexEntry(tv,[symbol:goto,syntax:[goes,to],inf:fin,num:sg]).
+lexEntry(tv,[symbol:goto,syntax:[goes,to],inf:fin,num:pl]).
+lexEntry(tv,[symbol:goto,syntax:[went,to],inf:inf,num:sg]).
+lexEntry(tv,[symbol:goto,syntax:[went,to],inf:inf,num:pl]).
+lexEntry(tv,[symbol:goto,syntax:[went,to],inf:fin,num:sg]).
+lexEntry(tv,[symbol:goto,syntax:[went,to],inf:fin,num:pl]).
+
 
 /*========================================================================
    Copula

diff --git a/hopre/pun_detection.pl b/hopre/pun_detection.pl
@@ -17,4 +17,15 @@
    similar(HomonymPhrase,Context1),
    append([_,Context2,_],AllWords),
    similar(HomonymPhrase,Context2),
-   dissimilar(Context1,Context2).
+   dissimilar(Context1,Context2).
+
+homophonePun(Sentences,Homophone1,Homophone2,Context1,Context2):-
+   allTexts(Sentences),
+   append(Sentences,AllWords),
+   append([_,Homophone1,_],AllWords),
+   homophone(Homophone1,Homophone2),
+   append([_,Context1,_],AllWords),
+   similar(Homophone1,Context1),
+   append([_,Context2,_],AllWords),
+   similar(Homophone2,Context2),
+   dissimilar(Context1,Context2).