Minor adjustments

app.py

@@ -82,7 +82,9 @@ def get_ai_response(messages):
             if chunk.choices and chunk.choices[0].delta.get("content"):
                 content = chunk.choices[0].delta["content"]
                 st.session_state.response += content
-                shown_message += content.replace("\n", "  \n")
+                shown_message += content.replace("\n", "  \n")\
+                                        .replace("<", "\\<")\
+                                        .replace(">", "\\>")
                 placeholder.markdown(shown_message)
 
     if st.session_state.prefill == st.session_state.response:
@@ -93,18 +95,22 @@ def get_ai_response(messages):
     st.session_state.is_streaming = False
     return st.session_state.response
 
-def replace_spaces_in_code_blocks(match):
-    return match.group(0).replace("  \n", "\n")
+def normalize_code_block(match):
+    return match.group(0).replace("  \n", "\n")\
+                         .replace("\\<", "<")\
+                         .replace("\\>", ">")
 
 code_block_pattern = r"(```.*?```)"
 
 def display_messages():
     for i, message in enumerate(st.session_state.messages):
         with st.chat_message(message["role"]):
-            shown_message = message["content"].replace("\n", "  \n")
+            shown_message = message["content"].replace("\n", "  \n")\
+                                              .replace("<", "\\<")\
+                                              .replace(">", "\\>")
             if "```" in shown_message:
                 # Replace "  \n" with "\n" within code blocks
-                shown_message = re.sub(code_block_pattern, replace_spaces_in_code_blocks, shown_message, flags=re.DOTALL)
+                shown_message = re.sub(code_block_pattern, normalize_code_block, shown_message, flags=re.DOTALL)
             st.markdown(shown_message)
             
             col1, col2, col3, col4 = st.columns([1, 1, 1, 1])
@@ -169,6 +175,8 @@ def display_messages():
     
 # Add sidebar for advanced settings
 with st.sidebar:
+    st.markdown("Help (Japanese): https://rentry.org/9hgneofz")
+    
     # Copy Conversation History button
     log_text = ""
     for message in st.session_state.messages:
@@ -240,7 +248,8 @@ with st.sidebar:
     model = st.selectbox("Model", options=model_list, index=0)
     system_prompt = st.text_area("System prompt", height=200)
 
-    st.session_state.prefill = st.text_area("Prefill", height=50, value=st.session_state.prefill, placeholder="It only works well with the Claude models.")
+    st.session_state.prefill = st.text_area("Prefill", height=50, value=st.session_state.prefill, placeholder="It only works well with the Claude models.",
+                                            help="You can prefill the assistant's responses. You can also directly type the @prefill command into the chat field (e.g., \"Write a novel. @prefill Sure! I'd be happy to write a novel for you.\")")
     save_prefill = st.toggle("Save the @prefill command input in the sidebar", value=True)
     
     temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=1.0, step=0.1)
@@ -314,7 +323,7 @@ if "retry_flag" in st.session_state and st.session_state.retry_flag == True:
     else:
         st.session_state.retry_flag = False
 
-if prompt := st.chat_input("What is up?"):
+if prompt := st.chat_input("Enter your message here..."):
     used_prefill = False
     prefill_pattern = r"([@＠](prefill|ぷれふぃる|プレフィル)\s?(.*))"
     prefill_match = re.search(prefill_pattern, prompt)
@@ -329,8 +338,12 @@ if prompt := st.chat_input("What is up?"):
     st.session_state.messages.append({"role": "user", "content": prompt})
     messages = st.session_state.messages.copy()
 
+    shown_message = prompt.replace("\n", "  \n")\
+                          .replace("<", "\\<")\
+                          .replace(">", "\\>")
+
     with st.chat_message("user"):
-        st.write(prompt)
+        st.write(shown_message)
 
     response = get_ai_response(messages)