update

app.py

@@ -68,13 +68,41 @@ def respond(
                                     if 'content' in delta:
                                         content = delta['content']
                                         if content:
-                                            content = content.replace('<', '&lt;').replace('>', '&gt;')
-                                            content = content.replace('*', '\\*')
                                             current_response += content
-                                            yield current_response
+                                            
+                                            if len(current_response) > 16:
+                                                if '<think>' in current_response:
+                                                    current_response = current_response.replace('<think>', '<details open><summary>Thinking</summary>\n\n```')
+                                                if '</think>' in current_response:
+                                                    current_response = current_response.replace('</think>', '```\n\n</details>')
+                                                if '**Final Answer**' in current_response:
+                                                    current_response = current_response.replace('**Final Answer**', '')
+                                                
+                                                formatted_response = current_response[:-16]
+                                                
+                                                formatted_response = formatted_response.replace('<', '&lt;').replace('>', '&gt;')
+                                                formatted_response = formatted_response.replace('&lt;details open&gt;', '<details open>')
+                                                formatted_response = formatted_response.replace('&lt;/details&gt;', '</details>')
+                                                formatted_response = formatted_response.replace('&lt;summary&gt;', '<summary>')
+                                                formatted_response = formatted_response.replace('&lt;/summary&gt;', '</summary>')
+                                                formatted_response = formatted_response.replace('*', '\\*')
+                                                yield formatted_response
                             except json.JSONDecodeError:
                                 continue
-                # print(f"[INFO] final response: {current_response}")
+                if current_response:
+                    # current_response = current_response.replace('<details open>', '<details>')
+                    if '**Final Answer**' in current_response:
+                        current_response = current_response.replace('**Final Answer**', '')
+                    
+                    formatted_response = current_response
+                    formatted_response = formatted_response.replace('<', '&lt;').replace('>', '&gt;')
+                    formatted_response = formatted_response.replace('&lt;details open&gt;', '<details open>')
+                    # formatted_response = formatted_response.replace('&lt;details&gt;', '<details>')
+                    formatted_response = formatted_response.replace('&lt;/details&gt;', '</details>')
+                    formatted_response = formatted_response.replace('&lt;summary&gt;', '<summary>')
+                    formatted_response = formatted_response.replace('&lt;/summary&gt;', '</summary>')
+                    formatted_response = formatted_response.replace('*', '\\*')
+                    yield formatted_response
             else:
                 print(f"[ERROR] Bad status code: {r.status_code}, response: {r.text}")
                 yield "Service temporarily unavailable"
@@ -100,7 +128,40 @@ demo = gr.ChatInterface(
             label="Top-p (nucleus sampling)",
         ),
     ],
-    type="messages"
+    type="messages",
+    css="""
+    .prose pre {
+        white-space: pre-wrap !important;
+        word-wrap: break-word !important;
+        overflow-wrap: break-word !important;
+        max-width: 100% !important;
+        margin-bottom: 1.5em !important;
+    }
+    .prose code {
+        white-space: pre-wrap !important;
+        word-wrap: break-word !important;
+        overflow-wrap: break-word !important;
+        max-width: 100% !important;
+    }
+    .prose pre code {
+        white-space: pre-wrap !important;
+        word-wrap: break-word !important;
+        overflow-wrap: break-word !important;
+        max-width: 100% !important;
+    }
+    .accordion {
+        margin: 0 !important;
+        border: none !important;
+    }
+    .accordion-header {
+        background: #f0f0f0 !important;
+        padding: 8px !important;
+        cursor: pointer !important;
+    }
+    .accordion-content {
+        padding: 8px !important;
+    }
+    """
 )