Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import os
 from PIL import Image
 import spaces
+import copy
 
 from kimi_vl.serve.frontend import reload_javascript
 from kimi_vl.serve.utils import (
@@ -88,7 +89,15 @@ def get_prompt(conversation) -> str:
     system_prompt = conversation.system_template.format(system_message=conversation.system_message)
     return system_prompt
 
+def highlight_thinking(msg: str) -> str:
+    msg = copy.deepcopy()
+    if "◁think▷" in msg:
+        msg = msg.replace("◁think▷", "<b style='color:blue;'>🤔Thinking...</b>\n")
+    if "◁/think▷" in msg:
+        msg = msg.replace("◁/think▷", "\n<b style='color:purple;'>💡Summary</b>\n")
 
+    return msg
+    
 @wrap_gen_fn
 @spaces.GPU(duration=180)
 def predict(
@@ -168,7 +177,7 @@ def predict(
             full_response += x
             response = strip_stop_words(full_response, stop_words)
             conversation.update_last_message(response)
-            gradio_chatbot_output[-1][1] = response
+            gradio_chatbot_output[-1][1] = highlight_thinking(response)
 
             yield gradio_chatbot_output, to_gradio_history(conversation), "Generating..."