Update app.py

app.py

@@ -991,16 +991,35 @@ elif page == "Sessions":
             def get_chatmodel_response(question):
                 session['flowmessages'].append(HumanMessage(content=question))
                 memory.save_context({"input": question}, {"output": ""})  # Save the input question to memory
+            
+                # Prepare the 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']])
+            
+                # Construct the prompt using the template
                 prompt = CUSTOM_PROMPT.format(chat_history=chat_history, user_message=question, language=language)
-                messages = [SystemMessage(content=prompt)]
-                # Use st.session_state['chat'] here
+            
+                # Ensure the prompt is not empty
+                if not prompt.strip():
+                    raise ValueError("The constructed prompt is empty. Check the message formatting.")
+            
+                # Use HumanMessage instead of SystemMessage
+                messages = [HumanMessage(content=prompt)]
+            
+                # Get the response from the chat model
                 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
+            
+                # Ensure AI response is not empty
+                if not answer or not answer.content.strip():
+                    raise ValueError("The AI response is empty. Check the prompt or input.")
+                
+                # Append the AI response to the session
+                session['flowmessages'].append(AIMessage(content=answer.content))
+                save_session(session)  # Save the session after appending new messages
+                
                 return answer.content
 
 
+
             input = st.text_input("Input: ", key="input")
             submit = st.button("Ask the question")