Update app.py

app.py

@@ -989,30 +989,32 @@ elif page == "Sessions":
             memory = ConversationBufferWindowMemory(k=5, return_messages=True)
 
             def get_chatmodel_response(question):
-                            # Append the user's question as a HumanMessage
-                            session['flowmessages'].append(HumanMessage(content=question))
-                            
-                            # Save the input question to memory
-                            memory.save_context({"input": question}, {"output": ""})  
-                            
-                            # Construct chat history for the prompt
-                            chat_history = "\n".join([f"User: {msg.content}" if isinstance(msg, HumanMessage) else f"Bot: {msg.content}" for msg in session['flowmessages']])
-                            
-                            # Format the prompt with the chat history and the user's message
-                            prompt = CUSTOM_PROMPT.format(chat_history=chat_history, user_message=question, language=language)
-                            
-                            # Create the messages list with the formatted prompt
-                            messages = [SystemMessage(content=prompt)]  # Ensure this is formatted as required
-                            
-                            # Use the chat model to get a response
-                            try:
-                                answer = st.session_state['chat'](messages)  # Pass list of messages to chat
-                                session['flowmessages'].append(AIMessage(content=answer.content))  # Handle response properly
-                                save_session(session)  # Ensure session is saved after adding new messages
-                                return answer.content
-                            except Exception as e:
-                                st.error(f"An error occurred: {str(e)}")
-                                return "Error generating response."
+                session['flowmessages'].append(HumanMessage(content=question))
+                memory.save_context({"input": question}, {"output": ""})  # Save the input question to memory
+                chat_history = "\n".join([f"User: {msg.content}" if isinstance(msg, HumanMessage) else f"Bot: {msg.content}" for msg in session['flowmessages']])
+                prompt = CUSTOM_PROMPT.format(chat_history=chat_history, user_message=question, language=language)
+                messages = [SystemMessage(content=prompt)]
+                # Use st.session_state['chat'] here
+                answer = st.session_state['chat'](messages)  # Pass list of messages to chat
+                session['flowmessages'].append(AIMessage(content=answer.content))  # Adjusted to handle response properly
+                save_session(session)  # Ensure session is saved after adding new messages
+                return answer.content
+
+
+            input = st.text_input("Input: ", key="input")
+            submit = st.button("Ask the question")
+
+            if submit:
+                response = get_chatmodel_response(input)
+                save_session(session)
+                st.experimental_rerun()
+            if "flowmessages" in session:
+                st.subheader("Chat")
+                for message in session['flowmessages']:
+                    if isinstance(message, HumanMessage):
+                        st.write(user_template.replace("{{MSG}}", message.content), unsafe_allow_html=True)
+                    elif isinstance(message, AIMessage):
+                        st.write(bot_template.replace("{{MSG}}", message.content), unsafe_allow_html=True)
 
             # Session rename functionality
             new_title = st.text_input("Rename session:", value=session["title"])
@@ -1033,6 +1035,7 @@ elif page == "Sessions":
                         mime='text/plain'
                     )
 
+
 # Define other tool functions here...
 def breathing_exercise():
     st.markdown("""