From 5633019411de09d5f71d63cae3688b13a82ee72f Mon Sep 17 00:00:00 2001
From: s444417 <mikolaj12341230@gmail.com>
Date: Mon, 12 Jun 2023 14:17:16 +0200
Subject: [PATCH] fix path issue

---
 .env_template                                        |  4 ++++
 .../{requirements.txt => requirements_old.txt}       |  0
 ai_talks/chat.py                                     |  9 ++++++++-
 ai_talks/src/utils/conversation.py                   | 12 +++++++++---
 requirements.txt                                     | 11 ++++++++++-
 5 files changed, 31 insertions(+), 5 deletions(-)
 create mode 100644 .env_template
 rename ai_talks/AMUseBotBackend/{requirements.txt => requirements_old.txt} (100%)

diff --git a/.env_template b/.env_template
new file mode 100644
index 0000000..d54ea95
--- /dev/null
+++ b/.env_template
@@ -0,0 +1,4 @@
+RECIPE_PATH=AMUseBotFront/ai_talks/AMUseBotBackend/recipe/ 
+DIALOG_PATH=AMUseBotFront/ai_talks/AMUseBotBackend/dialog/
+INTENT_DICT_PATH=ai_talks/AMUseBotBackend/utils/intent_dict.json
+MODEL_IDENTIFIER_PATH=ai_talks/AMUseBotBackend/models/NLU/roberta-base-cookdial.txt
\ No newline at end of file
diff --git a/ai_talks/AMUseBotBackend/requirements.txt b/ai_talks/AMUseBotBackend/requirements_old.txt
similarity index 100%
rename from ai_talks/AMUseBotBackend/requirements.txt
rename to ai_talks/AMUseBotBackend/requirements_old.txt
diff --git a/ai_talks/chat.py b/ai_talks/chat.py
index 56fb34c..e38fd4a 100644
--- a/ai_talks/chat.py
+++ b/ai_talks/chat.py
@@ -10,6 +10,8 @@ from PIL import Image
 from src.utils.conversation import get_user_input, show_chat_buttons, show_conversation
 from src.utils.lang import en
 
+import os
+from dotenv import load_dotenv
 
 if __name__ == '__main__':
     
@@ -37,6 +39,11 @@ if __name__ == '__main__':
     with open(css_file) as f:
         st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
 
+    load_dotenv()
+
+    DIALOG_PATH = os.getenv('DIALOG_PATH')
+    RECIPE_PATH = os.getenv('RECIPE_PATH')
+
     # Storing The Context
     if "locale" not in st.session_state:
         st.session_state.locale = en
@@ -57,7 +64,7 @@ if __name__ == '__main__':
     if "total_tokens" not in st.session_state:
         st.session_state.total_tokens = []
     if "dst" not in st.session_state:
-        st.session_state.dst = DST(recipe_path="AMUseBotFront/ai_talks/AMUseBotBackend/recipe/", dialog_path="AMUseBotFront/ai_talks/AMUseBotBackend/dialog/")
+        st.session_state.dst = DST(recipe_path=RECIPE_PATH, dialog_path=DIALOG_PATH)
     if "dp" not in st.session_state:
         st.session_state.dp = DP(dst=st.session_state.dst)
 
diff --git a/ai_talks/src/utils/conversation.py b/ai_talks/src/utils/conversation.py
index 28f2c46..67c9340 100644
--- a/ai_talks/src/utils/conversation.py
+++ b/ai_talks/src/utils/conversation.py
@@ -7,12 +7,18 @@ from streamlit_chat import message
 from .stt import show_voice_input
 from .tts import show_audio_player
 
-from AMUseBotBackend.src.DP.dp import DP
-from AMUseBotBackend.src.DST.dst import DST
 from AMUseBotBackend.src.NLU.nlu import NLU
 
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+INTENT_DICT_PATH = os.getenv('INTENT_DICT_PATH')
+MODEL_IDENTIFIER_PATH = os.getenv('MODEL_IDENTIFIER_PATH')
+
 @st.cache_resource
-def get_nlu_model(intent_dict_path = 'AMUseBotFront/ai_talks/AMUseBotBackend/utils/intent_dict.json', model_identifier_path = 'AMUseBotFront/ai_talks/AMUseBotBackend/models/NLU/roberta-base-cookdial.txt'):
+def get_nlu_model(intent_dict_path = INTENT_DICT_PATH, model_identifier_path = MODEL_IDENTIFIER_PATH):
     return NLU(intent_dict_path=intent_dict_path,
                             model_identifier_path=model_identifier_path)
 
diff --git a/requirements.txt b/requirements.txt
index dbe4ec8..86b9ba2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,13 @@ streamlit-bokeh-events>=0.1.2
 watchdog>=3.0.0
 setuptools~=65.5.0
 graphviz~=0.20.1
-Pillow~=9.5.0
\ No newline at end of file
+Pillow~=9.5.0
+pandas==1.5.2
+scikit_learn==1.2.0
+simpletransformers==0.63.9
+torch==1.11.0
+spacy==3.5.0
+rank_bm25==0.2.2
+tqdm==4.64.1
+nlp==0.4.0
+python-dotenv==1.0.0
\ No newline at end of file