error handling

src/lib/components/MonsterGenerator/MonsterGenerator.svelte

@@ -83,12 +83,49 @@ Assistant:`;
       state.currentStep = 'complete';
     } catch (err) {
       console.error('Workflow error:', err);
-      state.error = err instanceof Error ? err.message : 'An unknown error occurred';
+      
+      // Check for GPU quota error
+      if (err && typeof err === 'object' && 'message' in err) {
+        const errorMessage = String(err.message);
+        if (errorMessage.includes('exceeded your GPU quota') || errorMessage.includes('GPU quota')) {
+          state.error = 'GPU quota exceeded! You need to sign in with Hugging Face for free GPU time, or upgrade to Hugging Face Pro for more quota.';
+        } else {
+          state.error = errorMessage;
+        }
+      } else if (err instanceof Error) {
+        state.error = err.message;
+      } else {
+        state.error = 'An unknown error occurred';
+      }
     } finally {
       state.isProcessing = false;
     }
   }
   
+  function handleAPIError(error: any): never {
+    console.error('API Error:', error);
+    
+    // Check if it's a GPU quota error
+    if (error && typeof error === 'object' && 'message' in error) {
+      const errorMessage = String(error.message);
+      if (errorMessage.includes('exceeded your GPU quota') || errorMessage.includes('GPU quota')) {
+        throw new Error('GPU quota exceeded! You need to sign in with Hugging Face for free GPU time, or upgrade to Hugging Face Pro for more quota.');
+      }
+      throw new Error(errorMessage);
+    }
+    
+    // Check if error has a different structure (like the status object from the logs)
+    if (error && typeof error === 'object' && 'type' in error && error.type === 'status') {
+      const statusError = error as any;
+      if (statusError.message && statusError.message.includes('GPU quota')) {
+        throw new Error('GPU quota exceeded! You need to sign in with Hugging Face for free GPU time, or upgrade to Hugging Face Pro for more quota.');
+      }
+      throw new Error(statusError.message || 'API request failed');
+    }
+    
+    throw error;
+  }
+  
   async function captionImage() {
     state.currentStep = 'captioning';
     
@@ -96,18 +133,22 @@ Assistant:`;
       throw new Error('Caption service not available or no image provided');
     }
     
-    const output = await joyCaptionClient.predict("/stream_chat", [
-      state.userImage,
-      "Descriptive" as CaptionType,
-      "very long" as CaptionLength,
-      [], // extra_options
-      "", // name_input
-      ""  // custom_prompt
-    ]);
-    
-    const [prompt, caption] = output.data;
-    state.imageCaption = caption;
-    console.log('Caption generated:', caption);
+    try {
+      const output = await joyCaptionClient.predict("/stream_chat", [
+        state.userImage,
+        "Descriptive" as CaptionType,
+        "very long" as CaptionLength,
+        [], // extra_options
+        "", // name_input
+        ""  // custom_prompt
+      ]);
+      
+      const [prompt, caption] = output.data;
+      state.imageCaption = caption;
+      console.log('Caption generated:', caption);
+    } catch (error) {
+      handleAPIError(error);
+    }
   }
   
   async function generateMonsterConcept() {
@@ -121,21 +162,25 @@ Assistant:`;
     
     console.log('Generating monster concept with prompt:', conceptPrompt);
     
-    const output = await rwkvClient.predict(0, [
-      conceptPrompt,
-      300,   // maxTokens
-      0.7,   // temperature
-      0.8,   // topP
-      0.1,   // presencePenalty
-      0.1    // countPenalty
-    ]);
-    
-    console.log('RWKV output:', output);
-    state.monsterConcept = output.data[0];
-    console.log('Monster concept generated:', state.monsterConcept);
-    
-    if (!state.monsterConcept || state.monsterConcept.trim() === '') {
-      throw new Error('Failed to generate monster concept - received empty response');
+    try {
+      const output = await rwkvClient.predict(0, [
+        conceptPrompt,
+        300,   // maxTokens
+        0.7,   // temperature
+        0.8,   // topP
+        0.1,   // presencePenalty
+        0.1    // countPenalty
+      ]);
+      
+      console.log('RWKV output:', output);
+      state.monsterConcept = output.data[0];
+      console.log('Monster concept generated:', state.monsterConcept);
+      
+      if (!state.monsterConcept || state.monsterConcept.trim() === '') {
+        throw new Error('Failed to generate monster concept - received empty response');
+      }
+    } catch (error) {
+      handleAPIError(error);
     }
   }
   
@@ -150,21 +195,25 @@ Assistant:`;
     
     console.log('Generating image prompt from concept');
     
-    const output = await rwkvClient.predict(0, [
-      promptGenerationPrompt,
-      300,   // maxTokens
-      0.7,   // temperature
-      0.8,   // topP
-      0.1,   // presencePenalty
-      0.1    // countPenalty
-    ]);
-    
-    console.log('Image prompt output:', output);
-    state.imagePrompt = output.data[0];
-    console.log('Image prompt generated:', state.imagePrompt);
-    
-    if (!state.imagePrompt || state.imagePrompt.trim() === '') {
-      throw new Error('Failed to generate image prompt - received empty response');
+    try {
+      const output = await rwkvClient.predict(0, [
+        promptGenerationPrompt,
+        300,   // maxTokens
+        0.7,   // temperature
+        0.8,   // topP
+        0.1,   // presencePenalty
+        0.1    // countPenalty
+      ]);
+      
+      console.log('Image prompt output:', output);
+      state.imagePrompt = output.data[0];
+      console.log('Image prompt generated:', state.imagePrompt);
+      
+      if (!state.imagePrompt || state.imagePrompt.trim() === '') {
+        throw new Error('Failed to generate image prompt - received empty response');
+      }
+    } catch (error) {
+      handleAPIError(error);
     }
   }
   
@@ -175,30 +224,34 @@ Assistant:`;
       throw new Error('Image generation service not available or no prompt');
     }
     
-    const output = await fluxClient.predict("/infer", [
-      `${state.imagePrompt}\nNow generate an Anime-style image of the monster in an idle pose. The monster should not be attacking or in motion. The full monster must be visible within the frame.`,
-      0,      // seed
-      true,   // randomizeSeed
-      1024,   // width
-      1024,   // height
-      4       // steps
-    ]);
-    
-    const [image, usedSeed] = output.data;
-    let url: string | undefined;
-    
-    if (typeof image === "string") url = image;
-    else if (image && image.url) url = image.url;
-    else if (image && image.path) url = image.path;
-    
-    if (url) {
-      state.monsterImage = {
-        imageUrl: url,
-        seed: usedSeed,
-        prompt: state.imagePrompt
-      };
-    } else {
-      throw new Error('Failed to generate monster image');
+    try {
+      const output = await fluxClient.predict("/infer", [
+        `${state.imagePrompt}\nNow generate an Anime-style image of the monster in an idle pose with a white background. The monster should not be attacking or in motion. The full monster must be visible within the frame.`,
+        0,      // seed
+        true,   // randomizeSeed
+        1024,   // width
+        1024,   // height
+        4       // steps
+      ]);
+      
+      const [image, usedSeed] = output.data;
+      let url: string | undefined;
+      
+      if (typeof image === "string") url = image;
+      else if (image && image.url) url = image.url;
+      else if (image && image.path) url = image.path;
+      
+      if (url) {
+        state.monsterImage = {
+          imageUrl: url,
+          seed: usedSeed,
+          prompt: state.imagePrompt
+        };
+      } else {
+        throw new Error('Failed to generate monster image');
+      }
+    } catch (error) {
+      handleAPIError(error);
     }
   }