Upload 53 files

ai-routes.js

@@ -2,38 +2,31 @@
 const express = require('express');
 const router = express.Router();
 const OpenAI = require('openai');
-// Removed static require causing ERR_REQUIRE_ESM
-// const { GoogleGenAI } = require("@google/genai"); 
 const { ConfigModel, User } = require('./models');
 
-// --- AI Client Initialization ---
+// --- Key Management & Rotation ---
 
-let openAIClient = null;
-function getOpenRouter() {
-    if (openAIClient) return openAIClient;
-    // 使用环境变量或硬编码的备用 Key (仅供演示，实际应配置在 .env)
-    const apiKey = process.env.OPENROUTER_API_KEY; 
-    if (!apiKey) return null;
+// Fetch keys from DB + merge with ENV variables
+async function getKeyPool(type) {
+    const config = await ConfigModel.findOne({ key: 'main' });
+    const pool = [];
     
-    openAIClient = new OpenAI({
-        baseURL: "https://openrouter.ai/api/v1",
-        apiKey: apiKey,
-        defaultHeaders: {
-            "HTTP-Referer": "https://smart-school-system.com", 
-            "X-Title": "Smart School System", 
-        },
-    });
-    return openAIClient;
-}
+    // 1. Add DB keys first (Priority)
+    if (config && config.apiKeys && config.apiKeys[type] && Array.isArray(config.apiKeys[type])) {
+        config.apiKeys[type].forEach(k => {
+            if (k && k.trim()) pool.push(k.trim());
+        });
+    }
 
-// Initialize Gemini Client Dynamically (Lazy Load)
-let geminiClientInstance = null;
-async function getGeminiClient() {
-    if (geminiClientInstance) return geminiClientInstance;
-    // Dynamic import to support ESM package in CJS environment
-    const { GoogleGenAI } = await import("@google/genai");
-    geminiClientInstance = new GoogleGenAI({ apiKey: process.env.API_KEY });
-    return geminiClientInstance;
+    // 2. Add Environment keys as fallback
+    if (type === 'gemini' && process.env.API_KEY) {
+        if (!pool.includes(process.env.API_KEY)) pool.push(process.env.API_KEY);
+    }
+    if (type === 'openrouter' && process.env.OPENROUTER_API_KEY) {
+        if (!pool.includes(process.env.OPENROUTER_API_KEY)) pool.push(process.env.OPENROUTER_API_KEY);
+    }
+
+    return pool;
 }
 
 // --- Helpers ---
@@ -94,49 +87,6 @@ function convertGeminiToOpenAI(baseParams) {
     return messages;
 }
 
-// Helper to adapt params for Gemma (Remove systemInstruction and prepend to prompt)
-function prepareGemmaParams(baseParams, modelName) {
-    // Deep copy to avoid mutating original for other providers
-    const newParams = JSON.parse(JSON.stringify(baseParams));
-    newParams.model = modelName;
-
-    // Gemma models do not support systemInstruction in config, move it to text
-    if (newParams.config && newParams.config.systemInstruction) {
-        const sysInst = newParams.config.systemInstruction;
-        delete newParams.config.systemInstruction;
-
-        const systemPrefix = `[System Instruction: ${sysInst}]\n\n`;
-
-        if (Array.isArray(newParams.contents) && newParams.contents.length > 0) {
-            // Chat mode: Prepend to first user message
-            const firstMsg = newParams.contents[0];
-            if (firstMsg.role === 'user' && firstMsg.parts && firstMsg.parts.length > 0) {
-                const textPart = firstMsg.parts.find(p => p.text);
-                if (textPart) {
-                    textPart.text = systemPrefix + textPart.text;
-                } else {
-                    // Only has image/audio, add text part
-                    firstMsg.parts.unshift({ text: systemPrefix });
-                }
-            } else {
-                // If history starts with model (rare), prepend a new user message
-                newParams.contents.unshift({ role: 'user', parts: [{ text: systemPrefix }] });
-            }
-        } else if (newParams.contents && newParams.contents.parts) {
-            // Generate content mode (single object)
-            if (Array.isArray(newParams.contents.parts)) {
-                const textPart = newParams.contents.parts.find(p => p.text);
-                if (textPart) {
-                    textPart.text = systemPrefix + textPart.text;
-                } else {
-                    newParams.contents.parts.unshift({ text: systemPrefix });
-                }
-            }
-        }
-    }
-    return newParams;
-}
-
 // --- Dynamic Provider Management ---
 
 const PROVIDERS = {
@@ -167,83 +117,125 @@ function isQuotaError(e) {
 // --- Provider Callers (Generate Content) ---
 
 async function callGeminiProvider(baseParams) {
-    const client = await getGeminiClient(); // AWAIT client
+    const { GoogleGenAI } = await import("@google/genai");
     const primaryModels = ['gemini-2.5-flash', 'gemini-2.5-flash-lite'];
+    
+    // Get all available keys
+    const keys = await getKeyPool('gemini');
+    if (keys.length === 0) throw new Error("No Gemini API keys configured");
+
     let lastError = null;
-    for (const modelName of primaryModels) {
-        try {
-            console.log(`🚀 [AI Debug] Calling Gemini non-stream: ${modelName}`);
-            const currentParams = { ...baseParams, model: modelName };
-            return await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
-        } catch (e) {
-            lastError = e;
-            console.error(`⚠️ [AI Debug] Gemini ${modelName} Error:`, e.message);
-            if (isQuotaError(e)) continue; 
-            throw e; 
+
+    // KEY ROTATION LOOP
+    for (const apiKey of keys) {
+        const client = new GoogleGenAI({ apiKey });
+        
+        for (const modelName of primaryModels) {
+            try {
+                console.log(`🚀 [AI Debug] Calling Gemini non-stream: ${modelName} (Key ends: ${apiKey.slice(-4)})`);
+                const currentParams = { ...baseParams, model: modelName };
+                return await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
+            } catch (e) {
+                lastError = e;
+                console.error(`⚠️ [AI Debug] Gemini ${modelName} Error with key ${apiKey.slice(-4)}:`, e.message);
+                
+                // If it's a quota error, break the model loop to try next KEY
+                if (isQuotaError(e)) {
+                    break; // Try next key
+                }
+                // If it's NOT a quota error (e.g. invalid prompt), throw immediately to avoid burning other keys
+                throw e; 
+            }
         }
     }
-    throw lastError; 
+    // If we exhausted all keys with quota errors
+    throw lastError || new Error("All Gemini keys exhausted"); 
 }
 
 async function callOpenRouterProvider(baseParams) {
-    const openRouter = getOpenRouter();
-    if (!openRouter) throw new Error("OpenRouter not configured");
-
     const openRouterModels = [
         'qwen/qwen3-coder:free',
         'openai/gpt-oss-120b:free',
         'qwen/qwen3-235b-a22b:free',
         'tngtech/deepseek-r1t-chimera:free'
     ];
-
     const openAIMessages = convertGeminiToOpenAI(baseParams);
+    
+    const keys = await getKeyPool('openrouter');
+    if (keys.length === 0) throw new Error("No OpenRouter API keys configured");
+
     let lastError = null;
 
-    for (const modelName of openRouterModels) {
-        try {
-            console.log(`🛡️ [AI Debug] Switching to OpenRouter Model: ${modelName}`);
-            const completion = await openRouter.chat.completions.create({
-                model: modelName,
-                messages: openAIMessages,
-            });
-            if (!completion || !completion.choices || !completion.choices[0]) {
-                throw new Error(`Invalid response structure from ${modelName}`);
+    for (const apiKey of keys) {
+        const client = new OpenAI({
+            baseURL: "https://openrouter.ai/api/v1",
+            apiKey: apiKey,
+            defaultHeaders: {
+                "HTTP-Referer": "https://smart-school-system.com", 
+                "X-Title": "Smart School System", 
+            },
+        });
+
+        for (const modelName of openRouterModels) {
+            try {
+                console.log(`🛡️ [AI Debug] Switching to OpenRouter Model: ${modelName} (Key ends: ${apiKey.slice(-4)})`);
+                const completion = await client.chat.completions.create({
+                    model: modelName,
+                    messages: openAIMessages,
+                });
+                if (!completion || !completion.choices || !completion.choices[0]) {
+                    throw new Error(`Invalid response structure from ${modelName}`);
+                }
+                const content = completion.choices[0].message.content || "";
+                return { text: content }; 
+            } catch (e) {
+                lastError = e;
+                console.warn(`⚠️ [AI Debug] OpenRouter Model ${modelName} failed with key ${apiKey.slice(-4)}.`, e.message);
+                if (isQuotaError(e)) break; // Try next key
+                // Keep trying models if it's not a quota error? OpenRouter often has specific model downtimes.
+                // But for key rotation logic, usually 429 applies to account.
             }
-            const content = completion.choices[0].message.content || "";
-            // Mock a Gemini-like response object for compatibility
-            return { text: content }; // Correct property access for text
-        } catch (e) {
-            lastError = e;
-            console.warn(`⚠️ [AI Debug] OpenRouter Model ${modelName} failed.`, e.message);
         }
     }
-    throw lastError || new Error("OpenRouter failed");
+    throw lastError || new Error("OpenRouter failed on all keys");
 }
 
 async function callGemmaProvider(baseParams) {
-    const client = await getGeminiClient(); // AWAIT client
-    // Updated to Gemma 3 series as per user quota (including smaller ones)
-    const fallbackModels = ['gemma-3-27b-it', 'gemma-3-9b-it', 'gemma-3-4b-it']; 
+    // Gemma uses Gemini API Keys
+    const { GoogleGenAI } = await import("@google/genai");
+    const fallbackModels = ['gemma-3-27b-it', 'gemma-3-12b-it', 'gemma-3-4b-it']; 
+    const gemmaConfig = { ...baseParams.config };
+    if (gemmaConfig.systemInstruction) delete gemmaConfig.systemInstruction;
+
+    const keys = await getKeyPool('gemini'); // Shared pool
+    if (keys.length === 0) throw new Error("No Gemini API keys for Gemma");
+
     let lastError = null;
-    for (const modelName of fallbackModels) {
-        try {
-            console.log(`🛡️ [AI Debug] Switching to Backup (Gemma): ${modelName}`);
-            // Use adapter to remove systemInstruction
-            const currentParams = prepareGemmaParams(baseParams, modelName);
-            return await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
-        } catch (e) {
-            lastError = e;
-            console.warn(`⚠️ [AI Debug] Backup Model ${modelName} failed.`, e.message);
+    for (const apiKey of keys) {
+        const client = new GoogleGenAI({ apiKey });
+        for (const modelName of fallbackModels) {
+            try {
+                console.log(`🛡️ [AI Debug] Switching to Backup (Gemma): ${modelName} (Key ends: ${apiKey.slice(-4)})`);
+                const currentParams = { 
+                    ...baseParams, 
+                    model: modelName,
+                    config: gemmaConfig 
+                };
+                return await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
+            } catch (e) {
+                lastError = e;
+                console.warn(`⚠️ [AI Debug] Backup Model ${modelName} failed with key ${apiKey.slice(-4)}.`, e.message);
+                if (isQuotaError(e)) break; // Try next key
+            }
         }
     }
-    throw lastError || new Error("Gemma failed");
+    throw lastError || new Error("Gemma failed on all keys");
 }
 
 async function generateContentWithSmartFallback(baseParams) {
     // If input has audio, force Gemini as others likely don't support multimodal audio directly
     let hasAudio = false;
     if (baseParams.contents) {
-        // Handle both simple and complex content structures
         const contents = Array.isArray(baseParams.contents) ? baseParams.contents : [baseParams.contents];
         contents.forEach(c => {
             if (c.parts) c.parts.forEach(p => {
@@ -275,99 +267,128 @@ async function generateContentWithSmartFallback(baseParams) {
 // --- Streaming Helpers ---
 
 async function streamGemini(baseParams, res) {
-    const client = await getGeminiClient(); // AWAIT client
+    const { GoogleGenAI } = await import("@google/genai");
     const models = ['gemini-2.5-flash', 'gemini-2.5-flash-lite'];
+    const keys = await getKeyPool('gemini');
+    if (keys.length === 0) throw new Error("No Gemini API keys");
+
     let lastError = null;
-    for (const modelName of models) {
-        try {
-            console.log(`🌊 [AI Debug] STREAMING Gemini: ${modelName}`);
-            const currentParams = { ...baseParams, model: modelName };
-            const result = await client.models.generateContentStream(currentParams);
-            
-            let fullText = "";
-            // FIX: Iterate result directly, NOT result.stream
-            for await (const chunk of result) {
-                // FIX: Access property .text, NOT method .text()
-                const chunkText = chunk.text;
-                if (chunkText) {
-                    fullText += chunkText;
-                    res.write(`data: ${JSON.stringify({ text: chunkText })}\n\n`);
-                    if (res.flush) res.flush(); 
+
+    for (const apiKey of keys) {
+        const client = new GoogleGenAI({ apiKey });
+        for (const modelName of models) {
+            try {
+                console.log(`🌊 [AI Debug] STREAMING Gemini: ${modelName} (Key ends: ${apiKey.slice(-4)})`);
+                const currentParams = { ...baseParams, model: modelName };
+                const result = await client.models.generateContentStream(currentParams);
+                
+                let fullText = "";
+                for await (const chunk of result) {
+                    const chunkText = chunk.text;
+                    if (chunkText) {
+                        fullText += chunkText;
+                        res.write(`data: ${JSON.stringify({ text: chunkText })}\n\n`);
+                        if (res.flush) res.flush(); 
+                    }
                 }
+                return fullText; 
+            } catch (e) {
+                lastError = e;
+                console.error(`Streaming GEMINI failed with key ${apiKey.slice(-4)}: ${e.message}`);
+                if (isQuotaError(e)) break; // Try next key
+                throw e; 
             }
-            return fullText; 
-        } catch (e) {
-            lastError = e;
-            console.error(`Streaming GEMINI failed: ${e.message}`);
-            if (isQuotaError(e)) continue; 
-            throw e; 
         }
     }
-    throw lastError || new Error("Gemini streaming failed");
+    throw lastError || new Error("Gemini streaming failed on all keys");
 }
 
 async function streamOpenRouter(baseParams, res) {
-    const openRouter = getOpenRouter();
-    if (!openRouter) throw new Error("OpenRouter not configured");
-    
     const openRouterModels = ['qwen/qwen3-coder:free', 'openai/gpt-oss-120b:free'];
     const messages = convertGeminiToOpenAI(baseParams);
     
+    const keys = await getKeyPool('openrouter');
+    if (keys.length === 0) throw new Error("No OpenRouter API keys");
+
     let lastError = null;
-    for (const modelName of openRouterModels) {
-        try {
-            console.log(`🛡️ [AI Debug] STREAMING OpenRouter: ${modelName}`);
-            const stream = await openRouter.chat.completions.create({
-                model: modelName,
-                messages: messages,
-                stream: true
-            });
-            let fullText = '';
-            for await (const chunk of stream) {
-                const text = chunk.choices[0]?.delta?.content || '';
-                if (text) {
-                    fullText += text;
-                    res.write(`data: ${JSON.stringify({ text: text })}\n\n`);
-                    if (res.flush) res.flush();
+
+    for (const apiKey of keys) {
+        const client = new OpenAI({
+            baseURL: "https://openrouter.ai/api/v1",
+            apiKey: apiKey,
+            defaultHeaders: {
+                "HTTP-Referer": "https://smart-school-system.com", 
+                "X-Title": "Smart School System", 
+            },
+        });
+
+        for (const modelName of openRouterModels) {
+            try {
+                console.log(`🛡️ [AI Debug] STREAMING OpenRouter: ${modelName} (Key ends: ${apiKey.slice(-4)})`);
+                const stream = await client.chat.completions.create({
+                    model: modelName,
+                    messages: messages,
+                    stream: true
+                });
+                let fullText = '';
+                for await (const chunk of stream) {
+                    const text = chunk.choices[0]?.delta?.content || '';
+                    if (text) {
+                        fullText += text;
+                        res.write(`data: ${JSON.stringify({ text: text })}\n\n`);
+                        if (res.flush) res.flush();
+                    }
                 }
+                return fullText;
+            } catch (e) {
+                lastError = e;
+                console.warn(`[AI Debug] Stream OpenRouter ${modelName} failed with key ${apiKey.slice(-4)}`, e.message);
+                if (isQuotaError(e)) break; // Try next key
             }
-            return fullText;
-        } catch (e) {
-            lastError = e;
-            console.warn(`[AI Debug] Stream OpenRouter ${modelName} failed`, e.message);
         }
     }
-    throw lastError || new Error("OpenRouter stream failed");
+    throw lastError || new Error("OpenRouter stream failed on all keys");
 }
 
 async function streamGemma(baseParams, res) {
-    const client = await getGeminiClient(); // AWAIT client
-    // Updated to Gemma 3 series as per user quota
-    const models = ['gemma-3-27b-it', 'gemma-3-9b-it', 'gemma-3-4b-it'];
+    const { GoogleGenAI } = await import("@google/genai");
+    const models = ['gemma-3-27b-it', 'gemma-3-12b-it', 'gemma-3-4b-it'];
+    const gemmaConfig = { ...baseParams.config };
+    if (gemmaConfig.systemInstruction) delete gemmaConfig.systemInstruction;
+
+    const keys = await getKeyPool('gemini'); // Shared with Gemini
+    if (keys.length === 0) throw new Error("No Gemini keys for Gemma");
+
     let lastError = null;
-    for (const modelName of models) {
-        try {
-            console.log(`🛡️ [AI Debug] STREAMING Gemma: ${modelName}`);
-            // Use adapter to remove systemInstruction
-            const currentParams = prepareGemmaParams(baseParams, modelName);
-            const result = await client.models.generateContentStream(currentParams);
-            let fullText = "";
-            // FIX: Iterate result directly
-            for await (const chunk of result) {
-                const chunkText = chunk.text;
-                if (chunkText) {
-                    fullText += chunkText;
-                    res.write(`data: ${JSON.stringify({ text: chunkText })}\n\n`);
-                    if (res.flush) res.flush(); 
+    for (const apiKey of keys) {
+        const client = new GoogleGenAI({ apiKey });
+        for (const modelName of models) {
+            try {
+                console.log(`🛡️ [AI Debug] STREAMING Gemma: ${modelName} (Key ends: ${apiKey.slice(-4)})`);
+                const currentParams = { 
+                    ...baseParams, 
+                    model: modelName,
+                    config: gemmaConfig 
+                };
+                const result = await client.models.generateContentStream(currentParams);
+                let fullText = "";
+                for await (const chunk of result) {
+                    const chunkText = chunk.text;
+                    if (chunkText) {
+                        fullText += chunkText;
+                        res.write(`data: ${JSON.stringify({ text: chunkText })}\n\n`);
+                        if (res.flush) res.flush(); 
+                    }
                 }
+                return fullText;
+            } catch (e) {
+                lastError = e;
+                console.warn(`Streaming GEMMA ${modelName} failed with key ${apiKey.slice(-4)}:`, e.message);
+                if (isQuotaError(e)) break; // Try next key
             }
-            return fullText;
-        } catch (e) {
-            lastError = e;
-            console.warn(`Streaming GEMMA ${modelName} failed:`, e.message);
         }
     }
-    throw lastError || new Error("Gemma streaming failed");
+    throw lastError || new Error("Gemma streaming failed on all keys");
 }
 
 async function streamContentWithSmartFallback(baseParams, res) {
@@ -471,24 +492,39 @@ router.post('/chat', checkAIAccess, async (req, res) => {
         // Note: Other providers generally don't support high-quality TTS via simple API call yet, so we stick to Gemini for TTS or fallback to browser.
         if (answerText) {
             try {
-                const client = await getGeminiClient(); // AWAIT client
-                const ttsResponse = await client.models.generateContent({
-                    model: "gemini-2.5-flash-preview-tts", 
-                    contents: [{ parts: [{ text: answerText }] }],
-                    config: {
-                        responseModalities: ['AUDIO'],
-                        speechConfig: { voiceConfig: { prebuiltVoiceConfig: { voiceName: 'Kore' } } },
-                    },
-                });
-                const audioBytes = ttsResponse.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
+                // For TTS, we also try key rotation
+                const { GoogleGenAI } = await import("@google/genai");
+                const keys = await getKeyPool('gemini');
+                let audioBytes = null;
+
+                for (const apiKey of keys) {
+                    try {
+                        const client = new GoogleGenAI({ apiKey });
+                        const ttsResponse = await client.models.generateContent({
+                            model: "gemini-2.5-flash-preview-tts", 
+                            contents: [{ parts: [{ text: answerText }] }],
+                            config: {
+                                responseModalities: ['AUDIO'],
+                                speechConfig: { voiceConfig: { prebuiltVoiceConfig: { voiceName: 'Kore' } } },
+                            },
+                        });
+                        audioBytes = ttsResponse.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
+                        if (audioBytes) break; // Success
+                    } catch(e) {
+                        console.warn(`TTS Key ${apiKey.slice(-4)} failed`, e.message);
+                        if (isQuotaError(e)) continue;
+                        break; // Non-quota error, probably model unavailable, stop trying
+                    }
+                }
+
                 if (audioBytes) {
                     res.write(`data: ${JSON.stringify({ audio: audioBytes })}\n\n`);
                 } else {
-                    throw new Error("No audio data");
+                    // Signal fallback
+                    res.write(`data: ${JSON.stringify({ ttsSkipped: true })}\n\n`);
                 }
             } catch (ttsError) {
                 console.warn("⚠️ TTS Generation skipped (Quota or Error):", ttsError.message);
-                // Send explicit signal to frontend to fallback to browser TTS
                 res.write(`data: ${JSON.stringify({ ttsSkipped: true })}\n\n`);
             }
         }
@@ -535,7 +571,6 @@ router.post('/evaluate', checkAIAccess, async (req, res) => {
         // Handle different return types from fallback
         let rawText = "";
         if (typeof response.text === 'string') rawText = response.text;
-        // Check for new SDK structure just in case, though generateContent returns GenerateContentResponse which has .text property
         else if (response && response.response && response.response.text) rawText = response.response.text();
         else if (typeof response.text === 'function') rawText = response.text(); 
         
@@ -554,20 +589,29 @@ router.post('/evaluate', checkAIAccess, async (req, res) => {
         let feedbackAudio = null;
         if (resultJson.feedback) {
             try {
-                const client = await getGeminiClient(); // AWAIT client
-                const ttsResponse = await client.models.generateContent({
-                    model: "gemini-2.5-flash-preview-tts",
-                    contents: [{ parts: [{ text: resultJson.feedback }] }],
-                    config: {
-                        responseModalities: ['AUDIO'],
-                        speechConfig: { voiceConfig: { prebuiltVoiceConfig: { voiceName: 'Kore' } } },
-                    },
-                });
-                feedbackAudio = ttsResponse.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
+                const { GoogleGenAI } = await import("@google/genai");
+                const keys = await getKeyPool('gemini');
+                
+                for (const apiKey of keys) {
+                    try {
+                        const client = new GoogleGenAI({ apiKey });
+                        const ttsResponse = await client.models.generateContent({
+                            model: "gemini-2.5-flash-preview-tts",
+                            contents: [{ parts: [{ text: resultJson.feedback }] }],
+                            config: {
+                                responseModalities: ['AUDIO'],
+                                speechConfig: { voiceConfig: { prebuiltVoiceConfig: { voiceName: 'Kore' } } },
+                            },
+                        });
+                        feedbackAudio = ttsResponse.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
+                        if (feedbackAudio) break;
+                    } catch(e) {
+                        if (isQuotaError(e)) continue;
+                        break; 
+                    }
+                }
             } catch (ttsErr) {
                 console.warn("⚠️ TTS Generation failed:", ttsErr.message);
-                // We rely on frontend to use browser TTS if audio is null, 
-                // OR we can signal it in the JSON response if we want explicit control
             }
         }
 

models.js

@@ -116,7 +116,11 @@ const ConfigSchema = new mongoose.Schema({
     emailNotify: Boolean,
     enableAI: { type: Boolean, default: true },
     aiTotalCalls: { type: Number, default: 0 },
-    periodConfig: [{ period: Number, name: String, startTime: String, endTime: String }] // NEW
+    periodConfig: [{ period: Number, name: String, startTime: String, endTime: String }],
+    apiKeys: {
+        gemini: [String],
+        openrouter: [String]
+    }
 });
 const ConfigModel = mongoose.model('Config', ConfigSchema);
 

pages/AIAssistant.tsx

@@ -2,7 +2,7 @@
 import React, { useState, useRef, useEffect } from 'react';
 import { api } from '../services/api';
 import { AIChatMessage, SystemConfig, UserRole } from '../types';
-import { Bot, Mic, Square, Play, Volume2, Send, CheckCircle, Brain, Sparkles, Loader2, StopCircle, Trash2, Image as ImageIcon, X, AlertTriangle, ShieldCheck, Activity, Power, ExternalLink } from 'lucide-react';
+import { Bot, Mic, Square, Play, Volume2, Send, CheckCircle, Brain, Sparkles, Loader2, StopCircle, Trash2, Image as ImageIcon, X, AlertTriangle, ShieldCheck, Activity, Power, ExternalLink, Key, Plus, Save } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { Toast, ToastState } from '../components/Toast';
@@ -99,6 +99,12 @@ export const AIAssistant: React.FC = () => {
     const [assessmentResult, setAssessmentResult] = useState<{score: number, feedback: string, transcription?: string, audio?: string} | null>(null);
     const [selectedImage, setSelectedImage] = useState<File | null>(null);
 
+    // Admin Key Management State
+    const [geminiKeys, setGeminiKeys] = useState<string[]>([]);
+    const [openRouterKeys, setOpenRouterKeys] = useState<string[]>([]);
+    const [newGeminiKey, setNewGeminiKey] = useState('');
+    const [newOpenRouterKey, setNewOpenRouterKey] = useState('');
+
     // Audio Refs
     const mediaRecorderRef = useRef<MediaRecorder | null>(null);
     const audioChunksRef = useRef<Blob[]>([]);
@@ -116,6 +122,10 @@ export const AIAssistant: React.FC = () => {
                 
                 const cfg = await api.config.get();
                 setSystemConfig(cfg);
+                if (isAdmin && cfg.apiKeys) {
+                    setGeminiKeys(cfg.apiKeys.gemini || []);
+                    setOpenRouterKeys(cfg.apiKeys.openrouter || []);
+                }
             } catch (e) {
                 console.error("Init failed", e);
             } finally {
@@ -407,12 +417,49 @@ export const AIAssistant: React.FC = () => {
         }
     };
 
+    const handleAddKey = (type: 'gemini' | 'openrouter') => {
+        const key = type === 'gemini' ? newGeminiKey.trim() : newOpenRouterKey.trim();
+        if (!key) return;
+        
+        if (type === 'gemini') {
+            setGeminiKeys([...geminiKeys, key]);
+            setNewGeminiKey('');
+        } else {
+            setOpenRouterKeys([...openRouterKeys, key]);
+            setNewOpenRouterKey('');
+        }
+    };
+
+    const removeKey = (type: 'gemini' | 'openrouter', index: number) => {
+        if (type === 'gemini') {
+            setGeminiKeys(geminiKeys.filter((_, i) => i !== index));
+        } else {
+            setOpenRouterKeys(openRouterKeys.filter((_, i) => i !== index));
+        }
+    };
+
+    const saveApiKeys = async () => {
+        if (!systemConfig) return;
+        try {
+            await api.config.save({
+                ...systemConfig,
+                apiKeys: {
+                    gemini: geminiKeys,
+                    openrouter: openRouterKeys
+                }
+            });
+            setToast({ show: true, message: 'API 密钥配置已保存', type: 'success' });
+        } catch (e) {
+            setToast({ show: true, message: '保存失败', type: 'error' });
+        }
+    };
+
     if (configLoading) return <div className="flex justify-center items-center h-full"><Loader2 className="animate-spin text-blue-600"/></div>;
 
     // --- ADMIN VIEW ---
     if (isAdmin) {
         return (
-            <div className="p-6 md:p-10 max-w-5xl mx-auto space-y-8 animate-in fade-in">
+            <div className="p-6 md:p-10 max-w-5xl mx-auto space-y-8 animate-in fade-in pb-20 overflow-y-auto h-full">
                 {toast.show && <Toast message={toast.message} type={toast.type} onClose={()=>setToast({...toast, show: false})}/>}
                 <div className="flex items-center gap-4 border-b pb-6">
                     <div className="p-3 bg-blue-100 rounded-2xl text-blue-600">
@@ -420,7 +467,7 @@ export const AIAssistant: React.FC = () => {
                     </div>
                     <div>
                         <h1 className="text-2xl font-bold text-gray-800">AI 智能助教管理后台</h1>
-                        <p className="text-gray-500">监控 AI 服务状态与用量</p>
+                        <p className="text-gray-500">监控 AI 服务状态与用量，管理密钥池。</p>
                     </div>
                 </div>
 
@@ -434,7 +481,7 @@ export const AIAssistant: React.FC = () => {
                             </div>
                             <div className="text-sm text-gray-500 flex-1">
                                 <p className="mb-2">计费周期: 当前学期</p>
-                                <p>服务商: Google Gemini</p>
+                                <p>服务商: Gemini & OpenRouter</p>
                             </div>
                         </div>
                     </div>
@@ -455,6 +502,74 @@ export const AIAssistant: React.FC = () => {
                         </div>
                     </div>
                 </div>
+
+                {/* API Key Management */}
+                <div className="bg-white p-6 rounded-xl border border-gray-100 shadow-sm">
+                    <div className="flex justify-between items-center mb-6">
+                        <h3 className="font-bold text-gray-800 flex items-center"><Key className="mr-2 text-amber-500"/> 多线路密钥池配置</h3>
+                        <button onClick={saveApiKeys} className="bg-blue-600 text-white px-4 py-2 rounded-lg text-sm font-bold flex items-center gap-2 hover:bg-blue-700 shadow-sm">
+                            <Save size={16}/> 保存密钥配置
+                        </button>
+                    </div>
+                    
+                    <div className="grid grid-cols-1 md:grid-cols-2 gap-8">
+                        {/* Gemini Pool */}
+                        <div>
+                            <div className="flex items-center justify-between mb-2">
+                                <label className="text-sm font-bold text-gray-700">Google Gemini / Gemma 密钥池</label>
+                                <span className="text-xs bg-blue-100 text-blue-700 px-2 py-0.5 rounded-full">{geminiKeys.length} 个</span>
+                            </div>
+                            <p className="text-xs text-gray-400 mb-3">当一个 Key 额度耗尽时，系统将自动切换至下一个。所有 Key 耗尽后尝试 OpenRouter。</p>
+                            <div className="space-y-2 mb-3">
+                                {geminiKeys.map((k, idx) => (
+                                    <div key={idx} className="flex gap-2 items-center bg-gray-50 p-2 rounded border border-gray-200">
+                                        <div className="flex-1 font-mono text-xs text-gray-600 truncate">
+                                            {k.substring(0, 8)}...{k.substring(k.length - 6)}
+                                        </div>
+                                        <button onClick={() => removeKey('gemini', idx)} className="text-gray-400 hover:text-red-500"><Trash2 size={14}/></button>
+                                    </div>
+                                ))}
+                            </div>
+                            <div className="flex gap-2">
+                                <input 
+                                    className="flex-1 border border-gray-300 rounded px-3 py-1.5 text-sm outline-none focus:ring-2 focus:ring-blue-500"
+                                    placeholder="输入 Gemini API Key"
+                                    value={newGeminiKey}
+                                    onChange={e => setNewGeminiKey(e.target.value)}
+                                />
+                                <button onClick={() => handleAddKey('gemini')} className="bg-gray-100 hover:bg-gray-200 text-gray-600 px-3 py-1.5 rounded border border-gray-300"><Plus size={16}/></button>
+                            </div>
+                        </div>
+
+                        {/* OpenRouter Pool */}
+                        <div>
+                            <div className="flex items-center justify-between mb-2">
+                                <label className="text-sm font-bold text-gray-700">OpenRouter 密钥池</label>
+                                <span className="text-xs bg-purple-100 text-purple-700 px-2 py-0.5 rounded-full">{openRouterKeys.length} 个</span>
+                            </div>
+                            <p className="text-xs text-gray-400 mb-3">作为备用线路，当 Gemini 服务不可用或额度耗尽时启用。</p>
+                            <div className="space-y-2 mb-3">
+                                {openRouterKeys.map((k, idx) => (
+                                    <div key={idx} className="flex gap-2 items-center bg-gray-50 p-2 rounded border border-gray-200">
+                                        <div className="flex-1 font-mono text-xs text-gray-600 truncate">
+                                            {k.substring(0, 8)}...{k.substring(k.length - 6)}
+                                        </div>
+                                        <button onClick={() => removeKey('openrouter', idx)} className="text-gray-400 hover:text-red-500"><Trash2 size={14}/></button>
+                                    </div>
+                                ))}
+                            </div>
+                            <div className="flex gap-2">
+                                <input 
+                                    className="flex-1 border border-gray-300 rounded px-3 py-1.5 text-sm outline-none focus:ring-2 focus:ring-purple-500"
+                                    placeholder="输入 OpenRouter API Key"
+                                    value={newOpenRouterKey}
+                                    onChange={e => setNewOpenRouterKey(e.target.value)}
+                                />
+                                <button onClick={() => handleAddKey('openrouter')} className="bg-gray-100 hover:bg-gray-200 text-gray-600 px-3 py-1.5 rounded border border-gray-300"><Plus size={16}/></button>
+                            </div>
+                        </div>
+                    </div>
+                </div>
             </div>
         );
     }

types.ts

@@ -177,7 +177,11 @@ export interface SystemConfig {
   emailNotify: boolean;
   enableAI?: boolean;
   aiTotalCalls?: number;
-  periodConfig?: PeriodConfig[]; // Kept for backward compatibility or global defaults
+  periodConfig?: PeriodConfig[]; 
+  apiKeys?: {
+      gemini?: string[];
+      openrouter?: string[];
+  };
 }
 
 export interface SchoolCalendarEntry {
@@ -368,4 +372,4 @@ export interface AIChatMessage {
   audio?: string;
   isAudioMessage?: boolean;
   timestamp: number;
-}
+}