Update components/ai/CommentGeneratorPanel.tsx

components/ai/CommentGeneratorPanel.tsx

@@ -1,6 +1,7 @@
 
 import React, { useState, useEffect, useRef } from 'react';
 import { User, StudentProfile, SavedRecord, Zodiac, Personality, Hobby } from '../../types';
+import { api } from '../../services/api';
 import { FileSpreadsheet, Loader2, Sparkles, Copy, Save, Trash2, Download, RefreshCw, PenTool, Edit, Box, ChevronLeft, Settings2, Check, X } from 'lucide-react';
 import { Toast, ToastState } from '../Toast';
 import { ConfirmModal } from '../ConfirmModal';
@@ -67,6 +68,30 @@ export const CommentGeneratorPanel: React.FC<CommentGeneratorPanelProps> = ({ cu
         }
     }, [savedRecords, currentUser]);
 
+    // Auto-load class students
+    useEffect(() => {
+        const fetchClassStudents = async () => {
+            // If list is empty and user has a class, try to load from API
+            if (importedNames.length === 0 && currentUser?.homeroomClass) {
+                try {
+                    const allStudents = await api.students.getAll();
+                    const classStudents = allStudents
+                        .filter((s: any) => s.className === currentUser.homeroomClass)
+                        .map((s: any) => s.name);
+                    
+                    if (classStudents.length > 0) {
+                        setImportedNames(classStudents);
+                        setProfile(prev => ({ ...prev, name: classStudents[0] }));
+                        setToast({ show: true, message: `已加载 ${currentUser.homeroomClass} 学生名单`, type: 'success' });
+                    }
+                } catch (e) {
+                    console.error("Failed to load students", e);
+                }
+            }
+        };
+        fetchClassStudents();
+    }, [currentUser]);
+
     // --- Excel Import Logic ---
     const handleFileUpload = async (e: React.ChangeEvent<HTMLInputElement>) => {
         const file = e.target.files?.[0];
@@ -396,39 +421,39 @@ export const CommentGeneratorPanel: React.FC<CommentGeneratorPanelProps> = ({ cu
                         {/* 1. Header (Name Input) */}
                         <div className="p-4 border-b border-gray-100 bg-gray-50/50 shrink-0">
                             <label className="text-xs font-bold text-gray-500 uppercase block mb-2">当前学生</label>
-                            {importedNames.length > 0 ? (
-                                <div className="relative">
-                                    <select 
-                                        className="w-full border border-amber-300 rounded-xl p-3 bg-white text-base font-bold text-gray-800 shadow-sm focus:ring-2 focus:ring-amber-400 outline-none appearance-none"
-                                        value={profile.name}
-                                        onChange={e => setProfile({...profile, name: e.target.value})}
-                                    >
-                                        {importedNames.map(name => (
-                                            <option key={name} value={name}>
-                                                {isStudentDone(name) ? '✅' : '⬜'} {name}
-                                            </option>
-                                        ))}
-                                    </select>
-                                    <div className="absolute right-3 top-3.5 text-gray-400 pointer-events-none text-xs">▼</div>
-                                </div>
-                            ) : (
-                                <div className="flex gap-2">
+                            <div className="flex gap-2">
+                                {importedNames.length > 0 ? (
+                                    <div className="relative flex-1">
+                                        <select 
+                                            className="w-full border border-amber-300 rounded-xl p-3 bg-white text-base font-bold text-gray-800 shadow-sm focus:ring-2 focus:ring-amber-400 outline-none appearance-none"
+                                            value={profile.name}
+                                            onChange={e => setProfile({...profile, name: e.target.value})}
+                                        >
+                                            {importedNames.map(name => (
+                                                <option key={name} value={name}>
+                                                    {isStudentDone(name) ? '✅' : '⬜'} {name}
+                                                </option>
+                                            ))}
+                                        </select>
+                                        <div className="absolute right-3 top-3.5 text-gray-400 pointer-events-none text-xs">▼</div>
+                                    </div>
+                                ) : (
                                     <input 
                                         className="flex-1 border border-gray-300 rounded-xl p-3 text-sm focus:ring-2 focus:ring-amber-400 outline-none shadow-sm"
                                         placeholder="输入姓名..."
                                         value={profile.name}
                                         onChange={e => setProfile({...profile, name: e.target.value})}
                                     />
-                                    <button 
-                                        onClick={() => fileInputRef.current?.click()} 
-                                        className="px-4 bg-green-50 text-green-600 rounded-xl border border-green-200 hover:bg-green-100 flex items-center shadow-sm"
-                                        title="Excel 导入"
-                                    >
-                                        <FileSpreadsheet size={20}/>
-                                    </button>
-                                    <input type="file" accept=".xlsx,.xls,.csv" ref={fileInputRef} className="hidden" onChange={handleFileUpload} />
-                                </div>
-                            )}
+                                )}
+                                <button 
+                                    onClick={() => fileInputRef.current?.click()} 
+                                    className="px-3 bg-white text-green-600 rounded-xl border border-green-200 hover:bg-green-50 flex items-center justify-center shadow-sm shrink-0"
+                                    title="Excel 导入 / 覆盖名单"
+                                >
+                                    <FileSpreadsheet size={20}/>
+                                </button>
+                                <input type="file" accept=".xlsx,.xls,.csv" ref={fileInputRef} className="hidden" onChange={handleFileUpload} />
+                            </div>
                             {isImporting && <div className="text-xs text-gray-400 mt-2 flex items-center gap-1"><Loader2 size={10} className="animate-spin"/> 正在解析名单...</div>}
                         </div>