Minor adjustments

app.py

@@ -81,7 +81,12 @@ def normalize_code_block(match):
                          .replace("\\<", "<")\
                          .replace("\\>", ">")
 
+def normalize_inline(match):
+    return match.group(0).replace("\\<", "<")\
+                         .replace("\\>", ">")
+
 code_block_pattern = r"(```.*?```)"
+inline_pattern = r"`([^`\n]+?)`"
 
 def display_messages():
     for i, message in enumerate(st.session_state.messages):
@@ -93,6 +98,8 @@ def display_messages():
             if "```" in shown_message:
                 # Replace "  \n" with "\n" within code blocks
                 shown_message = re.sub(code_block_pattern, normalize_code_block, shown_message, flags=re.DOTALL)
+            if "`" in shown_message:
+                shown_message = re.sub(inline_pattern, normalize_inline, shown_message)
             st.markdown(shown_message)
             
             col1, col2, col3, col4 = st.columns([1, 1, 1, 1])
@@ -221,7 +228,7 @@ with st.sidebar:
             st.warning("Please enter a valid API Key.")
 
 # After Stop generating
-if "is_streaming" in st.session_state and st.session_state.is_streaming:
+if st.session_state.get("is_streaming"):
     st.session_state.messages.append({"role": "CHATBOT", "text": st.session_state.response})
     st.session_state.is_streaming = False
     if "retry_flag" in st.session_state and st.session_state.retry_flag:
@@ -231,7 +238,7 @@ if "is_streaming" in st.session_state and st.session_state.is_streaming:
 display_messages()
 
 # After Retry
-if "retry_flag" in st.session_state and st.session_state.retry_flag == True:
+if st.session_state.get("retry_flag"):
     if len(st.session_state.messages) > 0:
         prompt = st.session_state.messages[-1]["text"]
         messages = st.session_state.messages[:-1].copy()