diff --git a/__main__.py b/__main__.py
index 60285ea..45a9015 100644
--- a/__main__.py
+++ b/__main__.py
@@ -79,6 +79,7 @@ class ChatBot(ChatBotFrame):
         with gr.Box(elem_id='chat_box'):
             with gr.Row():
                 self.sm_upload = gr.UploadButton(label='🔗Upload', file_count='multiple', elem_classes='sm_btn').style(size='sm', full_width=False)
+                self.sm_code_block = gr.Button(value='⚛︎Code', elem_classes='sm_btn').style(size='sm', full_width=False)
                 gr.HTML(func_box.get_html("appearance_switcher.html").format(label=""), elem_id='user_input_tb', elem_classes="insert_block")
             with gr.Row():
                 self.txt = gr.Textbox(show_label=False,  placeholder="Input question here.", elem_classes='chat_input').style(container=False)
@@ -91,6 +92,7 @@ class ChatBot(ChatBotFrame):
         self.sm_upload.upload(on_file_uploaded, [self.sm_upload, self.chatbot, self.txt], [self.chatbot, self.txt]).then(
             fn=lambda: [gr.Tabs.update(selected='plug_tab'), gr.Column.update(visible=False)], inputs=None, outputs=[self.tabs_inputs, self.examples_column]
         )
+        self.sm_code_block.click(fn=lambda x: x+'\n```\n\n```', inputs=[self.txt], outputs=[self.txt])
 
     def draw_examples(self):
         with gr.Column(elem_id='examples_col') as self.examples_column:
diff --git a/func_box.py b/func_box.py
index a6df439..1a5b88c 100644
--- a/func_box.py
+++ b/func_box.py
@@ -504,7 +504,12 @@ def show_prompt_result(index, data: gr.Dataset, chatbot):
 
 def pattern_html(html):
     bs = BeautifulSoup(html, 'html.parser')
-    return bs.get_text(separator='')
+    md_message = bs.find('div', {'class': 'md-message'})
+    if md_message:
+        return md_message.get_text(separator='')
+    else:
+        return ""
+
 
 def thread_write_chat(chatbot, history):
     """
@@ -709,4 +714,4 @@ class JsonHandle:
 
 
 if __name__ == '__main__':
-    html_local_img("docs/imgs/openai-api-key-billing-paid-account.png")
\ No newline at end of file
+    print(pattern_html('<div class="raw-message hideM">很抱歉，作为一个文</div><div class="md-message"><p>很抱歉，作为一个文</p></div>'))
\ No newline at end of file