mycofi-earth-website/app/api/zine/regenerate-page/route.ts

import { NextRequest, NextResponse } from "next/server";
import { GoogleGenerativeAI } from "@google/generative-ai";
import { getZine, saveZine, getPageImagePath, readFileAsBase64, savePageImage } from "@/lib/storage";
import type { PageOutline } from "@/lib/gemini";

// Regeneration modes with denoising strengths
const MODE_STRENGTHS: Record<string, number> = {
  refine: 0.25,   // Keep most of image, minor tweaks
  revise: 0.5,    // Keep composition, change elements
  regenerate: 1.0 // Completely new image
};

export async function POST(request: NextRequest) {
  try {
    const body = await request.json();
    const { zineId, pageNumber, currentOutline, feedback, style, tone, mode = "regenerate" } = body;

    if (!zineId || !pageNumber || !currentOutline || !feedback) {
      return NextResponse.json(
        { error: "Missing required fields" },
        { status: 400 }
      );
    }

    const apiKey = process.env.GEMINI_API_KEY;
    if (!apiKey) {
      return NextResponse.json(
        { error: "GEMINI_API_KEY not configured" },
        { status: 500 }
      );
    }

    const genAI = new GoogleGenerativeAI(apiKey);

    // Verify zine exists
    const zine = await getZine(zineId);
    if (!zine) {
      return NextResponse.json(
        { error: "Zine not found" },
        { status: 404 }
      );
    }

    // Update outline based on feedback using Gemini
    const model = genAI.getGenerativeModel({ model: "gemini-2.0-flash" });

    const prompt = `You are refining a zine page based on user feedback.

Current page outline:
- Page Number: ${currentOutline.pageNumber}
- Type: ${currentOutline.type}
- Title: ${currentOutline.title}
- Key Points: ${currentOutline.keyPoints.join(", ")}
- Image Prompt: ${currentOutline.imagePrompt}

User feedback: "${feedback}"

Style: ${style}
Tone: ${tone}

Update the page outline to incorporate this feedback. Keep the same page number and type.

Return ONLY valid JSON (no markdown, no code blocks):
{
  "pageNumber": ${currentOutline.pageNumber},
  "type": "${currentOutline.type}",
  "title": "Updated title if needed",
  "keyPoints": ["Updated point 1", "Updated point 2"],
  "imagePrompt": "Updated detailed image prompt incorporating the feedback"
}`;

    const result = await model.generateContent(prompt);
    const response = result.response.text();

    // Parse the updated outline
    let jsonStr = response;
    if (response.includes("```")) {
      const match = response.match(/```(?:json)?\s*([\s\S]*?)```/);
      if (match) {
        jsonStr = match[1];
      }
    }

    const updatedOutline = JSON.parse(jsonStr.trim()) as PageOutline;

    let imageUrl: string;

    // Use img2img for refine/revise modes, full generation for regenerate
    if (mode === "refine" || mode === "revise") {
      // Get existing image for img2img
      const existingImagePath = await getPageImagePath(zineId, pageNumber);
      if (!existingImagePath) {
        throw new Error("Existing page image not found");
      }

      const existingImageBase64 = await readFileAsBase64(existingImagePath);
      const strength = MODE_STRENGTHS[mode];

      console.log(`Using FLUX img2img with strength ${strength} for ${mode} mode`);

      // Generate with FLUX img2img via Fal.ai
      const newImageBase64 = await generateWithFluxImg2Img(
        existingImageBase64,
        updatedOutline.imagePrompt,
        strength
      );

      // Save the new image
      await savePageImage(zineId, pageNumber, newImageBase64);
      imageUrl = `/api/zine/${zineId}?image=p${pageNumber}&t=${Date.now()}`;
    } else {
      // Full regeneration - use existing generate-page endpoint
      const generateResponse = await fetch(
        new URL("/api/zine/generate-page", request.url),
        {
          method: "POST",
          headers: {
            "Content-Type": "application/json",
          },
          body: JSON.stringify({
            zineId,
            pageNumber,
            outline: updatedOutline,
            style,
            tone,
          }),
        }
      );

      if (!generateResponse.ok) {
        throw new Error("Failed to regenerate image");
      }

      const generateResult = await generateResponse.json();
      imageUrl = generateResult.imageUrl;
    }

    // Update the zine outline
    zine.outline[pageNumber - 1] = updatedOutline;
    zine.updatedAt = new Date().toISOString();
    await saveZine(zine);

    return NextResponse.json({
      pageNumber,
      updatedOutline,
      imageUrl,
      mode,
      success: true,
    });
  } catch (error) {
    console.error("Page regeneration error:", error);
    return NextResponse.json(
      { error: error instanceof Error ? error.message : "Failed to regenerate page" },
      { status: 500 }
    );
  }
}

// FLUX img2img via Fal.ai - using synchronous endpoint
async function generateWithFluxImg2Img(
  imageBase64: string,
  prompt: string,
  strength: number
): Promise<string> {
  const falKey = process.env.FAL_KEY;
  if (!falKey) {
    throw new Error("FAL_KEY not configured");
  }

  console.log(`Calling Fal.ai FLUX img2img API (strength: ${strength})...`);

  // Use fal.run for synchronous execution (waits for result)
  const response = await fetch("https://fal.run/fal-ai/flux/dev/image-to-image", {
    method: "POST",
    headers: {
      "Authorization": `Key ${falKey}`,
      "Content-Type": "application/json",
    },
    body: JSON.stringify({
      image_url: `data:image/png;base64,${imageBase64}`,
      prompt: prompt,
      strength: strength,
      num_inference_steps: 28,
      guidance_scale: 3.5,
      image_size: {
        width: 768,
        height: 1024  // Portrait for zine pages
      },
      output_format: "png"
    }),
  });

  if (!response.ok) {
    const errorText = await response.text();
    console.error("Fal.ai error:", response.status, errorText);
    throw new Error(`Fal.ai API error: ${response.status} - ${errorText}`);
  }

  const result = await response.json();
  console.log("Fal.ai response received, extracting image...");

  if (result.images && result.images.length > 0) {
    const imageUrl = result.images[0].url;
    console.log("✅ Got image URL from Fal.ai, downloading...");
    return await fetchImageAsBase64(imageUrl);
  }

  console.error("Fal.ai response:", JSON.stringify(result).slice(0, 500));
  throw new Error("No image in Fal.ai response");
}

async function fetchImageAsBase64(url: string): Promise<string> {
  const response = await fetch(url);
  if (!response.ok) {
    throw new Error("Failed to fetch generated image");
  }
  const buffer = await response.arrayBuffer();
  return Buffer.from(buffer).toString("base64");
}