Update app.py

app.py

@@ -69,8 +69,10 @@ def bot_streaming(message, history, max_new_tokens=250):
         buffer += new_text
         generated_text_without_prompt = buffer
         time.sleep(0.01)
-        buffer = re.sub(r"<.*?>", "", buffer)
-        print(f"Buffer: {buffer}")
+    
+        buffer = re.sub(r"<(\w+)>", r"(Here begins the \1 stage)", buffer)  
+        buffer = re.sub(r"</(\w+)>", r"(Here ends the \1 stage)", buffer)  
+    
         yield buffer
 
 
@@ -85,7 +87,7 @@ demo = gr.ChatInterface(fn=bot_streaming, title="LLaVA-CoT",
           )
         ],
       cache_examples=False,
-      description="Upload an image, and start chatting about it. To learn more about LLaVA-CoT, visit [oir GitHub page](https://github.com/PKU-YuanGroup/LLaVA-CoT). ",
+      description="Upload an image, and start chatting about it. To learn more about LLaVA-CoT, visit [oir GitHub page](https://github.com/PKU-YuanGroup/LLaVA-CoT). Note: Since Gradio currently does not support displaying the <X> symbol in the output, we have replaced it with the expression (Here begins the X phase).",
       stop_btn="Stop Generation", 
       fill_height=True,
     multimodal=True)