Update app.py

app.py

@@ -44,19 +44,17 @@ def respond(
             }
         )
         response = ""
+        is_reasoning_end = True
 
         for index, event in enumerate(completion):
             if hasattr(event.choices[0].delta, 'reasoning_content'):
-                if index == 0:
-                    response += '> **开始思考**\n\n'
-                token = event.choices[0].delta.reasoning_content
+                separator = '--- **开始思考** ---' if index == 0 else ''
+                response += separator + '\n'
+                token = f'<span style="color: #1e88e5; background-color: #e8f5e9; padding: 2px 4px; border-radius: 3px;">{event.choices[0].delta.reasoning_content}</span>'
             else:
-                response += '> **结束思考**\n\n'
-                token = event.choices[0].delta.content
-            # if message.find('写一篇关于青春的五言绝句') != -1:
-            #     print(11111111111)
-            #     print(event)
-            response += token
+                response += '--- **结束思考** ---\n'
+                token = f'<span style="color: #43a047; background-color: #f1f8e9; padding: 2px 4px; border-radius: 3px;">{event.choices[0].delta.content}</span>'
+            response += token + '\n'
             yield response
     except Exception as e:
         raise gr.Error(f"发生错误: {str(e)}")