228 lines
9.0 KiB
TypeScript
228 lines
9.0 KiB
TypeScript
"use client";
|
|
|
|
import { useEffect, useState } from "react";
|
|
import type { ChatMessage, Citation } from "@duoc-thu/shared-types";
|
|
import { ChatPanel } from "./_components/ChatPanel";
|
|
import { Sidebar, ChatSession } from "./_components/Sidebar";
|
|
import { EvidencePanel } from "./_components/EvidencePanel";
|
|
import { MessageSquare, FileSearch, Menu, X, Layers } from "lucide-react";
|
|
import { cn } from "@duoc-thu/ui";
|
|
|
|
function createSessionId() {
|
|
return `session-${crypto.randomUUID()}`;
|
|
}
|
|
|
|
export default function ChatPage() {
|
|
const [sessions, setSessions] = useState<ChatSession[]>([]);
|
|
const [currentSessionId, setCurrentSessionId] = useState<string>("");
|
|
const [messagesBySession, setMessagesBySession] = useState<Record<string, ChatMessage[]>>({});
|
|
const [queryOverride, setQueryOverride] = useState<{ text: string; token: number } | null>(null);
|
|
|
|
useEffect(() => {
|
|
const id = createSessionId();
|
|
setSessions([{ id, title: "Phiên tra cứu mới", updatedAt: new Date().toISOString() }]);
|
|
setMessagesBySession({ [id]: [] });
|
|
setCurrentSessionId(id);
|
|
}, []);
|
|
|
|
// Citation & Evidence Panel State
|
|
const [citations, setCitations] = useState<Citation[]>([]);
|
|
const [activeCitationIndex, setActiveCitationIndex] = useState<number | null>(null);
|
|
|
|
// Responsive Drawer Toggles for Mobile/Tablet
|
|
const [showMobileSidebar, setShowMobileSidebar] = useState(false);
|
|
const [showMobileEvidence, setShowMobileEvidence] = useState(false);
|
|
const [showEvidenceDesktop, setShowEvidenceDesktop] = useState(true);
|
|
|
|
const handleNewChat = () => {
|
|
const newId = createSessionId();
|
|
const newSession: ChatSession = {
|
|
id: newId,
|
|
title: "Phiên tra cứu mới",
|
|
updatedAt: new Date().toISOString(),
|
|
};
|
|
setSessions((prev) => [newSession, ...prev]);
|
|
setMessagesBySession((prev) => ({ ...prev, [newId]: [] }));
|
|
setCurrentSessionId(newId);
|
|
setQueryOverride(null);
|
|
setCitations([]);
|
|
setActiveCitationIndex(null);
|
|
setShowMobileSidebar(false);
|
|
};
|
|
|
|
const handleSelectSession = (id: string) => {
|
|
setCurrentSessionId(id);
|
|
setQueryOverride(null);
|
|
setCitations([]);
|
|
setActiveCitationIndex(null);
|
|
setShowMobileSidebar(false);
|
|
};
|
|
|
|
const handleDeleteSession = (id: string) => {
|
|
setSessions((prev) => prev.filter((s) => s.id !== id));
|
|
setMessagesBySession((prev) => {
|
|
const next = { ...prev };
|
|
delete next[id];
|
|
return next;
|
|
});
|
|
if (currentSessionId === id) {
|
|
const remaining = sessions.filter((s) => s.id !== id);
|
|
if (remaining.length > 0) {
|
|
setCurrentSessionId(remaining[0].id);
|
|
} else {
|
|
handleNewChat();
|
|
}
|
|
}
|
|
};
|
|
|
|
const handleQuickQuery = (query: string) => {
|
|
setQueryOverride({ text: query, token: Date.now() });
|
|
setShowMobileSidebar(false);
|
|
};
|
|
|
|
const currentMessages = messagesBySession[currentSessionId] ?? [];
|
|
const setCurrentMessages: React.Dispatch<React.SetStateAction<ChatMessage[]>> = (update) => {
|
|
const sessionId = currentSessionId;
|
|
setMessagesBySession((prev) => {
|
|
const existing = prev[sessionId] ?? [];
|
|
const next = typeof update === "function" ? update(existing) : update;
|
|
return { ...prev, [sessionId]: next };
|
|
});
|
|
};
|
|
|
|
if (!currentSessionId) {
|
|
return <div className="flex flex-1 items-center justify-center text-sm text-txt-muted">Đang tạo phiên tra cứu an toàn...</div>;
|
|
}
|
|
|
|
const handleCitationClick = (citation: Citation, index: number, allCitations: Citation[]) => {
|
|
// Found live 2026-08-07: this used to only set the index into whatever
|
|
// `citations` array was last loaded (i.e. the MOST RECENT answer's), so
|
|
// clicking [1] on an older message showed a LATER message's unrelated
|
|
// drug in the evidence panel (reported live: clicking Omeprazol's own
|
|
// citation showed Kanamycin). The clicked message's own citation list
|
|
// must replace the panel's state, not just the index into a stale one.
|
|
setCitations(allCitations);
|
|
setActiveCitationIndex(index);
|
|
setShowMobileEvidence(true);
|
|
};
|
|
|
|
const handleCitationsLoaded = (newCitations: Citation[]) => {
|
|
setCitations(newCitations);
|
|
// Deliberately NOT auto-activating citation 1 here (removed
|
|
// 2026-08-07): this used to fire the beam connector line + card
|
|
// highlight on every single answer, unprompted, and — since
|
|
// `CitationBeamOverlay` only recomputes its coordinates on window
|
|
// resize/scroll, not on the content reflow a just-arrived answer
|
|
// itself causes — the line frequently ended up pointing at stale
|
|
// positions, i.e. exactly the "dây trích dẫn dính lung tung" (messy
|
|
// citation wire) reported live. The beam/highlight now only appears
|
|
// when the user actually clicks a citation, at which point the
|
|
// coordinates are computed fresh.
|
|
};
|
|
|
|
return (
|
|
<div className="flex flex-1 w-full h-[calc(100vh-6.5rem)] overflow-hidden bg-app relative">
|
|
{/* Mobile Header Bar Controls */}
|
|
<div className="lg:hidden absolute top-2 left-3 right-3 z-30 flex items-center justify-between pointer-events-none">
|
|
<button
|
|
onClick={() => setShowMobileSidebar(true)}
|
|
aria-label="Mở danh mục phiên"
|
|
className="pointer-events-auto flex items-center gap-1.5 px-3 py-1.5 rounded-full border border-border-subtle bg-surface/90 text-txt-primary text-xs font-semibold shadow-elevated backdrop-blur-md"
|
|
>
|
|
<Menu className="w-4 h-4 text-accent-primary" />
|
|
<span>Lịch sử</span>
|
|
</button>
|
|
|
|
<button
|
|
onClick={() => setShowMobileEvidence(!showMobileEvidence)}
|
|
aria-label="Mở thanh bằng chứng"
|
|
className="pointer-events-auto flex items-center gap-1.5 px-3 py-1.5 rounded-full border border-border-subtle bg-surface/90 text-txt-primary text-xs font-semibold shadow-elevated backdrop-blur-md"
|
|
>
|
|
<FileSearch className="w-4 h-4 text-accent-primary" />
|
|
<span>Bằng chứng ({citations.length})</span>
|
|
</button>
|
|
</div>
|
|
|
|
{/* Desktop Region 1: Navigation / Session Sidebar */}
|
|
<Sidebar
|
|
currentSessionId={currentSessionId}
|
|
sessions={sessions}
|
|
onSelectSession={handleSelectSession}
|
|
onNewChat={handleNewChat}
|
|
onDeleteSession={handleDeleteSession}
|
|
onQuickQuery={handleQuickQuery}
|
|
className="hidden lg:flex"
|
|
/>
|
|
|
|
{/* Mobile Drawer Region 1 */}
|
|
{showMobileSidebar && (
|
|
<div className="fixed inset-0 z-50 flex lg:hidden">
|
|
<div
|
|
className="fixed inset-0 bg-bg-overlay backdrop-blur-sm"
|
|
onClick={() => setShowMobileSidebar(false)}
|
|
/>
|
|
<div className="relative flex flex-col w-4/5 max-w-sm h-full bg-surface z-10 shadow-elevated animate-slide-up">
|
|
<Sidebar
|
|
currentSessionId={currentSessionId}
|
|
sessions={sessions}
|
|
onSelectSession={handleSelectSession}
|
|
onNewChat={handleNewChat}
|
|
onDeleteSession={handleDeleteSession}
|
|
onQuickQuery={handleQuickQuery}
|
|
className="w-full h-full border-r-0"
|
|
/>
|
|
</div>
|
|
</div>
|
|
)}
|
|
|
|
{/* Desktop Region 2: Primary Answer Workspace */}
|
|
<main className="flex-1 flex justify-center overflow-hidden relative">
|
|
<ChatPanel
|
|
key={currentSessionId}
|
|
sessionId={currentSessionId}
|
|
messages={currentMessages}
|
|
onMessagesChange={setCurrentMessages}
|
|
initialQuery={queryOverride?.text}
|
|
initialQueryToken={queryOverride?.token}
|
|
onCitationClick={handleCitationClick}
|
|
onCitationsLoaded={handleCitationsLoaded}
|
|
activeCitationIndex={activeCitationIndex}
|
|
className="w-full max-w-4xl h-full"
|
|
/>
|
|
</main>
|
|
|
|
{/* Desktop Region 3: Right Evidence & Source Inspector Panel */}
|
|
{showEvidenceDesktop && (
|
|
<EvidencePanel
|
|
citations={citations}
|
|
activeCitationIndex={activeCitationIndex}
|
|
onSelectCitation={(citation, index) => setActiveCitationIndex(index)}
|
|
onClose={() => setShowEvidenceDesktop(false)}
|
|
className="hidden lg:flex"
|
|
/>
|
|
)}
|
|
|
|
{/* Mobile Sheet Region 3: Evidence Bottom Sheet / Drawer */}
|
|
{showMobileEvidence && (
|
|
<div className="fixed inset-0 z-50 flex flex-col justify-end lg:hidden">
|
|
<div
|
|
className="fixed inset-0 bg-bg-overlay backdrop-blur-sm"
|
|
onClick={() => setShowMobileEvidence(false)}
|
|
/>
|
|
<div className="relative flex flex-col w-full h-4/5 bg-surface rounded-t-3xl z-10 shadow-elevated overflow-hidden animate-slide-up">
|
|
<EvidencePanel
|
|
citations={citations}
|
|
activeCitationIndex={activeCitationIndex}
|
|
onSelectCitation={(citation, index) => {
|
|
setActiveCitationIndex(index);
|
|
}}
|
|
onClose={() => setShowMobileEvidence(false)}
|
|
className="w-full h-full border-l-0"
|
|
/>
|
|
</div>
|
|
</div>
|
|
)}
|
|
</div>
|
|
);
|
|
}
|