frontend chat

app.py

@@ -1,4 +1,105 @@
 import streamlit as st
+import requests
+import simplejson as json
+
+
+URL = "http://localhost:7094/"
+
+full_response = ""
+message_placeholder = st.empty()
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+    st.session_state.conversation_id = None
+    st.session_state.history = []
+
+
+def random_response(
+        message,
+    ):
+    
+    params = {
+        'question': message[-1]['content'],
+        # 'conversation_id': Optional[str] = None,
+    }
+
+
+    question = {
+        'message': message[-1]['content'],
+    }
+
+    request_session = requests.Session()
+    stream = request_session.post(
+        f"{URL}stream",
+        params=params,
+        json=question,
+    )
+
+    for response in stream.iter_lines():
+        try:
+            response = json.loads(response.decode('utf-8'))
+        except json.decoder.JSONDecodeError:
+            continue
+        yield response
+    
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# Accept user input
+if prompt := st.chat_input("How can I help?"):
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    # Display user message in chat message container
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        message_placeholder = st.empty()
+        full_response = ""
+
+
+if len(st.session_state.messages) > 0:
+
+    for response in random_response(
+        # model=st.session_state["openai_model"],
+        message=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
+        # conversation_id=st.session_state.conversation_id,
+        # user=user,
+        # chat_type=chat_options[chat_option],
+        # stream=True,
+    ):
+        if 'answer' in response:
+            full_response += response['answer']
+            message_placeholder.markdown(full_response + "▌")
+        # if 'id' in response:
+        #     st.session_state.conversation_id=response["id"]
+        # # if 'type' in response and response['type'] == 'source':
+        # if 'next_questions' in response:
+        #     # st.write(response)
+        #     next_questions = response['next_questions']
+        #     count = 0
+        #     col1, col2= st.columns(2)
+        #     for next_question in next_questions:
+        #         count += 1
+        #         if count % 2 != 0:
+        #             col1.write(next_question)
+        #         else:
+        #             col2.write(next_question)
+
+    # message_placeholder.markdown(full_response)
+    st.session_state.messages.append(
+        {
+            "role": "assistant",
+            "content": full_response,
+            "prompt": prompt,
+            "history": st.session_state.history,
+            # "next_questions": next_questions,
+        }
+    )
+
+
+
 
 option = st.sidebar.selectbox(
    "Please select level of difficulty",
@@ -18,3 +119,6 @@ start = st.button("Start", type="primary")
 if start:
 
     st.write(":red[starting...]")
+
+
+

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+simplejson
+requests