Spaces:

DMindAI
/

DMind-1

Running on CPU Upgrade

App Files Files Community

nanova commited on 8 days ago

Commit

942e73f

1 Parent(s): bcf0f93

update format

Browse files

Files changed (1) hide show

app.py +24 -4

app.py CHANGED Viewed

@@ -69,29 +69,37 @@ def respond(
                                             if len(current_response) > 16:
                                                 if '<think>' in current_response:
-                                                    current_response = current_response.replace('<think>', '???+ Thinking\n```')
                                                 if '</think>' in current_response:
-                                                    current_response = current_response.replace('</think>', '```')
                                                 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('*', '\\*')
                                                 yield formatted_response
                             except json.JSONDecodeError:
                                 continue
                 if current_response:
                     if '<think>' in current_response:
-                        current_response = current_response.replace('<think>', '???+ Thinking\n```')
                     if '</think>' in current_response:
-                        current_response = current_response.replace('</think>', '```')
                     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('*', '\\*')
                     yield formatted_response
             else:
@@ -139,6 +147,18 @@ demo = gr.ChatInterface(
         overflow-wrap: break-word !important;
         max-width: 100% !important;
     }
     """
 )

                                             if len(current_response) > 16:
                                                 if '<think>' in current_response:
+                                                    current_response = current_response.replace('<think>', '<details open><summary>思考过程</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
                 if current_response:
                     if '<think>' in current_response:
+                        current_response = current_response.replace('<think>', '<details open><summary>思考过程</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
                     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
             else:
         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;
+    }
     """
 )