Update web/src/lib/api.ts

web/src/lib/api.ts

@@ -1,46 +1,25 @@
 // web/src/lib/api.ts
+// Aligns with api/server.py routes:
+// POST /api/login, /api/chat, /api/upload, /api/export, /api/summary, /api/feedback
+// GET  /api/memoryline
 
-export type ApiRole = "user" | "assistant";
-
-export type ApiChatMessage = {
-  role: ApiRole;
-  content: string;
-};
-
-export type ApiChatRequest = {
-  message: string;
-
-  // optional context
-  history?: Array<[string, string]>; // (user, assistant)
-  learning_mode?: string;            // e.g. "concept"
-  language_preference?: string;      // "English" | "中文" | "auto"
-  doc_type?: string;                 // "Syllabus" | ...
-  course_outline?: string[];         // parsed syllabus topics
-  rag_context?: string | null;       // if FE sends; usually BE builds
-  review_data?: string | null;       // for review mode
-};
-
-export type ApiChatResponse = {
-  answer: string;
-  history?: Array<[string, string]>;
-  references?: string[];
-};
+export type LearningMode = "general" | "concept" | "socratic" | "exam" | "assignment" | "summary";
+export type LanguagePref = "Auto" | "English" | "中文";
+export type DocType = "Syllabus" | "Lecture Slides / PPT" | "Literature Review / Paper" | "Other Course Document";
 
 const DEFAULT_TIMEOUT_MS = 20000;
 
-// If you later deploy to a different domain, set VITE_API_BASE in env
 function getBaseUrl() {
-  // Vite convention
+  // Vite env: VITE_API_BASE can be "", "http://localhost:8000", etc.
   const v = (import.meta as any)?.env?.VITE_API_BASE as string | undefined;
-  return (v && v.trim()) ? v.trim() : "";
+  return v && v.trim() ? v.trim() : "";
 }
 
 async function fetchWithTimeout(input: RequestInfo, init?: RequestInit, timeoutMs = DEFAULT_TIMEOUT_MS) {
   const controller = new AbortController();
   const id = setTimeout(() => controller.abort(), timeoutMs);
   try {
-    const res = await fetch(input, { ...init, signal: controller.signal });
-    return res;
+    return await fetch(input, { ...init, signal: controller.signal });
   } finally {
     clearTimeout(id);
   }
@@ -55,50 +34,166 @@ async function parseJsonSafe(res: Response) {
   }
 }
 
