Update app/api/image/route.ts

app/api/image/route.ts

@@ -2,9 +2,36 @@ import { NextRequest, NextResponse } from "next/server";
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import { HistoryItem, HistoryPart } from "@/lib/types";
 
-// Initialize the Google Gen AI client with your API key
-const GEMINI_API_KEY = process.env.GEMINI_API_KEY || "";
-const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
+// Function to get a random API key from environment variables
+function getRandomApiKey(): string {
+  // Get all environment variables
+  const envVars = process.env;
+  
+  // Collect all Gemini API keys from environment variables
+  const apiKeys: string[] = [];
+  
+  // Iterate through all environment variables to find Gemini API keys
+  for (const key in envVars) {
+    // Check if the key matches our pattern (GEMINI_API_KEY or GEMINI_API_KEY followed by a number)
+    if (key === 'GEMINI_API_KEY' || key.match(/^GEMINI_API_KEY\d+$/)) {
+      const apiKey = envVars[key];
+      if (apiKey && apiKey.trim() !== '') {
+        apiKeys.push(apiKey);
+      }
+    }
+  }
+  
+  // If no keys found, return empty string
+  if (apiKeys.length === 0) {
+    console.error("No Gemini API keys found in environment variables");
+    return "";
+  }
+  
+  // Select a random API key from the collection
+  const randomIndex = Math.floor(Math.random() * apiKeys.length);
+  console.log(`Using API key ${randomIndex + 1} of ${apiKeys.length} available keys`);
+  return apiKeys[randomIndex];
+}
 
 // Define the model ID for Gemini 2.0 Flash experimental
 const MODEL_ID = "gemini-2.0-flash-exp";
@@ -31,6 +58,18 @@ export async function POST(req: NextRequest) {
       );
     }
 
+    // Get a random API key
+    const GEMINI_API_KEY = getRandomApiKey();
+    if (!GEMINI_API_KEY) {
+      return NextResponse.json(
+        { error: "No valid API key found" },
+        { status: 500 }
+      );
+    }
+
+    // Initialize the Google Gen AI client with the selected API key
+    const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
+
     // Get the model with the correct configuration
     const model = genAI.getGenerativeModel({
       model: MODEL_ID,
@@ -181,4 +220,4 @@ export async function POST(req: NextRequest) {
       { status: 500 }
     );
   }
-}
+}