Update app.py

app.py

@@ -989,32 +989,30 @@ elif page == "Sessions":
             memory = ConversationBufferWindowMemory(k=5, return_messages=True)
 
             def get_chatmodel_response(question):
-                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)
+                            # 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 rename functionality
             new_title = st.text_input("Rename session:", value=session["title"])