From 5cbbe2fd3d2ec37d990c33f2d72018e526936c6f Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Mon, 24 Feb 2025 08:53:43 +0100
Subject: Fix model and provider in chat completion response Add login button
 to HuggingFace demo Custom conversation ids in chat ui Remove rate limiter in
 demo mode Improve YouTube support in Gemini

---
 g4f/gui/client/demo.html | 49 +++++++++++++++++++++++++++++++-----------------
 1 file changed, 32 insertions(+), 17 deletions(-)

(limited to 'g4f/gui/client/demo.html')

diff --git a/g4f/gui/client/demo.html b/g4f/gui/client/demo.html
index 1d75135f..36c2afed 100644
--- a/g4f/gui/client/demo.html
+++ b/g4f/gui/client/demo.html
@@ -183,25 +183,10 @@
             const isIframe = window.self !== window.top;
             const backendUrl = "{{backend_url}}";
             let url = new URL(window.location.href)
-            let params = new URLSearchParams(url.search);
             if (isIframe && backendUrl) {
-                if (params.get("get_gpu_token")) {
-                    window.addEventListener('DOMContentLoaded', async function() {
-                        const link = document.getElementById("new_window");
-                        link.href = `${backendUrl}${url.search}`;
-                        link.click();
-                    });
-                } else {
-                    window.location.replace(`${backendUrl}${url.search}`);
-                }
+                window.location.replace(`${backendUrl}${url.search}`);
                 return;
             }
-            if (params.get("__sign")) {
-                localStorage.setItem("HuggingSpace-api_key", params.get("__sign"));
-                if (!isIframe) {
-                    window.location.replace("/");
-                }
-            }
         })();
     </script>
     <script src="https://unpkg.com/es-module-shims@1.7.0/dist/es-module-shims.js"></script>
@@ -240,10 +225,13 @@
             <p>
                 <a href="https://huggingface.co/settings/tokens" target="_blank">Get Access Token</a>
             </p>
+            <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/sign-in-with-huggingface-xl-dark.svg" alt="Sign in with Hugging Face" style="cursor: pointer; display: none;" id="signin">
+            <button id="signout" style="display: none">Sign out</button>
         </form>
         <script type="module">
             import * as hub from "@huggingface/hub";
             import { init } from "@huggingface/space-header";
+            import { oauthLoginUrl, oauthHandleRedirectIfPresent } from "@huggingface/hub";
 
             const isIframe = window.self !== window.top;
             const button = document.querySelector('form a.button');
@@ -269,7 +257,6 @@
                     return;
                 }
                 localStorage.setItem("HuggingFace-api_key", accessToken);
-                localStorage.setItem("HuggingFace-user", JSON.stringify(user));
                 localStorage.setItem("user", user.name);
                 localStorage.setItem("report_error", "true")
                 location.href = "/chat/";
@@ -280,6 +267,34 @@
                 event.preventDefault();
                 check_access_token();
             });
+
+            let oauthResult = localStorage.getItem("oauth");
+            if (oauthResult) {
+                try {
+                    oauthResult = JSON.parse(oauthResult);
+                } catch {
+                    oauthResult = null;
+                }
+            }
+            oauthResult ||= await oauthHandleRedirectIfPresent();
+            if (oauthResult) {
+                localStorage.setItem("oauth", JSON.stringify(oauthResult));
+                localStorage.setItem("HuggingFace-api_key", oauthResult.accessToken);
+                localStorage.setItem("user", oauthResult.userInfo.fullname);
+                document.getElementById("signout").style.removeProperty("display");
+                document.getElementById("signout").onclick = async function() {
+                    localStorage.removeItem("oauth");
+                    localStorage.removeItem("HuggingFace-api_key");
+                    window.location.href = window.location.href.replace(/\?.*$/, '');
+                    window.location.reload();
+                }
+            } else {
+                document.getElementById("signin").style.removeProperty("display");
+                document.getElementById("signin").onclick = async function() {
+                    // prompt=consent to re-trigger the consent screen instead of silently redirecting
+                    window.location.href = (await oauthLoginUrl({clientId: 'ed074164-4f8d-4fb2-8bec-44952707965e', scopes: ['inference-api']})) + "&prompt=consent";
+                }
+            }
         </script>
 
         <!-- Footer -->
-- 
cgit v1.2.3