feat: add artifact panel
This commit is contained in:
@@ -0,0 +1,202 @@
|
|||||||
|
import { useState, useEffect } from "react";
|
||||||
|
import { Code2, Eye, X, Download, Copy, ExternalLink, Check, ChevronDown } from "lucide-react";
|
||||||
|
import { useTranslation } from "react-i18next";
|
||||||
|
import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter';
|
||||||
|
import { vscDarkPlus } from 'react-syntax-highlighter/dist/esm/styles/prism';
|
||||||
|
import { cn } from "@/lib/utils";
|
||||||
|
|
||||||
|
interface ArtifactPreviewTarget {
|
||||||
|
name: string;
|
||||||
|
mimeType: string;
|
||||||
|
previewUrl: string;
|
||||||
|
downloadUrl: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ArtifactPanelProps {
|
||||||
|
artifact: ArtifactPreviewTarget;
|
||||||
|
onClose: () => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function ArtifactPanel({ artifact, onClose }: ArtifactPanelProps) {
|
||||||
|
const { t } = useTranslation();
|
||||||
|
const [activeTab, setActiveTab] = useState<'code' | 'preview'>('preview');
|
||||||
|
const [code, setCode] = useState<string>('');
|
||||||
|
const [loadingCode, setLoadingCode] = useState(false);
|
||||||
|
const [copied, setCopied] = useState(false);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
// Reset state when artifact changes
|
||||||
|
setCode('');
|
||||||
|
if (artifact.mimeType.startsWith("image/") || artifact.mimeType.startsWith("application/pdf")) {
|
||||||
|
setActiveTab('preview');
|
||||||
|
} else {
|
||||||
|
setActiveTab('preview');
|
||||||
|
}
|
||||||
|
}, [artifact]);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (activeTab === 'code' && !code && artifact.downloadUrl) {
|
||||||
|
setLoadingCode(true);
|
||||||
|
fetch(artifact.downloadUrl)
|
||||||
|
.then(res => res.text())
|
||||||
|
.then(text => {
|
||||||
|
setCode(text);
|
||||||
|
setLoadingCode(false);
|
||||||
|
})
|
||||||
|
.catch(err => {
|
||||||
|
console.error("Failed to fetch code", err);
|
||||||
|
setCode("Failed to load code.");
|
||||||
|
setLoadingCode(false);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}, [activeTab, artifact.downloadUrl, code]);
|
||||||
|
|
||||||
|
const handleCopy = async () => {
|
||||||
|
if (code) {
|
||||||
|
try {
|
||||||
|
await navigator.clipboard.writeText(code);
|
||||||
|
setCopied(true);
|
||||||
|
setTimeout(() => setCopied(false), 2000);
|
||||||
|
} catch (err) {
|
||||||
|
console.error('Failed to copy text: ', err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const isCodeViewSupported = !artifact.mimeType.startsWith("image/") && !artifact.mimeType.startsWith("application/pdf");
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="h-full flex flex-col bg-background border-l border-border shadow-2xl">
|
||||||
|
{/* Header */}
|
||||||
|
<div className="flex items-center justify-between px-4 py-3 border-b border-border bg-background z-10">
|
||||||
|
<div className="flex items-center gap-2 min-w-0">
|
||||||
|
<span className="font-semibold text-sm truncate">{artifact.name}</span>
|
||||||
|
<ChevronDown className="h-4 w-4 text-muted-foreground shrink-0" />
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{isCodeViewSupported && (
|
||||||
|
<div className="flex items-center bg-muted/50 rounded-lg p-0.5 ml-4">
|
||||||
|
<button
|
||||||
|
onClick={() => setActiveTab('code')}
|
||||||
|
className={cn(
|
||||||
|
"flex items-center gap-1.5 px-3 py-1.5 rounded-md text-xs font-medium transition-all",
|
||||||
|
activeTab === 'code'
|
||||||
|
? "bg-background text-foreground shadow-sm"
|
||||||
|
: "text-muted-foreground hover:text-foreground hover:bg-background/50"
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
<Code2 className="h-4 w-4" />
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
onClick={() => setActiveTab('preview')}
|
||||||
|
className={cn(
|
||||||
|
"flex items-center gap-1.5 px-3 py-1.5 rounded-md text-xs font-medium transition-all",
|
||||||
|
activeTab === 'preview'
|
||||||
|
? "bg-background text-foreground shadow-sm"
|
||||||
|
: "text-muted-foreground hover:text-foreground hover:bg-background/50"
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
<Eye className="h-4 w-4" />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<div className="flex items-center gap-1 ml-auto">
|
||||||
|
{activeTab === 'code' && (
|
||||||
|
<button
|
||||||
|
onClick={handleCopy}
|
||||||
|
className="p-2 text-muted-foreground hover:text-foreground hover:bg-muted rounded-md transition-colors"
|
||||||
|
title={t('copy', 'Copy')}
|
||||||
|
>
|
||||||
|
{copied ? <Check className="h-4 w-4 text-emerald-500" /> : <Copy className="h-4 w-4" />}
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
<a
|
||||||
|
href={artifact.previewUrl}
|
||||||
|
target="_blank"
|
||||||
|
rel="noopener noreferrer"
|
||||||
|
className="p-2 text-muted-foreground hover:text-foreground hover:bg-muted rounded-md transition-colors"
|
||||||
|
title={t('openInNewTab', 'Open in new tab')}
|
||||||
|
>
|
||||||
|
<ExternalLink className="h-4 w-4" />
|
||||||
|
</a>
|
||||||
|
<a
|
||||||
|
href={artifact.downloadUrl}
|
||||||
|
download={artifact.name}
|
||||||
|
className="p-2 text-muted-foreground hover:text-foreground hover:bg-muted rounded-md transition-colors"
|
||||||
|
title={t('download', 'Download')}
|
||||||
|
>
|
||||||
|
<Download className="h-4 w-4" />
|
||||||
|
</a>
|
||||||
|
<div className="w-px h-4 bg-border mx-1" />
|
||||||
|
<button
|
||||||
|
onClick={onClose}
|
||||||
|
className="p-2 text-muted-foreground hover:text-foreground hover:bg-muted rounded-md transition-colors"
|
||||||
|
title={t('close', 'Close')}
|
||||||
|
>
|
||||||
|
<X className="h-4 w-4" />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Content */}
|
||||||
|
<div className="flex-1 min-h-0 relative bg-zinc-950">
|
||||||
|
{activeTab === 'preview' ? (
|
||||||
|
<div className="w-full h-full bg-white">
|
||||||
|
{artifact.mimeType.startsWith("image/") ? (
|
||||||
|
<img
|
||||||
|
src={artifact.previewUrl}
|
||||||
|
alt={artifact.name}
|
||||||
|
className="w-full h-full object-contain bg-muted/50"
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<iframe
|
||||||
|
title={artifact.name}
|
||||||
|
src={artifact.previewUrl}
|
||||||
|
className="w-full h-full border-0"
|
||||||
|
sandbox="allow-same-origin allow-scripts"
|
||||||
|
onLoad={(e) => {
|
||||||
|
try {
|
||||||
|
const doc = (e.target as HTMLIFrameElement).contentDocument;
|
||||||
|
if (doc) {
|
||||||
|
const style = doc.createElement('style');
|
||||||
|
style.textContent = `html, body { overflow: auto !important; }`;
|
||||||
|
doc.head.appendChild(style);
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error("Failed to inject styles into iframe", err);
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="w-full h-full overflow-auto text-[13px]">
|
||||||
|
{loadingCode ? (
|
||||||
|
<div className="h-full flex items-center justify-center text-muted-foreground">
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<div className="h-4 w-4 rounded-full border-2 border-primary border-t-transparent animate-spin" />
|
||||||
|
Loading code...
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<SyntaxHighlighter
|
||||||
|
language={artifact.name.split('.').pop() || 'text'}
|
||||||
|
style={vscDarkPlus}
|
||||||
|
customStyle={{
|
||||||
|
margin: 0,
|
||||||
|
padding: '1rem',
|
||||||
|
background: 'transparent',
|
||||||
|
minHeight: '100%'
|
||||||
|
}}
|
||||||
|
showLineNumbers
|
||||||
|
>
|
||||||
|
{code}
|
||||||
|
</SyntaxHighlighter>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
import { useState, useRef, useEffect } from "react";
|
import { useState, useRef, useEffect } from "react";
|
||||||
import { ScrollArea } from "@/components/ui/scroll-area";
|
import { ScrollArea } from "@/components/ui/scroll-area";
|
||||||
import { User, Loader2, ArrowUp, ChevronDown, Check, Square, Plus, Database, Wand2, CheckCircle2, Table, XCircle, Settings, ExternalLink, FileText, Download, Eye, Copy, Mic, X } from "lucide-react";
|
import { User, Loader2, ArrowUp, ChevronDown, Check, Square, Plus, Database, Wand2, CheckCircle2, Table, XCircle, Settings, ExternalLink, Download, Copy, Mic, X, Compass } from "lucide-react";
|
||||||
import { api } from "@/lib/api";
|
import { api } from "@/lib/api";
|
||||||
import { type ChartSpec } from "@/store/visualizationStore";
|
import { type ChartSpec } from "@/store/visualizationStore";
|
||||||
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
|
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
|
||||||
@@ -13,7 +13,7 @@ import { useTranslation } from "react-i18next";
|
|||||||
import { InlineVisualizationCard } from "./InlineVisualizationCard";
|
import { InlineVisualizationCard } from "./InlineVisualizationCard";
|
||||||
import { useProjectStore } from "@/store/projectStore";
|
import { useProjectStore } from "@/store/projectStore";
|
||||||
import { SlashCommandMenu } from "./SlashCommandMenu";
|
import { SlashCommandMenu } from "./SlashCommandMenu";
|
||||||
import { Dialog, DialogContent, DialogHeader, DialogTitle } from "@/components/ui/dialog";
|
import { ArtifactPanel } from "./ArtifactPanel";
|
||||||
|
|
||||||
interface Message {
|
interface Message {
|
||||||
id: string;
|
id: string;
|
||||||
@@ -63,6 +63,7 @@ interface ArtifactPreviewTarget {
|
|||||||
name: string;
|
name: string;
|
||||||
mimeType: string;
|
mimeType: string;
|
||||||
previewUrl: string;
|
previewUrl: string;
|
||||||
|
downloadUrl: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
const REPORT_HTML_BLOCK_REGEX = /<!--\s*REPORT_HTML_START\s*-->([\s\S]*?)<!--\s*REPORT_HTML_END\s*-->/i;
|
const REPORT_HTML_BLOCK_REGEX = /<!--\s*REPORT_HTML_START\s*-->([\s\S]*?)<!--\s*REPORT_HTML_END\s*-->/i;
|
||||||
@@ -143,19 +144,6 @@ interface SessionData {
|
|||||||
}>;
|
}>;
|
||||||
}
|
}
|
||||||
|
|
||||||
const formatArtifactSize = (size: number): string => {
|
|
||||||
if (!Number.isFinite(size) || size < 0) return "0 B";
|
|
||||||
const units = ["B", "KB", "MB", "GB", "TB"];
|
|
||||||
let value = size;
|
|
||||||
let unitIndex = 0;
|
|
||||||
while (value >= 1024 && unitIndex < units.length - 1) {
|
|
||||||
value /= 1024;
|
|
||||||
unitIndex += 1;
|
|
||||||
}
|
|
||||||
const fixed = value >= 10 || unitIndex === 0 ? 0 : 1;
|
|
||||||
return `${value.toFixed(fixed)} ${units[unitIndex]}`;
|
|
||||||
};
|
|
||||||
|
|
||||||
const normalizeArtifacts = (raw: unknown): MessageArtifact[] => {
|
const normalizeArtifacts = (raw: unknown): MessageArtifact[] => {
|
||||||
if (!Array.isArray(raw)) return [];
|
if (!Array.isArray(raw)) return [];
|
||||||
return raw.reduce<MessageArtifact[]>((acc, item) => {
|
return raw.reduce<MessageArtifact[]>((acc, item) => {
|
||||||
@@ -1357,7 +1345,8 @@ export function ChatInterface() {
|
|||||||
};
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="flex flex-col h-full bg-background relative">
|
<div className="flex h-full bg-background relative overflow-hidden">
|
||||||
|
<div className={cn("flex flex-col h-full transition-all duration-300", artifactPreview ? "w-full md:w-1/2 border-r border-border hidden md:flex" : "w-full")}>
|
||||||
<ScrollArea className="flex-1 min-h-0">
|
<ScrollArea className="flex-1 min-h-0">
|
||||||
{/* Hidden file input available in all states */}
|
{/* Hidden file input available in all states */}
|
||||||
<input
|
<input
|
||||||
@@ -1558,34 +1547,39 @@ export function ChatInterface() {
|
|||||||
{msg.artifacts && msg.artifacts.length > 0 ? (
|
{msg.artifacts && msg.artifacts.length > 0 ? (
|
||||||
<div className="mt-4 grid gap-2 sm:grid-cols-2">
|
<div className="mt-4 grid gap-2 sm:grid-cols-2">
|
||||||
{msg.artifacts.map((artifact, artifactIndex) => (
|
{msg.artifacts.map((artifact, artifactIndex) => (
|
||||||
<div key={`${msg.id}-artifact-${artifactIndex}`} className="rounded-xl border border-border bg-muted/50/60 px-3 py-2.5">
|
<div
|
||||||
<div className="flex items-center gap-2.5">
|
key={`${msg.id}-artifact-${artifactIndex}`}
|
||||||
<div className="h-8 w-8 rounded-lg bg-background border border-border flex items-center justify-center text-muted-foreground shrink-0">
|
className="rounded-xl border border-border bg-background px-4 py-3 cursor-pointer hover:bg-muted/50 transition-colors flex items-center justify-between gap-4"
|
||||||
<FileText className="h-4 w-4" />
|
onClick={() => {
|
||||||
|
if (artifact.previewable && artifact.preview_url) {
|
||||||
|
setArtifactPreview({
|
||||||
|
name: artifact.name,
|
||||||
|
mimeType: artifact.mime_type,
|
||||||
|
previewUrl: artifact.preview_url,
|
||||||
|
downloadUrl: artifact.download_url
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<div className="flex items-center gap-3 min-w-0 flex-1">
|
||||||
|
<div className="h-10 w-10 rounded-full bg-background border border-border flex items-center justify-center text-foreground shrink-0">
|
||||||
|
<Compass className="h-5 w-5" />
|
||||||
</div>
|
</div>
|
||||||
<div className="min-w-0 flex-1">
|
<div className="min-w-0 flex-1">
|
||||||
<div className="text-sm font-medium text-foreground/90 truncate">{artifact.name}</div>
|
<div className="text-sm font-bold text-foreground truncate">{artifact.name}</div>
|
||||||
<div className="text-[11px] text-muted-foreground">{formatArtifactSize(artifact.size)}</div>
|
<div className="text-xs text-muted-foreground truncate">{artifact.mime_type === "text/html" ? "HTML file" : artifact.mime_type}</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div className="mt-2 flex items-center gap-2">
|
<div className="flex items-center gap-2 flex-shrink-0">
|
||||||
{artifact.previewable && artifact.preview_url ? (
|
|
||||||
<button
|
|
||||||
onClick={() => setArtifactPreview({ name: artifact.name, mimeType: artifact.mime_type, previewUrl: artifact.preview_url || "" })}
|
|
||||||
className="inline-flex items-center gap-1.5 text-xs px-2.5 py-1.5 rounded-md border border-border text-foreground/80 hover:bg-background transition-colors"
|
|
||||||
>
|
|
||||||
<Eye className="h-3.5 w-3.5" />
|
|
||||||
{t('preview')}
|
|
||||||
</button>
|
|
||||||
) : null}
|
|
||||||
<a
|
<a
|
||||||
href={artifact.download_url}
|
href={artifact.download_url}
|
||||||
target="_blank"
|
target="_blank"
|
||||||
rel="noopener noreferrer"
|
rel="noopener noreferrer"
|
||||||
className="inline-flex items-center gap-1.5 text-xs px-2.5 py-1.5 rounded-md border border-border text-foreground/80 hover:bg-background transition-colors"
|
className="inline-flex items-center gap-1.5 text-sm px-3 py-1.5 rounded-md text-foreground/80 hover:bg-muted transition-colors whitespace-nowrap"
|
||||||
|
onClick={(e) => e.stopPropagation()}
|
||||||
>
|
>
|
||||||
<Download className="h-3.5 w-3.5" />
|
<Download className="h-4 w-4" />
|
||||||
{t('download')}
|
<span>{t('download')}</span>
|
||||||
</a>
|
</a>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -1644,42 +1638,17 @@ export function ChatInterface() {
|
|||||||
})}
|
})}
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
<Dialog open={Boolean(artifactPreview)} onOpenChange={(open) => {
|
|
||||||
if (!open) setArtifactPreview(null);
|
|
||||||
}}>
|
|
||||||
<DialogContent className="sm:max-w-[min(1100px,95vw)] h-[85vh] flex flex-col">
|
|
||||||
<DialogHeader>
|
|
||||||
<DialogTitle>{artifactPreview?.name || t('artifactPreview')}</DialogTitle>
|
|
||||||
</DialogHeader>
|
|
||||||
<div className="flex-1 min-h-0 rounded-lg border border-border bg-background overflow-hidden">
|
|
||||||
{artifactPreview?.mimeType.startsWith("image/") ? (
|
|
||||||
<img
|
|
||||||
src={artifactPreview.previewUrl}
|
|
||||||
alt={artifactPreview.name}
|
|
||||||
className="w-full h-full object-contain bg-muted/50"
|
|
||||||
/>
|
|
||||||
) : artifactPreview ? (
|
|
||||||
<iframe
|
|
||||||
title={artifactPreview.name}
|
|
||||||
src={artifactPreview.previewUrl}
|
|
||||||
className="w-full h-full border-0"
|
|
||||||
onLoad={(e) => {
|
|
||||||
try {
|
|
||||||
const doc = (e.target as HTMLIFrameElement).contentDocument;
|
|
||||||
if (doc) {
|
|
||||||
const style = doc.createElement('style');
|
|
||||||
style.textContent = `html, body { overflow: auto !important; }`;
|
|
||||||
doc.head.appendChild(style);
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
console.error("Failed to inject styles into iframe", err);
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
) : null}
|
|
||||||
</div>
|
</div>
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
{/* Artifact Side Panel */}
|
||||||
|
{artifactPreview && (
|
||||||
|
<div className="w-full md:w-1/2 h-full absolute md:relative inset-0 md:inset-auto z-50">
|
||||||
|
<ArtifactPanel
|
||||||
|
artifact={artifactPreview}
|
||||||
|
onClose={() => setArtifactPreview(null)}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user