fixed issue with the /?isVisited=true call

hf_server.js

@@ -3,7 +3,7 @@
  * All rights reserved.
  */
 //@ts-check
-const VERSION = '0.6.0.hf.015.c';
+const VERSION = '0.6.0.hf.015.d';
 
 const express = require('express');
 const http = require('http');
@@ -98,88 +98,95 @@ const COMMON_STYLES =
         }
     </style>`;
 
-const SCRIPTS = 
-    `<script>
-        // Script to check Hugging Face login status and control 'Clone Repo' button
+    function getScriptsHtml(url)
+    {
 
-        function startServer() {
-            document.getElementById('startServerButton').classList.remove
-            ('highlight-button-green');
-            document.getElementById('startServerButton').disabled = true;
+        const scripts_html = 
+            `<script>
+                // Script to check Hugging Face login status and control 'Clone Repo' button
 
+                function startServer() {
+                    document.getElementById('startServerButton').classList.remove
+                    ('highlight-button-green');
+                    document.getElementById('startServerButton').disabled = true;
 
-            fetch('/start-omnitool-server')
-                .then(response => response.json())
-                .then(data => startLogPolling());
 
-            
-        }
-        
-        function exitIframe() {
-            if (window.self !== window.top) 
-            {
-                // open a new window to get out of the iframe, then wait 1s,  then fetch burst-iframe then reload
-                window.open(${global.LOCAL_URL}+"?isVisited=true", '_blank');
-            }
-            else
-            {
-                fetch('/burst-iframe').then(response => response.json()).then(data => {window.location.reload();});
-            }
-        }
+                    fetch('/start-omnitool-server')
+                        .then(response => response.json())
+                        .then(data => startLogPolling());
 
-        function refreshPage() {
-            window.location.reload();
-        }
+                    
+                }
+                
+                function exitIframe() {
+                    if (window.self !== window.top) 
+                    {
+                        // open a new window to get out of the iframe, then wait 1s,  then fetch burst-iframe then reload
+                        window.open("${global.PROTOCOL}://${global.LOCAL_URL}/?isVisited=true", '_blank');
+                    }
+                    else
+                    {
+                        fetch('/burst-iframe').then(response => response.json()).then(data => {window.location.reload();});
+                    }
+                }
 
-        function startLogPolling() {
-
-            const interval = setInterval(() => {
-                fetch('/omnitool-logs')
-                    .then(response => response.json())
-                    .then(data => {
-                        const logContainer = document.getElementById('logContainer');
-                        logContainer.innerText = data.logs.join("");
-                        if (data.ready) {
-                            
-                            scrollToBottom(logContainer);
-                            document.getElementById('refreshButton').disabled = false;
-                            document.getElementById('refreshButton').classList.add('highlight-button');
-                            clearInterval(interval);
-
-                        }
-                        else {
-                            scrollToBottom(logContainer);
-                        }
-                    });
-            }, 500);
-
-            const interval2 = setInterval(() => {
-                fetch('/omnitool-logs')
-                    .then(response => response.json())
-                    .then(data => {
-                        const logContainer = document.getElementById('logContainer');
-                        logContainer.innerText = data.logs.join("");
-                        if (data.ready) {
-                            
-                            scrollToBottom(logContainer);
-                            document.getElementById('refreshButton').disabled = false;
-                            document.getElementById('refreshButton').classList.add('highlight-button');
-                            clearInterval(interval);
-
-                        }
-                        else {
-                            scrollToBottom(logContainer);
-                        }
-                    });
-            }, 500);
-        }
+                function refreshPage() {
+                    window.location.reload();
+                }
 
-        function scrollToBottom(element) {
-            element.scrollTop = element.scrollHeight;
-        }
-        
-        startLogPolling();
-    </script>`;
+                function startLogPolling() {
+
+                    const interval = setInterval(() => {
+                        fetch('/omnitool-logs')
+                            .then(response => response.json())
+                            .then(data => {
+                                const logContainer = document.getElementById('logContainer');
+                                logContainer.innerText = data.logs.join("");
+                                if (data.ready) {
+                                    
+                                    scrollToBottom(logContainer);
+                                    document.getElementById('refreshButton').disabled = false;
+                                    document.getElementById('refreshButton').classList.add('highlight-button');
+                                    clearInterval(interval);
+
+                                }
+                                else {
+                                    scrollToBottom(logContainer);
+                                }
+                            });
+                    }, 500);
+
+                    const interval2 = setInterval(() => {
+                        fetch('/omnitool-logs')
+                            .then(response => response.json())
+                            .then(data => {
+                                const logContainer = document.getElementById('logContainer');
+                                logContainer.innerText = data.logs.join("");
+                                if (data.ready) {
+                                    
+                                    scrollToBottom(logContainer);
+                                    document.getElementById('refreshButton').disabled = false;
+                                    document.getElementById('refreshButton').classList.add('highlight-button');
+                                    clearInterval(interval);
+
+                                }
+                                else {
+                                    scrollToBottom(logContainer);
+                                }
+                            });
+                    }, 500);
+                }
+
+                function scrollToBottom(element) {
+                    element.scrollTop = element.scrollHeight;
+                }
+                
+                startLogPolling();
+            </script>`;
+
+        console.log(`scripts_html = ${scripts_html}`);
+        return scripts_html;
+    }
 
 const INTRO_MESSAGE  =                 
     `<h1>Omnitool.ai on Hugging Face</h1>
@@ -289,7 +296,7 @@ async function checkOmnitoolStatus()
         }
     } catch (error)
     {
-        console.log('Please press [CREATE Omnitool Server] button');
+        console.log('Please press [START] button');
         setOmnitoolRunning(false);
     }
 }
