Upload 53 files

ai-routes.js

@@ -2,7 +2,7 @@
 const express = require('express');
 const router = express.Router();
 const OpenAI = require('openai');
-const { ConfigModel, User } = require('./models');
+const { ConfigModel, User, AIUsageModel } = require('./models');
 
 // --- Key Management & Rotation ---
 
@@ -29,6 +29,23 @@ async function getKeyPool(type) {
     return pool;
 }
 
+// --- Usage Tracking Helper ---
+async function recordUsage(model, provider) {
+    try {
+        const today = new Date().toISOString().split('T')[0];
+        // Upsert the usage record for this specific day/model/provider combination
+        await AIUsageModel.findOneAndUpdate(
+            { date: today, model: model, provider: provider },
+            { $inc: { count: 1 } },
+            { upsert: true, new: true }
+        );
+        // Also increment total stats for backward compatibility
+        await ConfigModel.findOneAndUpdate({ key: 'main' }, { $inc: { aiTotalCalls: 1 } }, { upsert: true });
+    } catch (e) {
+        console.error("Failed to record AI usage stats:", e);
+    }
+}
+
 // --- Helpers ---
 
 const wait = (ms) => new Promise(resolve => setTimeout(resolve, ms));
@@ -147,7 +164,12 @@ async function callGeminiProvider(baseParams) {
             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);
+                const result = await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
+                
+                // Track Usage on Success
+                recordUsage(modelName, PROVIDERS.GEMINI);
+                
+                return result;
             } catch (e) {
                 lastError = e;
                 console.error(`⚠️ [AI Debug] Gemini ${modelName} Error with key ${apiKey.slice(-4)}:`, e.message);
@@ -201,14 +223,16 @@ async function callOpenRouterProvider(baseParams) {
                 if (!completion || !completion.choices || !completion.choices[0]) {
                     throw new Error(`Invalid response structure from ${modelName}`);
                 }
+                
+                // Track Usage on Success
+                recordUsage(modelName, PROVIDERS.OPENROUTER);
+
                 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.
             }
         }
     }
@@ -236,7 +260,12 @@ async function callGemmaProvider(baseParams) {
                     model: modelName,
                     config: gemmaConfig 
                 };
