try to fix the nb of panels issue

src/app/main.tsx

@@ -75,16 +75,13 @@ export default function Main() {
         currentPanel < nbTotalPanels;
         currentPanel += nbPanelsToGenerate
       ) {
-        if (currentPanel > (nbTotalPanels / 2)) {
-          console.log("good, we are half way there, hold tight!")
-          // setWaitABitMore(true)
-        }
         try {
           const candidatePanels = await getStoryContinuation({
             preset,
             stylePrompt,
             userStoryPrompt,
             nbPanelsToGenerate,
+            nbTotalPanels,
             existingPanels,
           })
           console.log("LLM generated some new panels:", candidatePanels)
@@ -123,6 +120,10 @@ export default function Main() {
           setGeneratingStory(false)
           break
         }
+        if (currentPanel > (nbTotalPanels / 2)) {
+          console.log("good, we are half way there, hold tight!")
+          // setWaitABitMore(true)
+        }
       }
    
       /*

src/app/queries/getStoryContinuation.ts

@@ -8,13 +8,15 @@ export const getStoryContinuation = async ({
   stylePrompt = "",
   userStoryPrompt = "",
   nbPanelsToGenerate = 2,
+  nbTotalPanels = 8,
   existingPanels = [],
 }: {
   preset: Preset;
-  stylePrompt: string;
-  userStoryPrompt: string;
-  nbPanelsToGenerate: number;
-  existingPanels: GeneratedPanel[];
+  stylePrompt?: string;
+  userStoryPrompt?: string;
+  nbPanelsToGenerate?: number;
+  nbTotalPanels?: number;
+  existingPanels?: GeneratedPanel[];
 }): Promise<GeneratedPanel[]> => {
 
   let panels: GeneratedPanel[] = []
@@ -29,6 +31,7 @@ export const getStoryContinuation = async ({
       preset,
       prompt,
       nbPanelsToGenerate,
+      nbTotalPanels,
       existingPanels,
     })
 
@@ -49,7 +52,7 @@ export const getStoryContinuation = async ({
     // console.log("LLM step failed due to:", err)
     // console.log("we are now switching to a degraded mode, using 4 similar panels")
     panels = []
-    for (let p = startAt; p < endAt; p++) {
+    for (let p = startAt; p < endAt && p; p++) {
       panels.push({
         panel: p,
         instructions: joinWords([

src/app/queries/predictNextPanels.ts

@@ -11,11 +11,13 @@ export const predictNextPanels = async ({
   preset,
   prompt = "",
   nbPanelsToGenerate = 2,
+  nbTotalPanels = 8,
   existingPanels = [],
 }: {
   preset: Preset;
   prompt: string;
-  nbPanelsToGenerate: number;
+  nbPanelsToGenerate?: number;
+  nbTotalPanels?: number;
   existingPanels: GeneratedPanel[];
 }): Promise<GeneratedPanel[]> => {
   // console.log("predictNextPanels: ", { prompt, nbPanelsToGenerate })
@@ -28,12 +30,20 @@ export const predictNextPanels = async ({
     ? ` To help you, here are the previous panels and their captions (note: if you see an anomaly here eg. no caption or the same description repeated multiple times, do not hesitate to fix the story): ${JSON.stringify(existingPanels, null, 2)}`
     : ''
 
+
+  const firstNextOrLast =
+    existingPanels.length === 0
+      ? "first"
+      : (nbTotalPanels - existingPanels.length) === nbTotalPanels
+      ? "last"
+      : "next"
+
   const query = createZephyrPrompt([
     {
       role: "system",
       content: [
         `You are a writer specialized in ${preset.llmPrompt}`,
-        `Please write detailed drawing instructions and short (2-3 sentences long) speech captions for the next ${nbPanelsToGenerate} panels of a new story, but keep it open-ended (it will be continued and expanded later). Please make sure each of those ${nbPanelsToGenerate} panels include info about character gender, age, origin, clothes, colors, location, lights, etc.`,
+        `Please write detailed drawing instructions and short (2-3 sentences long) speech captions for the ${firstNextOrLast} ${nbPanelsToGenerate} panels (out of ${nbTotalPanels}) of a new story, but keep it open-ended (it will be continued and expanded later). Please make sure each of those ${nbPanelsToGenerate} panels include info about character gender, age, origin, clothes, colors, location, lights, etc.`,
         `Give your response as a VALID JSON array like this: \`Array<{ panel: number; instructions: string; caption: string; }>\`.`,
         // `Give your response as Markdown bullet points.`,
         `Be brief in your ${nbPanelsToGenerate} instructions and narrative captions, don't add your own comments. The captions must be captivating, smart, entertaining. Be straight to the point, and never reply things like "Sure, I can.." etc. Reply using valid JSON!! Important: Write valid JSON!`