adding chat bot

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 # Custom imports 
 from multipage import MultiPage
-from pages import text_gen, fill_mask, home_page, translation # import your pages here
+from pages import text_gen, fill_mask, home_page, translation, chat # import your pages here
 
 # Create an instance of the app 
 app = MultiPage()
@@ -9,6 +9,7 @@ app = MultiPage()
 
 # Add all your applications (pages) here
 app.add_page("Home Page", home_page.app)
+app.add_page("Chat Bot", chat.app)
 app.add_page("Mask Fill", fill_mask.app)
 app.add_page("Text Generation", text_gen.app)
 app.add_page("Translation", translation.app)

pages/chat.py

@@ -0,0 +1,42 @@
+import streamlit as st
+from streamlit_chat import message as st_message
+from transformers import BlenderbotTokenizer
+from transformers import BlenderbotForConditionalGeneration
+
+
+def app():
+    @st.experimental_singleton
+    def get_models():
+        # it may be necessary for other frameworks to cache the model
+        # seems pytorch keeps an internal state of the conversation
+        model_name = "facebook/blenderbot-400M-distill"
+        tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
+        model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
+        return tokenizer, model
+
+
+    if "history" not in st.session_state:
+        st.session_state.history = []
+
+    st.title("IoT Chat Robot")
+    st.write('Just chat with a friendly and smart AI')
+    st.markdown('## ')
+
+
+    def generate_answer():
+        tokenizer, model = get_models()
+        user_message = st.session_state.input_text
+        inputs = tokenizer(st.session_state.input_text, return_tensors="pt")
+        result = model.generate(**inputs)
+        message_bot = tokenizer.decode(
+            result[0], skip_special_tokens=True
+        )  # .replace("<s>", "").replace("</s>", "")
+
+        st.session_state.history.append({"message": user_message, "is_user": True})
+        st.session_state.history.append({"message": message_bot, "is_user": False})
+
+
+    st.text_input("Hi, nice to meet you AI!", key="input_text", on_change=generate_answer)
+
+    for chat in st.session_state.history:
+        st_message(**chat)  # unpacking

requirements.txt

@@ -4,4 +4,5 @@ altair
 pandas
 sacremoses
 sentencepiece
-torch
+torch
+streamlit_chat