-export async function apiChat(payload: ApiChatRequest): Promise<ApiChatResponse> {
+function errMsg(data: any, fallback: string) {
+  return (data && (data.error || data.detail || data.message)) ? String(data.error || data.detail || data.message) : fallback;
+}
+
+// --------------------
+// /api/login
+// --------------------
+export type ApiLoginReq = {
+  name: string;
+  user_id: string;
+};
+
+export type ApiLoginResp =
+  | { ok: true; user: { name: string; user_id: string } }
+  | { ok: false; error: string };
+
+export async function apiLogin(payload: ApiLoginReq): Promise<ApiLoginResp> {
   const base = getBaseUrl();
-  const res = await fetchWithTimeout(`${base}/api/chat`, {
+  const res = await fetchWithTimeout(`${base}/api/login`, {
     method: "POST",
     headers: { "Content-Type": "application/json" },
     body: JSON.stringify(payload),
   });
 
   const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiLogin failed (${res.status})`));
+  return data as ApiLoginResp;
+}
 
-  if (!res.ok) {
-    const msg =
-      (data && (data.error || data.message)) ||
-      `apiChat failed (${res.status})`;
-    throw new Error(msg);
-  }
+// --------------------
+// /api/chat
+// --------------------
+export type ApiChatReq = {
+  user_id: string;
+  message: string;
+  learning_mode: string;        // backend expects string (not strict union)
+  language_preference?: string; // "Auto" | "English" | "中文"
+  doc_type?: string;            // "Syllabus" | "Lecture Slides / PPT" | ...
+};
+
+export type ApiChatRef = { source_file?: string; section?: string };
+
+export type ApiChatResp = {
+  reply: string;
+  session_status_md: string;
+  refs: ApiChatRef[];
+  latency_ms: number;
+};
+
+export async function apiChat(payload: ApiChatReq): Promise<ApiChatResp> {
+  const base = getBaseUrl();
+  const res = await fetchWithTimeout(`${base}/api/chat`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({
+      language_preference: "Auto",
+      doc_type: "Syllabus",
+      ...payload,
+    }),
+  }, 60000); // chat can be slow
 
-  // Be tolerant: map common server shapes
-  if (data?.answer) return data as ApiChatResponse;
-  if (data?.response) return { answer: data.response, history: data.history, references: data.references };
-  if (typeof data === "string") return { answer: data };
+  const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiChat failed (${res.status})`));
 
-  return { answer: "" };
+  // backend returns { reply, session_status_md, refs, latency_ms }
+  return data as ApiChatResp;
 }
 
-export async function apiUpload(file: File, docType: string) {
+// --------------------
+// /api/upload
+// --------------------
+export type ApiUploadResp = {
+  ok: boolean;
+  added_chunks?: number;
+  status_md?: string;
+  error?: string;
+};
+
+export async function apiUpload(args: { user_id: string; doc_type: string; file: File }): Promise<ApiUploadResp> {
   const base = getBaseUrl();
   const fd = new FormData();
-  fd.append("file", file);
-  fd.append("doc_type", docType);
+  fd.append("user_id", args.user_id);
+  fd.append("doc_type", args.doc_type);
+  fd.append("file", args.file);
 
-  const res = await fetchWithTimeout(`${base}/api/upload`, {
+  const res = await fetchWithTimeout(`${base}/api/upload`, { method: "POST", body: fd }, 120000);
+  const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiUpload failed (${res.status})`));
+  return data as ApiUploadResp;
+}
+
+// --------------------
+// /api/export
+// --------------------
+export async function apiExport(payload: { user_id: string; learning_mode: string }): Promise<{ markdown: string }> {
+  const base = getBaseUrl();
+  const res = await fetchWithTimeout(`${base}/api/export`, {
     method: "POST",
-    body: fd,
-  }, 60000);
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify(payload),
+  });
 
   const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiExport failed (${res.status})`));
+  return data as { markdown: string };
+}
 
-  if (!res.ok) {
-    const msg =
-      (data && (data.error || data.message)) ||
-      `apiUpload failed (${res.status})`;
-    throw new Error(msg);
-  }
+// --------------------
+// /api/summary
+// --------------------
+export async function apiSummary(payload: { user_id: string; learning_mode: string; language_preference?: string }): Promise<{ markdown: string }> {
+  const base = getBaseUrl();
+  const res = await fetchWithTimeout(`${base}/api/summary`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ language_preference: "Auto", ...payload }),
+  });
 
-  return data;
+  const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiSummary failed (${res.status})`));
+  return data as { markdown: string };
+}
+
+// --------------------
+// /api/feedback
+// --------------------
+export type ApiFeedbackReq = {
+  user_id: string;
+  rating: "helpful" | "not_helpful";
+  assistant_message_id?: string;
+  assistant_text: string;
+  user_text?: string;
+  comment?: string;
+  tags?: string[];
+  refs?: string[];
+  learning_mode?: string;
+  doc_type?: string;
+  timestamp_ms?: number;
+};
+
+export async function apiFeedback(payload: ApiFeedbackReq): Promise<{ ok: boolean }> {
+  const base = getBaseUrl();
+  const res = await fetchWithTimeout(`${base}/api/feedback`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify(payload),
+  });
+
+  const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiFeedback failed (${res.status})`));
+  return data as { ok: boolean };
+}
+
+// --------------------
+// /api/memoryline
+// --------------------
+export async function apiMemoryline(user_id: string): Promise<{ next_review_label: string; progress_pct: number }> {
+  const base = getBaseUrl();
+  const res = await fetchWithTimeout(`${base}/api/memoryline?user_id=${encodeURIComponent(user_id)}`, { method: "GET" });
+  const data = await parseJsonSafe(res);
+  if (!res.ok) throw new Error(errMsg(data, `apiMemoryline failed (${res.status})`));
+  return data as { next_review_label: string; progress_pct: number };
 }