s444354/AMUseBot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

integrate chatbot

Browse Source
This commit is contained in:
s444417 2023-05-31 21:11:09 +02:00
parent 0d3b7d0a31
commit 8a3669e2e6
3 changed files with 69 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ai_talks/AMUseBotBackend Submodule

@ -0,0 +1 @@
Subproject commit 01d547dc221b2bd81ccbe24da37b792e9b176b37

23
ai_talks/chat.py
View File

@ -1,12 +1,19 @@
from pathlib import Path from pathlib import Path
from random import randrange from random import randrange
from AMUseBotBackend.src.DP.dp import DP
from AMUseBotBackend.src.DST.dst import DST
from AMUseBotBackend.src.NLU.nlu import NLU
import graphviz import graphviz
import streamlit as st import streamlit as st
from PIL import Image from PIL import Image
from src.utils.conversation import get_user_input, show_chat_buttons, show_conversation from src.utils.conversation import get_user_input, show_chat_buttons, show_conversation
from src.utils.lang import en from src.utils.lang import en
if __name__ == '__main__':
# --- PATH SETTINGS --- # --- PATH SETTINGS ---
current_dir: Path = Path(__file__).parent if "__file__" in locals() else Path.cwd() current_dir: Path = Path(__file__).parent if "__file__" in locals() else Path.cwd()
css_file: Path = current_dir / "src/styles/.css" css_file: Path = current_dir / "src/styles/.css"
@ -50,7 +57,13 @@ if "costs" not in st.session_state:
st.session_state.costs = [] st.session_state.costs = []
if "total_tokens" not in st.session_state: if "total_tokens" not in st.session_state:
st.session_state.total_tokens = [] 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/")
if "dp" not in st.session_state:
st.session_state.dp = DP(dst=st.session_state.dst)
if "nlu" not in st.session_state:
st.session_state.nlu = NLU(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 show_graph(): def show_graph():
# Create a graphlib graph object # Create a graphlib graph object
@ -66,14 +79,12 @@ def show_graph():
graph.edge(st.session_state.generated[x], st.session_state.past[x+1]) graph.edge(st.session_state.generated[x], st.session_state.past[x+1])
except: except:
pass pass
st.graphviz_chart(graph) st.graphviz_chart(graph)
def main() -> None: def main() -> None:
c1, c2 = st.columns(2) c1, c2 = st.columns(2)
with c1, c2: with c1, c2:
c1.selectbox(label=st.session_state.locale.select_placeholder1, key="model", options=AI_MODEL_OPTIONS)
st.session_state.input_kind = c2.radio( st.session_state.input_kind = c2.radio(
label=st.session_state.locale.input_kind, label=st.session_state.locale.input_kind,
options=(st.session_state.locale.input_kind_1, st.session_state.locale.input_kind_2), options=(st.session_state.locale.input_kind_1, st.session_state.locale.input_kind_2),
@ -84,11 +95,10 @@ def main() -> None:
options=(st.session_state.locale.radio_text1, st.session_state.locale.radio_text2), options=(st.session_state.locale.radio_text1, st.session_state.locale.radio_text2),
horizontal=True, horizontal=True,
) )
match role_kind: if role_kind == st.session_state.locale.radio_text1:
case st.session_state.locale.radio_text1:
c2.selectbox(label=st.session_state.locale.select_placeholder2, key="role", c2.selectbox(label=st.session_state.locale.select_placeholder2, key="role",
options=st.session_state.locale.ai_role_options) options=st.session_state.locale.ai_role_options)
case st.session_state.locale.radio_text2: elif role_kind == st.session_state.locale.radio_text2:
c2.text_input(label=st.session_state.locale.select_placeholder3, key="role") c2.text_input(label=st.session_state.locale.select_placeholder3, key="role")
if st.session_state.user_text: if st.session_state.user_text:
@ -96,7 +106,6 @@ def main() -> None:
show_conversation() show_conversation()
get_user_input() get_user_input()
show_chat_buttons() show_chat_buttons()

13
ai_talks/src/utils/conversation.py
View File

@ -65,10 +65,17 @@ def show_conversation() -> None:
{"role": "user", "content": st.session_state.user_text}, {"role": "user", "content": st.session_state.user_text},
] ]
ai_content = "Dummy respone from AI" # ai_content = "Dummy respone from AI"
intents = st.session_state.nlu.predict(st.session_state.user_text)
st.session_state.dst.update_dialog_history(
system_message='',
user_message=st.session_state.user_text,
intents=intents,
)
system_message = st.session_state.dp.generate_response(intents)
# delete random before deploying with our model # delete random before deploying with our model
random_str = ''.join(choices(string.ascii_uppercase + string.digits, k=5)) #random_str = ''.join(choices(string.ascii_uppercase + string.digits, k=5))
ai_content += random_str ai_content = system_message
st.session_state.messages.append({"role": "assistant", "content": ai_content}) st.session_state.messages.append({"role": "assistant", "content": ai_content})
if ai_content: if ai_content:
show_chat(ai_content, st.session_state.user_text) show_chat(ai_content, st.session_state.user_text)