add chatbot

app.py

@@ -16,15 +16,23 @@ def main():
     chat_history = []
     context = "You are an helpfully assistant in a school. You are helping a student with his homework."
     chat = llm.get_chat(context=context)
+    user_input = st.text_input("User:")
+    button = st.button("Send")
+    chat_area = st.empty()
+
     while True:
-        user_input = st.text_input("User:")
-        if st.button("Send"):
+        print(user_input)
+        if button:
             if user_input:
                 chat_history.append(("User", user_input))
                 bot_response = chat.answerStoreHistory(qn=user_input)
                 chat_history.append(("Bot", bot_response))
+                print(chat_history)
+
+            chat_area.text(format_chat_history(chat_history))
 
-        st.text_area("Chat History:", value=format_chat_history(chat_history), readonly=True)
+if __name__ == "__main__":
+    main()
 
 
 

chatbot.py

@@ -0,0 +1,44 @@
+import streamlit as st
+from streamlit_chat import message
+
+from LLM import LLM
+
+st.set_page_config(
+    page_title="Streamlit Chat - Demo",
+    page_icon=":robot:"
+)
+
+# API_URL = "https://api-inference.huggingface.co/models/facebook/blenderbot-400M-distill"
+# headers = {"Authorization": st.secrets['api_key']}
+
+st.header("Streamlit Chat - Demo")
+# st.markdown("[Github](https://github.com/ai-yash/st-chat)")
+
+generated = []
+past = []
+
+def query(chat, payload):
+    response = chat.answerStoreHistory(qn=payload)
+    return response
+
+
+def get_text():
+    input_text = st.text_input("You: ", "Hello, how are you?", key="input")
+    return input_text
+
+
+user_input = get_text()
+llm = LLM("gpt2")
+chat = llm.get_chat(context="You are an helpfully assistant in a school. You are helping a student with his homework.")
+
+if user_input:
+    output = chat.answerStoreHistory(qn=user_input)
+
+    past.append(user_input)
+    generated.append(output)
+
+if generated:
+
+    for i in range(len(generated) - 1, -1, -1):
+        message(generated[i], key=str(i))
+        message(past[i], is_user=True, key=str(i) + '_user')

debug.py

@@ -0,0 +1,4 @@
+from streamlit.web import bootstrap
+
+real_script = 'app.py'
+bootstrap.run(real_script, f'run.py {real_script}', [], {})