feat: 优化dalle3图片显示

modules/models/DALLE3.py

@@ -35,4 +35,4 @@ class OpenAI_DALLE3_Client(BaseLLMModel):
                 msg = str(e)
                 match = re.search(r"'message': '([^']*)'", msg)
                 return match.group(1), 0
-        return f'<img src="{response.data[0].url}"> {response.data[0].revised_prompt}', 0
+        return f'<!-- S O PREFIX --><a data-fancybox="gallery" target="_blank" href="{response.data[0].url}"><img src="{response.data[0].url}" /></a><!-- E O PREFIX -->{response.data[0].revised_prompt}', 0

modules/utils.py

@@ -243,7 +243,7 @@ def clip_rawtext(chat_message, need_escape=True):
     message_clipped = chat_message[: hr_match.start()] if hr_match else chat_message
     # second, avoid agent-prefix being escaped
     agent_prefix_pattern = (
-        r'(<!-- S O PREFIX --><p class="agent-prefix">.*?<\/p><!-- E O PREFIX -->)'
+        r'(<!-- S O PREFIX -->.*?<!-- E O PREFIX -->)'
     )
     # agent_matches = re.findall(agent_prefix_pattern, message_clipped)
     agent_parts = re.split(agent_prefix_pattern, message_clipped, flags=re.DOTALL)
@@ -257,6 +257,7 @@ def clip_rawtext(chat_message, need_escape=True):
                     else f'<pre class="fake-pre">{part}</pre>'
                 )
         else:
+            part = part.replace(' data-fancybox="gallery"', '')
             final_message += part
     return final_message
 

modules/webui.py

@@ -55,6 +55,7 @@ def reload_javascript():
     js = javascript_html()
     js += '<script async type="module" src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>'
     js += '<script async type="module" src="http://spin.js.org/spin.umd.js"></script><link type="text/css" href="https://spin.js.org/spin.css" rel="stylesheet" />'
+    js += '<script async src="https://cdn.jsdelivr.net/npm/@fancyapps/[email protected]/dist/fancybox/fancybox.umd.js"></script><link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@fancyapps/[email protected]/dist/fancybox/fancybox.css" />'
     
     meta = """
         <meta name="apple-mobile-web-app-title" content="川虎 Chat">

web_assets/javascript/ChuanhuChat.js

@@ -349,6 +349,7 @@ function chatbotContentChanged(attempt = 1, force = false) {
             disableSendBtn();
             updateSlider();
             updateCheckboxes();
+            bindFancyBox();
 
             gradioApp().querySelectorAll('#chuanhu-chatbot .message-wrap .message.bot').forEach(addChuanhuButton);
 

web_assets/javascript/utils.js

@@ -71,6 +71,16 @@ function downloadFile(fileUrl, filename = "", format = "", retryTimeout = 200, m
 }
     
 
+function bindFancyBox() {
+    Fancybox.bind('[data-fancybox]', {
+        Carousel: {
+            Panzoom: {
+                decelFriction: 0.5
+            }
+        }
+    });
+}
+
 
 /* NOTE: These reload functions are not used in the current version of the code.
  *       From stable-diffusion-webui

web_assets/stylesheet/chatbot.css

@@ -243,6 +243,14 @@ hr.append-display {
     border-radius: 8px !important;
     margin: 4px !important
 }
+.message.bot img {
+    border-radius: 8px !important;
+    width: 512px;
+    max-height: unset !important;
+    max-width: 100% !important;
+    margin: unset !important;
+    margin-bottom: .8em !important;
+}
 
 /* custom buttons */
 .chuanhu-btn {