fix: artifact render error (ChatGPTNextWeb#5306)

fix: artifact render error
HaiyaoTec · Aug 21, 2024 · 4ec6b06 · 4ec6b06
1 parent 1748dd6
commit 4ec6b06
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/components/artifacts.tsx b/app/components/artifacts.tsx
@@ -81,10 +81,10 @@ export const HTMLPreview = forwardRef<HTMLPreviewHander, HTMLPreviewProps>(
 
     const srcDoc = useMemo(() => {
       const script = `<script>new ResizeObserver((entries) => parent.postMessage({id: '${frameId}', height: entries[0].target.clientHeight}, '*')).observe(document.body)</script>`;
-      if (props.code.includes("</head>")) {
-        props.code.replace("</head>", "</head>" + script);
+      if (props.code.includes("<!DOCTYPE html>")) {
+        props.code.replace("<!DOCTYPE html>", "<!DOCTYPE html>" + script);
       }
-      return props.code + script;
+      return script + props.code;
     }, [props.code, frameId]);
 
     const handleOnLoad = () => {