Fixing formater

chatbot.py

@@ -131,18 +131,18 @@ def render_chatbot(code, output, error):
         st.markdown(f'<div class="chat-message user-message">{escape(q)}</div>', unsafe_allow_html=True)
 
         def format_response(txt):
-            parts = txt.split('')
-            result = ''
-            for j, part in enumerate(parts):
-                if j % 2 == 1:
-                    lines = part.splitlines()
-                    if lines and lines[0].isalpha():
-                        lines = lines[1:]
-                    code_html = escape("\n".join(lines))
-                    result += f'<pre><code>{code_html}</code></pre>'
-                else:
-                    result += escape(part)
-            return result
+        parts = txt.split('```')
+        result = ''
+        for j, part in enumerate(parts):
+            if j % 2 == 1:
+                lines = part.splitlines()
+                if lines and lines[0].isalpha():
+                    lines = lines[1:]
+                code_html = escape("\n".join(lines))
+                result += f'<pre><code>{code_html}</code></pre>'
+            else:
+                result += escape(part)
+        return result
 
         formatted = format_response(a)
         st.markdown(f'<div class="chat-message bot-message">{formatted}</div>', unsafe_allow_html=True)