Minor adjustments

app.py

@@ -100,7 +100,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):
@@ -111,6 +116,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])
@@ -300,7 +307,7 @@ with st.sidebar:
             st.warning("Please enter a valid Project ID.")
 
 # 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": "assistant", "content": st.session_state.response})
     st.session_state.is_error = False
     st.session_state.is_streaming = False
@@ -311,7 +318,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:
         messages = st.session_state.messages.copy()
         response = get_ai_response(messages)