improve preview for videos by showing the streaming response as it co…

…mes in
mrylil · Mar 8, 2024 · f09b4c7 · f09b4c7
1 parent cacf78a
commit f09b4c7
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/frontend/src/components/preview/extractHtml.ts b/frontend/src/components/preview/extractHtml.ts
@@ -1,4 +1,16 @@
+// Not robust enough to support <html lang='en'> for instance
 export function extractHtml(code: string): string {
-  const htmlStartIndex = code.indexOf("<html>");
-  return htmlStartIndex !== -1 ? code.slice(htmlStartIndex) : "";
+  const lastHtmlStartIndex = code.lastIndexOf("<html>");
+  let htmlEndIndex = code.indexOf("</html>", lastHtmlStartIndex);
+
+  if (lastHtmlStartIndex !== -1) {
+    // If "</html>" is found, adjust htmlEndIndex to include the "</html>" tag
+    if (htmlEndIndex !== -1) {
+      htmlEndIndex += "</html>".length;
+      return code.slice(lastHtmlStartIndex, htmlEndIndex);
+    }
+    // If "</html>" is not found, return the rest of the string starting from the last "<html>"
+    return code.slice(lastHtmlStartIndex);
+  }
+  return "";
 }