reload on error

Author: ddPn08

src/scripts/webui/__inject_webui.js

@@ -1,7 +1,10 @@
 /* eslint-disable */
 // @ts-nocheck
 
-if (typeof window['__flat_registered'] === 'undefined') {
+const webui_url = new URL('__WEBUI_URL')
+
+if (webui_url.origin !== document.location.origin) document.location.href = webui_url.href
+else if (typeof window['__flat_registered'] === 'undefined') {
     const { ipcRenderer } = require('electron')
     window['__flat_registered'] = true
     gradioApp().addEventListener('click', (e) => {

src/web/pages/webui/launcher.tsx

@@ -39,7 +39,7 @@ export const createArgs = () => {
     if (typeof val === 'string' && !val) continue
     if (!key.startsWith('webui/args')) continue
     const arg = key.split('/').slice(-1)[0]
-    if (arg === 'custom') result += val
+    if (arg === 'custom') result += ` ${val}`
     else if (typeof val === 'boolean' && val) result += ` --${arg}`
     else if (typeof val === 'string') result += ` --${arg} "${val}"`.replace(/\\/g, '\\\\')
     else result += ` --${arg} ${val}`

src/web/pages/webui/ui.tsx

@@ -96,7 +96,7 @@ export const UI: Component = () => {
       }
     })
     webview.addEventListener('dom-ready', () => {
-      webview.executeJavaScript(__inject_webui)
+      webview.executeJavaScript(__inject_webui.replace('__WEBUI_URL', url))
     })
   }