Update app.py

app.py

@@ -110,10 +110,14 @@ if prompt := st.chat_input("What legal question or issue do you need help with?"
             if response.choices and response.choices[0].message:
                 full_response = response.choices[0].message.content
 
+                # Ensure the response does not include 'User:'
+                if full_response.endswith('User:'):
+                    full_response = full_response.replace('User:', '').strip()
+
                 # Stream the full response to the chat using st.write
                 with st.chat_message("assistant"):
                     st.markdown(full_response)
                 
                 st.session_state.messages.append({"role": "assistant", "content": full_response})
         except Exception as e:
-            st.error(f"An error occurred: {e}")
+            st.error(f"An error occurred: {e}")