Update app.py

app.py

@@ -53,7 +53,8 @@ def format_message(content):
     final_content = re.sub(r"(<br>)*</details>(<br>)*", "</details>", final_content)
     final_content = re.sub(r"(<br>)*(<div class=[^>]+>)(<br>)*", r"\2", final_content)
     
-    return final_content
+    # Return as gr.HTML to ensure proper rendering
+    return gr.HTML(final_content)
 
 def user_input(user_message, history):
     if history is None:
@@ -126,26 +127,27 @@ def predict(message, chat_history, system_msg, temperature, top_p, repetition_pe
 
 # JavaScript to ensure the details tag is working properly
 js_script = """
-    <script>
-        function initDetails() {
-            document.querySelectorAll('.think-container').forEach(detail => {
-                // 默认关闭所有思考框
-                detail.open = false;
-                
-                // 添加箭头图标交互
-                const summary = detail.querySelector('.think-summary');
-                summary.style.listStyle = 'disclosure-closed';
-                
-                detail.addEventListener('toggle', () => {
-                    summary.style.listStyle = detail.open ? 'disclosure-open' : 'disclosure-closed';
-                });
+<script>
+    function initDetails() {
+        const detailsElements = document.querySelectorAll('.think-container');
+        detailsElements.forEach((detail) => {
+            // 默认关闭所有思考框
+            detail.open = false;
+            
+            // 添加箭头图标交互
+            const summary = detail.querySelector('.think-summary');
+            summary.style.listStyle = 'disclosure-closed';
+            
+            detail.addEventListener('toggle', () => {
+                summary.style.listStyle = detail.open ? 'disclosure-open' : 'disclosure-closed';
             });
-        }
-        
-        // 初始化及每次聊天更新后执行
-        document.addEventListener('DOMContentLoaded', initDetails);
-        window.addEventListener('vue:mounted', initDetails);  // Gradio 的 DOM 更新事件
-    </script>
+        });
+    }
+    
+    // 初始化及每次聊天更新后执行
+    document.addEventListener('DOMContentLoaded', initDetails);
+    window.addEventListener('vue:mounted', initDetails);  // Gradio 的 DOM 更新事件
+</script>
 """
 
 with gr.Blocks(css=CSS, title="Tifa-Deepsex-COT-14B") as demo: