update the csv corpus to generate / rewrite article

src/app/api/article-rewrite/route.ts

@@ -0,0 +1,44 @@
+import { openai } from '@ai-sdk/openai';
+import { generateText } from 'ai';
+import { langfuse } from '@/lib/langfuse';
+import { AI_CONFIG } from '@/config/ai';
+
+export async function POST(req: Request) {
+  try {
+    const {
+      grade,
+      unit,
+      textbookVocab,
+      additionalVocab,
+      grammar,
+      topic,
+      inputArticle,
+    } = await req.json();
+
+    const langfusePrompt = await langfuse.prompt.get('article/rewrite');
+
+    const prompt = langfusePrompt.prompt
+      .replace(/\{\{grade_values\}\}/g, Array.isArray(grade) ? grade.join(', ') : grade)
+      .replace(/\{\{unit_values\}\}/g, Array.isArray(unit) ? unit.join(', ') : unit)
+      .replace(/\{\{textbook_vocab_values\}\}/g, textbookVocab || '')
+      .replace(/\{\{additional_vocab_values\}\}/g, additionalVocab || '')
+      .replace(/\{\{grammar_values\}\}/g, Array.isArray(grammar) ? grammar.join(', ') : grammar)
+      .replace(/\{\{topic_values\}\}/g, Array.isArray(topic) ? topic.join(', ') : topic)
+      .replace(/\{\{input_article_value\}\}/g, inputArticle || '');
+
+    const result = await generateText({
+      model: openai(AI_CONFIG.model),
+      prompt,
+    });
+
+    return Response.json({
+      article: result.text,
+      createdAt: new Date().toISOString(),
+    });
+  } catch (error) {
+    console.error('Error rewriting article:', error);
+    return Response.json({ error: 'Failed to rewrite article' }, { status: 500 });
+  }
+}
+
+

src/app/api/vocab/route.ts

@@ -0,0 +1,154 @@
+import fs from 'fs';
+import path from 'path';
+
+type VocabRequest = {
+  publisher: string[];
+  grade: string[];
+  unit: string[];
+};
+
+// Map UI labels to CSV values
+function mapGradeToCsv(uiGrade: string): string {
+  const map: Record<string, string> = {
+    '七年級上學期': '七上',
+    '七年級下學期': '七下',
+    '八年級上學期': '八上',
+    '八年級下學期': '八下',
+    '九年級上學期': '九上',
+    '九年級下學期': '九下',
+  };
+  return map[uiGrade] ?? uiGrade;
+}
+
+function mapUnitToCsv(uiUnit: string): string {
+  const normalized = uiUnit.replace(/\s+/g, '');
+  const map: Record<string, string> = {
+    '第一課': 'lesson 1',
+    '第二課': 'lesson 2',
+    '第三課': 'lesson 3',
+    '第四課': 'lesson 4',
+    '第五課': 'lesson 5',
+    '第六課': 'lesson 6',
+  };
+  return map[normalized] ?? uiUnit;
+}
+
+function parseCSVLine(line: string): string[] {
+  const result: string[] = [];
+  let current = '';
+  let inQuotes = false;
+  let i = 0;
+
+  while (i < line.length) {
+    const char = line[i];
+
+    if (char === '"') {
+      if (inQuotes && line[i + 1] === '"') {
+        current += '"';
+        i += 2;
+        continue;
+      } else {
+        inQuotes = !inQuotes;
+      }
+    } else if (char === ',' && !inQuotes) {
+      result.push(current.trim());
+      current = '';
+    } else {
+      current += char;
+    }
+
+    i++;
+  }
+
+  result.push(current.trim());
+  return result;
+}
+
+function parseCSVEntry(lines: string[], startIndex: number): { values: string[]; nextIndex: number } {
+  let currentLine = startIndex;
+  let fullLine = '';
+  let openQuotes = 0;
+
+  while (currentLine < lines.length) {
+    const line = lines[currentLine];
+    fullLine += (fullLine ? '\n' : '') + line;
+
+    for (const ch of line) {
+      if (ch === '"') openQuotes++;
+    }
+
+    if (openQuotes % 2 === 0) {
+      break;
+    }
+
+    currentLine++;
+  }
+
+  const values = parseCSVLine(fullLine);
+  return { values, nextIndex: currentLine + 1 };
+}
+
+export async function POST(req: Request) {
+  try {
+    const { publisher, grade, unit } = (await req.json()) as VocabRequest;
+
+    const filePath = path.join(process.cwd(), 'public', 'middle_school_english_vocabulary_overview.csv');
+    const csvContent = fs.readFileSync(filePath, 'utf-8');
+
+    const lines = csvContent.split('\n');
+    if (lines.length === 0) {
+      return Response.json({ vocab: [], text: '' });
+    }
+
+    // Header indices
+    const header = parseCSVLine(lines[0]);
+    const idxPublisher = header.indexOf('出版社');
+    const idxGrade = header.indexOf('年級');
+    const idxUnit = header.indexOf('單元');
+    const idxWord = header.indexOf('單字');
+
+    if (idxPublisher === -1 || idxGrade === -1 || idxUnit === -1 || idxWord === -1) {
+      return Response.json({ vocab: [], text: '' });
+    }
+
+    const gradeSet = new Set(grade.map(mapGradeToCsv));
+    const unitSet = new Set(unit.map(mapUnitToCsv));
+    const publisherSet = new Set(publisher);
+
+    const vocabSet = new Set<string>();
+
+    let i = 1;
+    while (i < lines.length) {
+      if (!lines[i] || !lines[i].trim()) {
+        i++;
+        continue;
+      }
+      const entry = parseCSVEntry(lines, i);
+      const values = entry.values;
+
+      const recPublisher = values[idxPublisher];
+      const recGrade = values[idxGrade];
+      const recUnit = values[idxUnit];
+      const recWord = values[idxWord];
+
+      const matchPublisher = publisherSet.size === 0 || publisherSet.has(recPublisher);
+      const matchGrade = gradeSet.size === 0 || gradeSet.has(recGrade);
+      const matchUnit = unitSet.size === 0 || unitSet.has(recUnit);
+
+      if (matchPublisher && matchGrade && matchUnit && recWord) {
+        vocabSet.add(recWord);
+      }
+
+      i = entry.nextIndex;
+    }
+
+    const vocab = Array.from(vocabSet);
+    const text = vocab.join(', ');
+    return Response.json({ vocab, text });
+  } catch (error) {
+    console.error('Error filtering vocab:', error);
+    return Response.json({ vocab: [], text: '' }, { status: 500 });
+  }
+}
+
+

src/components/article/ArticleGenerationModal.tsx

@@ -92,13 +92,37 @@ export function ArticleGenerationModal({
 
   // Update textbook vocabulary when selections change
   useEffect(() => {
-    if (formData.grade.length > 0 && formData.unit.length > 0 && formData.publisher.length > 0) {
-      // Mock vocabulary update - in real implementation this would call an API
-      const mockVocab = "apple, book, computer, desk, elephant, friend, green, happy, important, jump";
-      setFormData(prev => ({ ...prev, textbookVocab: mockVocab }));
-    } else {
-      setFormData(prev => ({ ...prev, textbookVocab: '' }));
-    }
+    let cancelled = false;
+    const fetchVocab = async () => {
+      if (formData.grade.length > 0 && formData.unit.length > 0 && formData.publisher.length > 0) {
+        try {
+          const res = await fetch('/api/vocab', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+              publisher: formData.publisher,
+              grade: formData.grade,
+              unit: formData.unit,
+            }),
+          });
+          if (!res.ok) throw new Error('Failed to fetch vocab');
+          const data: { vocab: string[]; text: string } = await res.json();
+          if (!cancelled) {
+            setFormData(prev => ({ ...prev, textbookVocab: data.text }));
+          }
+        } catch (e) {
+          if (!cancelled) {
+            setFormData(prev => ({ ...prev, textbookVocab: '' }));
+          }
+        }
+      } else {
+        if (!cancelled) {
+          setFormData(prev => ({ ...prev, textbookVocab: '' }));
+        }
+      }
+    };
+    fetchVocab();
+    return () => { cancelled = true; };
   }, [formData.grade, formData.unit, formData.publisher]);
 
   const handleCheckboxChange = (field: keyof ArticleGenerationParams, value: string) => {