Update app.py

app.py

@@ -125,14 +125,74 @@ st.sidebar.caption(f"Session ID: {genuuid()}")
 # Main chat interface
 st.header("Chat with the Agents")
 user_id = st.text_input("User ID:", key="user_id")
-user_input = st.text_input("Write your message here:", key="user_input")
-
-if 'history' not in st.session_state:
-    st.session_state.history = []
+# user_input = st.text_input("Write your message here:", key="user_input")
+
+# if 'history' not in st.session_state:
+#     st.session_state.history = []
+
+# if st.button("Send"):
+#     # Placeholder for processing the input and generating a response
+#     data = ChatRequestClient(
+#         user_id=user_id,
+#         user_input=user_input,
+#         numberOfQuestions=numberOfQuestions,
+#         welcomeMessage="",
+#         llm1=llm1,
+#         tokens1=tokens1,
+#         temperature1=temp1,
+#         persona1SystemMessage=persona1SystemMessage,
+#         persona2SystemMessage=persona2SystemMessage,
+#         userMessage2=userMessage2,
+#         llm2=llm2,
+#         tokens2=tokens2,
+#         temperature2=temp2
+#     )
+#     response = call_chat_api(data)
+    
+    
+    st.markdown(f"##### Time take: {format_elapsed_time(response['elapsed_time'])}")
+    st.markdown(f"##### Question Count : {response['count']} of {numberOfQuestions}")
+    
+    
+    # # {"count":count, "user_id":user_id,"time_stamp":time_stamp, "elapsed_time":elapsed_time, "content":content}
 
-if st.button("Send"):
-    # Placeholder for processing the input and generating a response
-    data = ChatRequestClient(
+    # st.session_state.history.append("You: " + user_input)
+                
+    # st.session_state.history.append("Agent: " + response['content'])  # Using 'response' after it's defined
+    # for message in st.session_state.history:
+    #     st.write(message)
+# session state
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = [
+        AIMessage(content="Hello, I am the GSK Reg Intel Assistant. How can I help you?"),
+    ]
+
+# conversation
+for message in st.session_state.chat_history:
+    if isinstance(message, AIMessage):
+        with st.chat_message("AI"):
+            st.write(message.content)
+    elif isinstance(message, HumanMessage):
+        with st.chat_message("Human"):
+            st.write(message.content)
+
+# user input
+user_query = st.chat_input("Type your message here...")
+if user_query is not None and user_query != "":
+    st.session_state.chat_history.append(HumanMessage(content=user_query))
+
+    with st.chat_message("Human"):
+        st.markdown(user_query)
+
+
+        # Display the table in the sidebar
+        st.sidebar.write("### Knowledge Base Results")
+        st.sidebar.dataframe(df)  # Adjust height as needed
+    else:
+        st.sidebar.write("No relevant knowledge base results found.")
+    
+    with st.chat_message("AI"):
+        data = ChatRequestClient(
         user_id=user_id,
         user_input=user_input,
         numberOfQuestions=numberOfQuestions,
@@ -148,17 +208,5 @@ if st.button("Send"):
         temperature2=temp2
     )
     response = call_chat_api(data)
-    
-    
-    st.markdown(f"##### Time take: {format_elapsed_time(response['elapsed_time'])}")
-    st.markdown(f"##### Question Count : {response['count']} of {numberOfQuestions}")
-    
-    
-    # {"count":count, "user_id":user_id,"time_stamp":time_stamp, "elapsed_time":elapsed_time, "content":content}
-
-    st.session_state.history.append("You: " + user_input)
-                
-    st.session_state.history.append("Agent: " + response['content'])  # Using 'response' after it's defined
-    for message in st.session_state.history:
-        st.write(message)
-    
+        response = st.write_stream(call_chat_api(data))
+        st.session_state.chat_history.append(AIMessage(content=response))