-                return await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
+                const result = await callAIWithRetry(() => client.models.generateContent(currentParams), 1);
+                
+                // Track Usage on Success
+                recordUsage(modelName, PROVIDERS.GEMMA);
+                
+                return result;
             } catch (e) {
                 lastError = e;
                 console.warn(`⚠️ [AI Debug] Backup Model ${modelName} failed with key ${apiKey.slice(-4)}.`, e.message);
@@ -297,6 +326,9 @@ async function streamGemini(baseParams, res) {
                 const currentParams = { ...baseParams, model: modelName };
                 const result = await client.models.generateContentStream(currentParams);
                 
+                // Track Usage on Success (Stream Start)
+                recordUsage(modelName, PROVIDERS.GEMINI);
+
                 let fullText = "";
                 for await (const chunk of result) {
                     const chunkText = chunk.text;
@@ -352,6 +384,10 @@ async function streamOpenRouter(baseParams, res) {
                     messages: messages,
                     stream: true
                 });
+                
+                // Track Usage on Success (Stream Start)
+                recordUsage(modelName, PROVIDERS.OPENROUTER);
+
                 let fullText = '';
                 for await (const chunk of stream) {
                     const text = chunk.choices[0]?.delta?.content || '';
@@ -393,6 +429,10 @@ async function streamGemma(baseParams, res) {
                     config: gemmaConfig 
                 };
                 const result = await client.models.generateContentStream(currentParams);
+                
+                // Track Usage on Success (Stream Start)
+                recordUsage(modelName, PROVIDERS.GEMMA);
+
                 let fullText = "";
                 for await (const chunk of result) {
                     const chunkText = chunk.text;
@@ -476,6 +516,63 @@ const checkAIAccess = async (req, res, next) => {
 
 // --- Routes ---
 
+// GET /api/ai/stats (New Detailed Stats Endpoint)
+router.get('/stats', checkAIAccess, async (req, res) => {
+    try {
+        const config = await ConfigModel.findOne({ key: 'main' });
+        const totalCalls = config ? (config.aiTotalCalls || 0) : 0;
+        
+        // Get Today's Date
+        const todayStr = new Date().toISOString().split('T')[0];
+        
+        // Get dates for the last 7 days
+        const last7Days = [];
+        for (let i = 6; i >= 0; i--) {
+            const d = new Date();
+            d.setDate(d.getDate() - i);
+            last7Days.push(d.toISOString().split('T')[0]);
+        }
+
+        // Aggregate usage data from AIUsageModel
+        const allUsage = await AIUsageModel.find({});
+        
+        // 1. Calculate Today's Calls
+        let todayCount = 0;
+        allUsage.forEach(u => {
+            if (u.date === todayStr) todayCount += u.count;
+        });
+
+        // 2. Calculate Weekly Trend (Date -> Count)
+        const dailyTrend = last7Days.map(date => {
+            const count = allUsage.filter(u => u.date === date).reduce((sum, u) => sum + u.count, 0);
+            return { date, count };
+        });
+
+        // 3. Calculate Model Distribution (Model Name -> Total Count)
+        const modelStats = {};
+        allUsage.forEach(u => {
+            if (!modelStats[u.model]) modelStats[u.model] = 0;
+            modelStats[u.model] += u.count;
+        });
+        
+        const modelDistribution = Object.keys(modelStats).map(key => ({
+            name: key,
+            value: modelStats[key]
+        })).sort((a,b) => b.value - a.value);
+
+        res.json({
+            totalCalls,
+            todayCount,
+            dailyTrend,
+            modelDistribution
+        });
+
+    } catch (e) {
+        console.error("Stats Error:", e);
+        res.status(500).json({ error: e.message });
+    }
+});
+
 // POST /api/ai/reset-pool (New Endpoint)
 router.post('/reset-pool', checkAIAccess, (req, res) => {
     resetProviderOrder();
@@ -557,7 +654,6 @@ router.post('/chat', checkAIAccess, async (req, res) => {
             }
         }
 
-        await ConfigModel.findOneAndUpdate({ key: 'main' }, { $inc: { aiTotalCalls: 1 } }, { upsert: true });
         res.write('data: [DONE]\n\n');
         res.end();
 
@@ -643,8 +739,6 @@ router.post('/evaluate', checkAIAccess, async (req, res) => {
             }
         }
 
-        await ConfigModel.findOneAndUpdate({ key: 'main' }, { $inc: { aiTotalCalls: 1 } }, { upsert: true });
-
         res.json({
             ...resultJson,
             audio: feedbackAudio,

models.js

@@ -1,8 +1,7 @@
 
 const mongoose = require('mongoose');
 
-// ... (Previous Models)
-
+// ... (Previous Models: School, User, Student, Course, Score, Class, Subject, Exam Models - No Change)
 const SchoolSchema = new mongoose.Schema({ name: String, code: String });
 const School = mongoose.model('School', SchoolSchema);
 
@@ -110,7 +109,7 @@ const ConfigSchema = new mongoose.Schema({
     semesters: [String], 
     allowRegister: Boolean, 
     allowAdminRegister: Boolean, 
-    allowPrincipalRegister: Boolean,
+    allowPrincipalRegister: Boolean, 
     allowStudentRegister: Boolean, 
     maintenanceMode: Boolean, 
     emailNotify: Boolean,
@@ -241,7 +240,6 @@ const FeedbackSchema = new mongoose.Schema({
 });
 const FeedbackModel = mongoose.model('Feedback', FeedbackSchema);
 
-// NEW: Todo Model
 const TodoSchema = new mongoose.Schema({
     userId: String,
     content: String,
@@ -250,9 +248,20 @@ const TodoSchema = new mongoose.Schema({
 });
 const TodoModel = mongoose.model('Todo', TodoSchema);
 
+// NEW: Detailed AI Usage Stats
+const AIUsageSchema = new mongoose.Schema({
+    date: String, // Format: YYYY-MM-DD
+    model: String,
+    provider: String,
+    count: { type: Number, default: 0 }
+});
+// Optimize lookups by date and model
+AIUsageSchema.index({ date: 1, model: 1, provider: 1 }, { unique: true });
+const AIUsageModel = mongoose.model('AIUsage', AIUsageSchema);
+
 module.exports = {
     School, User, Student, Course, Score, ClassModel, SubjectModel, ExamModel, ScheduleModel,
     ConfigModel, NotificationModel, GameSessionModel, StudentRewardModel, LuckyDrawConfigModel, GameMonsterConfigModel, GameZenConfigModel,
     AchievementConfigModel, TeacherExchangeConfigModel, StudentAchievementModel, AttendanceModel, LeaveRequestModel, SchoolCalendarModel,
-    WishModel, FeedbackModel, TodoModel
+    WishModel, FeedbackModel, TodoModel, AIUsageModel
 };

pages/AIAssistant.tsx

@@ -2,10 +2,11 @@
 import React, { useState, useRef, useEffect } from 'react';
 import { api } from '../services/api';
 import { AIChatMessage, SystemConfig, UserRole, OpenRouterModelConfig } from '../types';
-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, ArrowUp, ArrowDown } 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, ArrowUp, ArrowDown, BarChart2 } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { Toast, ToastState } from '../components/Toast';
+import { BarChart, Bar, XAxis, YAxis, CartesianGrid, Tooltip, ResponsiveContainer, PieChart, Pie, Cell, Legend } from 'recharts';
 
 // Utility to handle Base64 conversion
 const blobToBase64 = (blob: Blob): Promise<string> => {
@@ -92,6 +93,9 @@ const DEFAULT_OR_MODELS = [
     { id: 'tngtech/deepseek-r1t-chimera:free', name: 'DeepSeek R1T', isCustom: false }
 ];
 
+// Colors for Pie Chart
+const PIE_COLORS = ['#3b82f6', '#10b981', '#f59e0b', '#ef4444', '#8b5cf6', '#ec4899', '#06b6d4'];
+
 export const AIAssistant: React.FC = () => {
     const currentUser = api.auth.getCurrentUser();
     const isAdmin = currentUser?.role === UserRole.ADMIN;
@@ -100,6 +104,14 @@ export const AIAssistant: React.FC = () => {
     const [systemConfig, setSystemConfig] = useState<SystemConfig | null>(null);
     const [configLoading, setConfigLoading] = useState(true);
     const [toast, setToast] = useState<ToastState>({ show: false, message: '', type: 'success' });
+    
+    // Admin Stats State
+    const [detailedStats, setDetailedStats] = useState<{
+        totalCalls: number;
+        todayCount: number;
+        dailyTrend: {date: string, count: number}[];
+        modelDistribution: {name: string, value: number}[];
+    } | null>(null);
 
     // Chat State with Persistence
     const [activeTab, setActiveTab] = useState<'chat' | 'assessment'>('chat');
@@ -160,12 +172,16 @@ 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 || []);
-                }
                 if (isAdmin) {
+                    if (cfg.apiKeys) {
+                        setGeminiKeys(cfg.apiKeys.gemini || []);
+                        setOpenRouterKeys(cfg.apiKeys.openrouter || []);
+                    }
                     setOrModels(cfg.openRouterModels && cfg.openRouterModels.length > 0 ? cfg.openRouterModels : DEFAULT_OR_MODELS);
+                    
+                    // Fetch Detailed Stats
+                    const stats = await api.ai.getStats();
+                    setDetailedStats(stats);
                 }
             } catch (e) {
                 console.error("Init failed", e);
@@ -557,16 +573,62 @@ export const AIAssistant: React.FC = () => {
                 </div>
 
                 <div className="grid grid-cols-1 md:grid-cols-2 gap-6">
-                    <div className="bg-white p-6 rounded-xl border border-gray-100 shadow-sm">
-                        <h3 className="font-bold text-gray-800 mb-4 flex items-center"><Activity className="mr-2 text-purple-500"/> 调用统计</h3>
-                        <div className="flex items-center gap-4">
-                            <div className="bg-purple-50 p-4 rounded-xl text-center flex-1">
-                                <div className="text-3xl font-black text-purple-600">{systemConfig?.aiTotalCalls || 0}</div>
-                                <div className="text-xs text-purple-700 font-bold uppercase mt-1">总调用次数</div>
+                    {/* STATS PANEL */}
+                    <div className="bg-white p-6 rounded-xl border border-gray-100 shadow-sm md:col-span-2">
+                        <h3 className="font-bold text-gray-800 mb-6 flex items-center"><Activity className="mr-2 text-purple-500"/> 调用数据分析</h3>
+                        
+                        <div className="grid grid-cols-2 md:grid-cols-4 gap-4 mb-8">
+                            <div className="bg-purple-50 p-4 rounded-xl text-center">
+                                <div className="text-3xl font-black text-purple-600">{detailedStats?.totalCalls || 0}</div>
+                                <div className="text-xs text-purple-700 font-bold uppercase mt-1">历史累计调用</div>
                             </div>
-                            <div className="text-sm text-gray-500 flex-1">
-                                <p className="mb-2">计费周期: 当前学期</p>
-                                <p>服务商: Gemini & OpenRouter</p>
+                            <div className="bg-blue-50 p-4 rounded-xl text-center">
+                                <div className="text-3xl font-black text-blue-600">{detailedStats?.todayCount || 0}</div>
+                                <div className="text-xs text-blue-700 font-bold uppercase mt-1">今日调用次数</div>
+                            </div>
+                            <div className="md:col-span-2 bg-gray-50 p-4 rounded-xl flex items-center justify-center text-gray-500 text-sm">
+                                <span className="mr-2 font-bold">当前状态:</span> 
+                                {systemConfig?.enableAI ? <span className="text-green-600 font-bold flex items-center"><CheckCircle size={14} className="mr-1"/> 服务运行中</span> : <span className="text-red-500 font-bold flex items-center"><X size={14} className="mr-1"/> 服务已暂停</span>}
+                            </div>
+                        </div>
+
+                        <div className="grid grid-cols-1 md:grid-cols-2 gap-8">
+                            {/* Weekly Trend Chart */}
+                            <div className="h-64">
+                                <h4 className="text-sm font-bold text-gray-600 mb-4 flex items-center"><BarChart2 size={16} className="mr-2"/> 近7日调用趋势</h4>
+                                <ResponsiveContainer width="100%" height="100%">
+                                    <BarChart data={detailedStats?.dailyTrend || []}>
+                                        <CartesianGrid strokeDasharray="3 3" vertical={false}/>
+                                        <XAxis dataKey="date" tick={{fontSize: 10}} tickFormatter={(val) => val.slice(5)}/>
+                                        <YAxis allowDecimals={false}/>
+                                        <Tooltip cursor={{fill: 'transparent'}} contentStyle={{borderRadius: '8px', border: 'none', boxShadow: '0 4px 12px rgba(0,0,0,0.1)'}}/>
+                                        <Bar dataKey="count" fill="#8b5cf6" radius={[4, 4, 0, 0]} name="调用次数" />
+                                    </BarChart>
+                                </ResponsiveContainer>
+                            </div>
+
+                            {/* Model Distribution Pie */}
+                            <div className="h-64">
+                                <h4 className="text-sm font-bold text-gray-600 mb-4 flex items-center"><Brain size={16} className="mr-2"/> 模型使用分布</h4>
+                                <ResponsiveContainer width="100%" height="100%">
+                                    <PieChart>
+                                        <Pie
+                                            data={detailedStats?.modelDistribution || []}
+                                            cx="50%"
+                                            cy="50%"
+                                            innerRadius={60}
+                                            outerRadius={80}
+                                            paddingAngle={5}
+                                            dataKey="value"
+                                        >
+                                            {(detailedStats?.modelDistribution || []).map((entry, index) => (
+                                                <Cell key={`cell-${index}`} fill={PIE_COLORS[index % PIE_COLORS.length]} />
+                                            ))}
+                                        </Pie>
+                                        <Tooltip />
+                                        <Legend layout="vertical" verticalAlign="middle" align="right" wrapperStyle={{fontSize: '10px'}}/>
+                                    </PieChart>
+                                </ResponsiveContainer>
                             </div>
                         </div>
                     </div>
@@ -707,7 +769,8 @@ export const AIAssistant: React.FC = () => {
         );
     }
 
-    // --- MAINTENANCE MODE VIEW (For Teachers) ---
+    // ... (Rest of component remains same)
+    // --- MAINTENANCE MODE VIEW ---
     if (!systemConfig?.enableAI) {
         return (
             <div className="flex flex-col items-center justify-center h-full text-center p-6 space-y-4">
@@ -720,7 +783,7 @@ export const AIAssistant: React.FC = () => {
         );
     }
 
-    // --- TEACHER VIEW (Chat UI) ---
+    // ... (Teacher View - Chat UI - Same as before)
     return (
         <div className="h-full flex flex-col bg-slate-50 overflow-hidden relative">
             {toast.show && <Toast message={toast.message} type={toast.type} onClose={()=>setToast({...toast, show: false})}/>}

services/api.ts

@@ -271,7 +271,8 @@ export const api = {
   ai: {
       chat: (data: { text?: string, audio?: string, history?: { role: string, text?: string }[] }) => request('/ai/chat', { method: 'POST', body: JSON.stringify(data) }),
       evaluate: (data: { question: string, audio?: string, image?: string }) => request('/ai/evaluate', { method: 'POST', body: JSON.stringify(data) }),
-      resetPool: () => request('/ai/reset-pool', { method: 'POST' }), // NEW
+      resetPool: () => request('/ai/reset-pool', { method: 'POST' }), 
+      getStats: () => request('/ai/stats'), // NEW Detailed Stats
   },
 
   todos: { // NEW