@@ -380,6 +387,8 @@ async function handleGetRoot(req, res)
         page_message += `<div>DUPLICATING and using your own SPACE is FREE. You can do it by pressing this button: <a href="${HF_SPACE_DUPLICATE_URL}" target="_blank" class="button-like-link" id="duplicateRepoButton">DUPLICATE SPACE</a></div><div>However, consider choosing the 20 Gig Storage option (~ 5 USD/month) to persist keys, recipes and outputs between server restarts.</div>`;
         page_message += '<p></p>';
         
+        const SCRIPTS = getScriptsHtml();
+
         let buttonsHTML = `
         <a href="${HF_SPACE_DUPLICATE_URL}" target="_blank" class="button-like-link" id="duplicateRepoButton">DUPLICATE SPACE</a>`;
 
@@ -398,6 +407,8 @@ async function handleGetRoot(req, res)
             </body>
         </html>`; 
 
+        console.log(`page_html = ${page_html}`);
+
         res.writeHead(200, { 'Content-Type': 'text/html' });
         res.end(page_html);
         return;
@@ -416,7 +427,7 @@ async function handleGetRoot(req, res)
         let buttons_html = `
             <button id="exitIframeButton" class="${gotoButtonClass}" onclick="exitIframe()">START</button>
             `;
-        
+        const SCRIPTS = getScriptsHtml();
         const page_html = `
             <html>
                 <head>
@@ -432,7 +443,7 @@ async function handleGetRoot(req, res)
                 </body>
             </html>
             `; 
-
+        console.log(`page_html = ${page_html}`);
         res.writeHead(200, { 'Content-Type': 'text/html' });
         res.end(page_html);
         return;
@@ -451,6 +462,7 @@ async function handleGetRoot(req, res)
             <button id="refreshButton" class="${''}" disabled onclick="refreshPage()">REFRESH</button>
             `;
 
+        const SCRIPTS = getScriptsHtml();
         const page_html = `
             <html>
                 <head>
@@ -468,6 +480,7 @@ async function handleGetRoot(req, res)
             </html>
             `; 
 
+        console.log(`page_html = ${page_html}`);    
         res.writeHead(200, { 'Content-Type': 'text/html' });
         res.end(page_html);
         return;