test hf signin

README.md

@@ -7,6 +7,8 @@ sdk: docker
 pinned: true
 app_port: 3002
 license: mit
+
+hf_oauth: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

components/main/index.tsx

@@ -19,11 +19,13 @@ const categories = [
   {
     key: "my-own",
     label: "My generations",
+    isLogged: true,
     icon: <HiHeart className="text-2xl" />,
   },
 ];
 
 export const Main = () => {
+  const user = false;
   const { list_styles, style, setStyle, loading } = useInputGeneration();
   const [category, setCategory] = useState<string>("community");
   const [advancedSettings, setAdvancedSettings] = useState<boolean>(false);
@@ -34,17 +36,31 @@ export const Main = () => {
         <div className="flex flex-col lg:flex-row items-center justify-between w-full">
           <InputGeneration />
           <div className="items-center justify-center lg:justify-end gap-5 w-full mt-6 lg:mt-0 hidden lg:flex">
-            {categories.map(({ key, label, icon }) => (
-              <Button
-                key={key}
-                disabled={loading}
-                theme={key !== category ? "white" : "primary"}
-                onClick={() => setCategory(key)}
-              >
-                {icon}
-                {label}
-              </Button>
-            ))}
+            {categories.map(({ key, label, icon, isLogged }) =>
+              isLogged && !user ? (
+                <Button
+                  key={key}
+                  theme="white"
+                  onClick={() =>
+                    console.log(
+                      `https://huggingface.co/oauth/authorize?redirect_uri=${process.env.REDIRECT_URI}&scope=openid%20profile&client_id=${process.env.CLIENT_ID}&state=${process.env.STATE}`
+                    )
+                  }
+                >
+                  Sign in with Hugging Face
+                </Button>
+              ) : (
+                <Button
+                  key={key}
+                  disabled={loading}
+                  theme={key !== category ? "white" : "primary"}
+                  onClick={() => setCategory(key)}
+                >
+                  {icon}
+                  {label}
+                </Button>
+              )
+            )}
           </div>
         </div>
       </div>

utils/checker/prompt.ts

@@ -8,7 +8,7 @@ export const isTextNSFW = async (prompt: string, headers: any) => {
       headers: headers,
     })
     const res = await request.clone().json().catch(() => ({}));
-    const isNSFW = res?.[0]?.find((item: { label: string, score: number }) => item?.label === "NSFW")?.score > 0.92 ?? false
+    const isNSFW = res?.[0]?.find((item: { label: string, score: number }) => item?.label === "NSFW")?.score > 0.73 ?? false
     resolve(isNSFW)
   })
 }