From 5641552a385ae7434df0205e88767115726a6b7a Mon Sep 17 00:00:00 2001
From: Karol Cyganik <s495715@labs.wmi.amu.edu.pl>
Date: Tue, 16 Apr 2024 12:28:58 +0200
Subject: [PATCH] add mock model

---
 dialog_model.py | 69 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 dialog_model.py

diff --git a/dialog_model.py b/dialog_model.py
new file mode 100644
index 0000000..270b37e
--- /dev/null
+++ b/dialog_model.py
@@ -0,0 +1,69 @@
+from typing import Any
+
+
+class Model():
+    def __init__(self):
+        pass
+
+    def __call__(self, prompt) -> Any:
+        nlu = NLU()
+        dst = DST()
+        dp = DP()
+        nlg = NLG()
+
+        msg = prompt
+
+        rama_nlu = nlu(msg)
+        rama_dst = dst(rama_nlu)
+        rama_dp = dp(rama_dst)
+        text = nlg(rama_dp)
+
+        return text
+
+
+class NLU():
+    def __init__(self):
+        pass
+
+    def __call__(self, prompt) -> Any:
+        msg = prompt
+        if "imie" in msg:
+            return "jakie imie"
+
+
+class DST():
+    def __init__(self):
+        self.msgs = []
+
+    def __call__(self, msg) -> Any:
+        self.msgs.append(msg)
+        return msg
+
+
+class DP():
+    def __init__(self):
+        pass
+
+    def __call__(self, msg) -> Any:
+        if "imie" in msg:
+            return "imieMSG"
+        else:
+            return None
+
+
+class NLG():
+    def __init__(self):
+        pass
+
+    def __call__(self, msg) -> Any:
+        if msg == "imieMSG":
+            return "Mam na imie JARVIS"
+        else:
+            return "Nie rozumiem"
+
+
+if __name__ == "__main__":
+    model = Model()
+
+    ans = model(prompt="Jak masz na imie")
+    print(ans)