feat. Check if user authentication occurs when the “submit” or “improved prompt” button is clicked.

src/components/prompt-input.tsx

@@ -10,6 +10,7 @@ import { ColorPanel } from "./color-panel"
 import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from "@/components/ui/tooltip"
 import { FullscreenToggle } from "./ui/fullscreen-toggle"
 import { AuthErrorPopup } from "./auth-error-popup"
+import { getInferenceToken } from "@/lib/auth"
 
 interface PromptInputProps {
   onSubmit: (prompt: string, colors: string[]) => Promise<void>;
@@ -43,10 +44,24 @@ export function PromptInput({
     }
   }, [improveError, onImproveError]);
 
+  const checkAuth = async (): Promise<boolean> => {
+    try {
+      const token = await getInferenceToken();
+      return !!token;
+    } catch (error) {
+      setShowAuthError(true);
+      return false;
+    }
+  }
+
   const handleSubmit = async (e: React.FormEvent) => {
     e.preventDefault();
     if (prompt.trim() === '' || isLoading) return;
 
+    // Check for authentication
+    const isAuthenticated = await checkAuth();
+    if (!isAuthenticated) return;
+
     // Clear any previous errors
     setImproveError(null);
     await onSubmit(prompt, selectedColors);
@@ -55,6 +70,10 @@ export function PromptInput({
   const improvePrompt = async () => {
     if (prompt.trim() === '' || isImprovingPrompt || isLoading) return;
 
+    // Check for authentication
+    const isAuthenticated = await checkAuth();
+    if (!isAuthenticated) return;
+
     // Clear previous errors
     setImproveError(null);
     setShowAuthError(false);

src/lib/auth.ts

@@ -0,0 +1,20 @@
+/**
+ * Client-side authentication utilities
+ */
+
+/**
+ * Get the inference token from cookies
+ * @returns Promise that resolves to the token string or throws an error if no token is found
+ */
+export async function getInferenceToken(): Promise<string> {
+  // Get token from cookie
+  const cookies = document.cookie.split('; ');
+  const tokenCookie = cookies.find(cookie => cookie.startsWith('hf_token='));
+  const token = tokenCookie?.split('=')[1];
+
+  if (!token) {
+    throw new Error('Authentication required');
+  }
+
+  return token;
+}