From ad5d09fbbecbe0839343f360d8ff895381ba9650 Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Fri, 7 Feb 2025 13:13:09 +0100
Subject: [PATCH] Fix show new Window Button

---
 g4f/gui/client/demo.html            | 10 +++++-----
 g4f/gui/client/static/js/chat.v1.js |  9 ++++-----
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/g4f/gui/client/demo.html b/g4f/gui/client/demo.html
index d94ad864588..4ecdd6d04ab 100644
--- a/g4f/gui/client/demo.html
+++ b/g4f/gui/client/demo.html
@@ -181,7 +181,6 @@
     <script>
         (async () => {
             const isIframe = window.self !== window.top;
-            const button = document.querySelector('form a.button');
             let url = new URL(window.location.href)
             let params = new URLSearchParams(url.search);
             if (params.get("__sign")) {
@@ -190,9 +189,6 @@
                     window.location.replace("/");
                 }
             }
-            if (isIframe) {
-                button.classList.remove('hidden');
-            }
         })();
     </script>
     <script src="https://unpkg.com/es-module-shims@1.7.0/dist/es-module-shims.js"></script>
@@ -236,7 +232,11 @@
             import * as hub from "@huggingface/hub";
             import { init } from "@huggingface/space-header";
 
-            if (window.self === window.top) {
+            const isIframe = window.self !== window.top;
+            const button = document.querySelector('form a.button');
+            if (isIframe) {
+                button.classList.remove('hidden');
+            } else {
                 init("roxky/g4f-space");
             }
 
diff --git a/g4f/gui/client/static/js/chat.v1.js b/g4f/gui/client/static/js/chat.v1.js
index 320a8e98971..7f64a993401 100644
--- a/g4f/gui/client/static/js/chat.v1.js
+++ b/g4f/gui/client/static/js/chat.v1.js
@@ -811,19 +811,18 @@ async function add_message_chunk(message, message_id, provider, scroll, finish_m
         log_storage.appendChild(p);
         await api("log", {...message, provider: provider_storage[message_id]});
     } else if (message.type == "preview") {
-        if (content_map.inner.clientHeight > 200)
-            content_map.inner.style.height = content_map.inner.clientHeight + "px";
         if (img = content_map.inner.querySelector("img"))
             if (!img.complete)
                 return;
             else
                 img.src = message.images;
-        else
-           content_map.inner.innerHTML = markdown_render(message.preview);
+        else {
+            content_map.inner.innerHTML = markdown_render(message.preview);
+            await register_message_images();
+        }
     } else if (message.type == "content") {
         message_storage[message_id] += message.content;
         update_message(content_map, message_id, null, scroll);
-        content_map.inner.style.height = "";
     } else if (message.type == "log") {
         let p = document.createElement("p");
         p.innerText = message.log;