Files
sucker.severian.dev/src/app/page.tsx

774 lines
32 KiB
TypeScript

"use client";
import { useState, useEffect } from "react";
import { Separator } from "@/components/ui/separator";
import { Button } from "@/components/ui/button";
import { Card, CardContent } from "@/components/ui/card";
import {
Accordion,
AccordionContent,
AccordionItem,
AccordionTrigger,
} from "@/components/ui/accordion";
import {
Dialog,
DialogContent,
DialogHeader,
DialogTitle,
DialogDescription,
} from "@/components/ui/dialog";
import { Input } from "@/components/ui/input";
import { Png } from "@/lib/png";
import { ChevronUp, ChevronDown, Copy } from "lucide-react";
import {
CollapsibleContent,
Collapsible,
CollapsibleTrigger,
} from "@/components/ui/collapsible";
interface Card {
id: string;
name: string;
first_mes: string;
description: string;
personality: string;
mes_example: string;
scenario: string;
avatarUrl?: string;
hasVersions?: boolean;
versionCount?: number;
messageCount?: number;
initialVersion?: {
name: string;
first_mes: string;
description: string;
personality: string;
mes_example: string;
scenario: string;
};
}
export default function Home() {
const [isInstructionsOpen, setIsInstructionsOpen] = useState(false);
const [cards, setCards] = useState<Card[]>([]);
const [dialogOpen, setDialogOpen] = useState(false);
const [selectedCardIndex, setSelectedCardIndex] = useState<number | null>(
null
);
const [characterUrl, setCharacterUrl] = useState("");
const [avatarPath, setAvatarPath] = useState("");
const [isMetadataOpen, setIsMetadataOpen] = useState(false);
const [isRefreshing, setIsRefreshing] = useState(false);
const [changesDialogOpen, setChangesDialogOpen] = useState(false);
const [selectedChanges, setSelectedChanges] = useState<any>(null);
const [showFullText, setShowFullText] = useState(false);
const fetchCards = async () => {
try {
setIsRefreshing(true);
const response = await fetch("/api/proxy");
const data = await response.json();
if (data.cards) {
setCards((prevCards) => {
return data.cards.map((newCard: Card) => ({
...newCard,
avatarUrl:
prevCards.find((c) => c.id === newCard.id)?.avatarUrl ||
newCard.avatarUrl,
}));
});
}
} catch (error) {
console.error("Error fetching cards:", error);
} finally {
setIsRefreshing(false);
}
};
useEffect(() => {
fetchCards();
}, []);
const downloadJson = (card: Card) => {
// Use initial version for download, or current version if no initial version available
const downloadData = card.initialVersion ? {
name: card.initialVersion.name,
first_mes: card.initialVersion.first_mes,
description: card.initialVersion.description,
personality: card.initialVersion.personality,
mes_example: card.initialVersion.mes_example,
scenario: card.initialVersion.scenario,
} : {
name: card.name,
first_mes: card.first_mes,
description: card.description,
personality: card.personality,
mes_example: card.mes_example,
scenario: card.scenario,
};
const element = document.createElement("a");
const file = new Blob([JSON.stringify(downloadData, null, 2)], {
type: "application/json",
});
element.href = URL.createObjectURL(file);
element.download = `${card.name.replace(/[^a-zA-Z0-9\-_]/g, '_')}.json`;
document.body.appendChild(element);
element.click();
document.body.removeChild(element);
};
const downloadChanges = async (card: Card) => {
try {
const response = await fetch(`/api/proxy?changes=true&cardId=${card.id}`);
if (!response.ok) {
throw new Error('Failed to fetch changes');
}
const changesData = await response.json();
const element = document.createElement("a");
const file = new Blob([JSON.stringify(changesData, null, 2)], {
type: "application/json",
});
element.href = URL.createObjectURL(file);
element.download = `${card.name.replace(/[^a-zA-Z0-9\-_]/g, '_')}_changes.json`;
document.body.appendChild(element);
element.click();
document.body.removeChild(element);
} catch (error) {
console.error("Error downloading changes:", error);
alert("Failed to download changes. The card may not have version history.");
}
};
const viewChanges = async (card: Card) => {
try {
const response = await fetch(`/api/proxy?changes=true&cardId=${card.id}`);
if (!response.ok) {
throw new Error('Failed to fetch changes');
}
const changesData = await response.json();
setSelectedChanges(changesData);
setShowFullText(false); // Reset to diff view by default
setChangesDialogOpen(true);
} catch (error) {
console.error("Error fetching changes:", error);
alert("Failed to fetch changes. The card may not have version history.");
}
};
const downloadPng = async (card: Card, cardId: string) => {
if (!card.avatarUrl) return;
try {
const img = new Image();
img.src = `/api/proxy/image?url=${encodeURIComponent(card.avatarUrl)}`;
await new Promise((resolve, reject) => {
img.onload = resolve;
img.onerror = reject;
});
const canvas = document.createElement("canvas");
canvas.width = img.width;
canvas.height = img.height;
const ctx = canvas.getContext("2d");
if (!ctx) throw new Error("Could not get canvas context");
ctx.drawImage(img, 0, 0);
const pngBlob = await new Promise<Blob>((resolve) => {
canvas.toBlob((blob) => {
if (blob) resolve(blob);
else throw new Error("Could not convert to PNG");
}, "image/png");
});
const arrayBuffer = await pngBlob.arrayBuffer();
// Use initial version for PNG embedding, or current version if no initial version available
const pngData = card.initialVersion ? {
name: card.initialVersion.name,
first_mes: card.initialVersion.first_mes,
description: card.initialVersion.description,
personality: card.initialVersion.personality,
mes_example: card.initialVersion.mes_example,
scenario: card.initialVersion.scenario,
} : {
name: card.name,
first_mes: card.first_mes,
description: card.description,
personality: card.personality,
mes_example: card.mes_example,
scenario: card.scenario,
};
const cardData = JSON.stringify(pngData);
const newImageData = Png.Generate(arrayBuffer, cardData);
const newFileName = `${
card.name.replace(/[^a-zA-Z0-9\-_]/g, '_') || "character"
}.png`;
const newFile = new File([newImageData], newFileName, {
type: "image/png",
});
const link = URL.createObjectURL(newFile);
const a = document.createElement("a");
a.download = newFileName;
a.href = link;
a.click();
URL.revokeObjectURL(link);
} catch (error) {
console.error("Error generating PNG:", error);
alert("Couldn't export this character card, sorry.");
}
};
const handleOpenDialog = (index: number) => {
setSelectedCardIndex(index);
setDialogOpen(true);
setCharacterUrl("");
setAvatarPath("");
setIsMetadataOpen(false);
};
const handleOpenMetadata = () => {
const match = characterUrl.match(/characters\/([\w-]+)/);
if (match && match[1]) {
const characterId = match[1].split("_")[0];
window.open(
`https://janitorai.com/hampter/characters/${characterId}`,
"_blank"
);
setIsMetadataOpen(true);
}
};
const handleFetchAvatar = async () => {
if (selectedCardIndex === null) return;
try {
const avatarUrl = `https://ella.janitorai.com/bot-avatars/${avatarPath}`;
const updatedCards = [...cards];
updatedCards[selectedCardIndex] = {
...updatedCards[selectedCardIndex],
avatarUrl,
};
setCards(updatedCards);
setDialogOpen(false);
} catch (error) {
console.error("Error fetching avatar:", error);
}
};
const copyToClipboard = (text: string) => {
navigator.clipboard.writeText(text);
};
return (
<main className="min-h-screen bg-background text-foreground">
<div className="container mx-auto px-4 py-8">
<div className="flex justify-between items-center mb-4">
<div>
<h1 className="text-3xl font-bold">Sucker v2.0</h1>
<p className="text-sm text-muted-foreground">
Now with multimessage support! Tracks changes to character descriptions and scenarios across multiple extractions.
</p>
</div>
<Button
onClick={fetchCards}
variant="outline"
disabled={isRefreshing}
>
{isRefreshing ? "Refreshing..." : "Refresh"}
</Button>
</div>
<Separator className="my-4" />
<div className="mb-8">
<div className="bg-blue-50 dark:bg-blue-950 border border-blue-200 dark:border-blue-800 rounded-lg p-4">
<div className="flex flex-col justify-between">
<span className="text-lg font-semibold text-blue-800 dark:text-blue-200">
New: Multimessage Support
</span>
<p className="text-sm text-muted-foreground">
Sucker now tracks changes to character descriptions and scenarios across multiple messages. Cards with multiple versions show a version badge and offer a "Download Changes" button to get the change history.
</p>
</div>
</div>
</div>
<Collapsible
open={isInstructionsOpen}
onOpenChange={setIsInstructionsOpen}
className="w-full mb-8"
>
<div className="flex items-center justify-between">
<h2 className="text-2xl font-semibold">How to Use</h2>
<CollapsibleTrigger asChild>
<Button variant="ghost" size="sm" className="w-9 p-0">
{isInstructionsOpen ? (
<ChevronUp className="h-4 w-4" />
) : (
<ChevronDown className="h-4 w-4" />
)}
<span className="sr-only">Toggle instructions</span>
</Button>
</CollapsibleTrigger>
</div>
<CollapsibleContent className="mt-4">
<div className="prose dark:prose-invert max-w-none">
<p className="mb-2">
Follow every instruction here to the letter because it's all you
need to know and I have no intent of helping you further.
</p>
<ol className="list-decimal list-inside">
<li className="mb-2">
Put <code>https://sucker.severian.dev/api/proxy</code> in your
API settings, any value for model and key.
</li>
<li className="mb-2">
REQUIRED: Set your custom prompt to <code>&lt;.&gt;</code>
</li>
<li className="mb-2">
REQUIRED: Set your persona (or create a new one) with the name <code>&#123;user&#125;</code> and the description should only have <code>.</code> in it.
</li>
<li className="mb-2">
Save settings and refresh the page. Not this page. <i>That</i>{" "}
page.
</li>
<li className="mb-2">
Start a new chat with a character or multiple.
</li>
<li className="mb-2">
You can either send a dot to let sucker make a best guess about the char name, or send the char name yourself and it'll be used instead.
</li>
<li className="mb-2">
Hit the Refresh button here, and the cards should appear here.
</li>
<li className="mb-2">
Download the JSON files or go through a little more effort to
get PNGs instead.
</li>
</ol>
<p className="mb-2">
Extractions will only last for 10 minutes, after which they're
discarded. Reloading the page will remove any attached avatars.
I'm not storing shit.
</p>
<p className="mb-2">
<strong>New:</strong> If you send multiple messages with the same character name,
Sucker will track changes to the description and scenario fields. Cards with
multiple versions will show a version badge and offer a "Download Changes"
button to get a detailed change history with timestamps.
</p>
</div>
</CollapsibleContent>
</Collapsible>
<div className="max-w-3xl mx-auto space-y-6">
{cards.length === 0 ? (
<Card>
<CardContent className="pt-6 text-center text-muted-foreground">
<p>No extractions yet.</p>
</CardContent>
</Card>
) : (
cards.map((card, index) => (
<Card key={index}>
<CardContent className="pt-6">
<Accordion type="single" collapsible className="w-full">
<AccordionItem value={`card-${index}`}>
<AccordionTrigger className="text-xl font-semibold">
<div className="flex items-center gap-2">
{card.name || "Unnamed Card"}
<div className="flex gap-1">
{card.hasVersions && (
<span className="text-sm bg-blue-100 dark:bg-blue-900 text-blue-800 dark:text-blue-200 px-2 py-1 rounded-full">
v{card.versionCount}
</span>
)}
{card.messageCount && card.messageCount > 1 && (
<span className="text-sm bg-green-100 dark:bg-green-900 text-green-800 dark:text-green-200 px-2 py-1 rounded-full">
{card.messageCount} msgs
</span>
)}
</div>
</div>
</AccordionTrigger>
<AccordionContent>
<div id={`card-${index}`} className="space-y-4 mt-4">
{(card.initialVersion?.description || card.description) && (
<Accordion type="single" collapsible>
<AccordionItem value="description">
<AccordionTrigger>Description</AccordionTrigger>
<AccordionContent>
<div className="flex justify-between">
<pre className="whitespace-pre-wrap font-sans text-sm">{card.initialVersion?.description || card.description}</pre>
<Button
variant="ghost"
size="icon"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(card.initialVersion?.description || card.description);
}}
>
<Copy className="h-4 w-4" />
</Button>
</div>
</AccordionContent>
</AccordionItem>
</Accordion>
)}
{(card.initialVersion?.first_mes || card.first_mes) && (
<Accordion type="single" collapsible>
<AccordionItem value="first-message">
<AccordionTrigger>
First Message
</AccordionTrigger>
<AccordionContent>
<div className="flex justify-between">
<pre className="whitespace-pre-wrap font-sans text-sm">{card.initialVersion?.first_mes || card.first_mes}</pre>
<Button
variant="ghost"
size="icon"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(card.initialVersion?.first_mes || card.first_mes);
}}
>
<Copy className="h-4 w-4" />
</Button>
</div>
</AccordionContent>
</AccordionItem>
</Accordion>
)}
{(card.initialVersion?.scenario || card.scenario) && (
<Accordion type="single" collapsible>
<AccordionItem value="scenario">
<AccordionTrigger>Scenario</AccordionTrigger>
<AccordionContent>
<div className="flex justify-between">
<pre className="whitespace-pre-wrap font-sans text-sm">{card.initialVersion?.scenario || card.scenario}</pre>
<Button
variant="ghost"
size="icon"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(card.initialVersion?.scenario || card.scenario);
}}
>
<Copy className="h-4 w-4" />
</Button>
</div>
</AccordionContent>
</AccordionItem>
</Accordion>
)}
{(card.initialVersion?.mes_example || card.mes_example) && (
<Accordion type="single" collapsible>
<AccordionItem value="example-messages">
<AccordionTrigger>
Example Messages
</AccordionTrigger>
<AccordionContent>
<div className="flex justify-between">
<pre className="whitespace-pre-wrap font-sans text-sm">{card.initialVersion?.mes_example || card.mes_example}</pre>
<Button
variant="ghost"
size="icon"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(card.initialVersion?.mes_example || card.mes_example);
}}
>
<Copy className="h-4 w-4" />
</Button>
</div>
</AccordionContent>
</AccordionItem>
</Accordion>
)}
{(card.initialVersion?.personality || card.personality) && (
<Accordion type="single" collapsible>
<AccordionItem value="personality">
<AccordionTrigger>Personality</AccordionTrigger>
<AccordionContent>
<div className="flex justify-between">
<pre className="whitespace-pre-wrap font-sans text-sm">{card.initialVersion?.personality || card.personality}</pre>
<Button
variant="ghost"
size="icon"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(card.initialVersion?.personality || card.personality);
}}
>
<Copy className="h-4 w-4" />
</Button>
</div>
</AccordionContent>
</AccordionItem>
</Accordion>
)}
</div>
</AccordionContent>
</AccordionItem>
</Accordion>
<div className="flex flex-row space-x-2 mt-6">
<Button
onClick={() => downloadJson(card)}
variant="default"
>
Download JSON
</Button>
{card.hasVersions && (
<>
<Button
onClick={() => viewChanges(card)}
variant="outline"
>
View Changes
</Button>
<Button
onClick={() => downloadChanges(card)}
variant="outline"
>
Download Changes
</Button>
</>
)}
{!card.avatarUrl ? (
<Button
onClick={() => handleOpenDialog(index)}
variant="outline"
>
Fetch Avatar (required for PNG)
</Button>
) : (
<Button
onClick={() => downloadPng(card, `card-${index}`)}
variant="default"
disabled={!card.avatarUrl}
>
Download PNG
</Button>
)}
</div>
</CardContent>
</Card>
))
)}
</div>
</div>
<Dialog open={dialogOpen} onOpenChange={setDialogOpen}>
<DialogContent>
<DialogHeader>
<DialogTitle>
{isMetadataOpen ? "Enter Avatar Path" : "Enter Character URL"}
</DialogTitle>
<DialogDescription>
{isMetadataOpen
? "Look for the avatar field in the opened tab and paste the value here."
: "Enter the Janitor character URL (https://janitorai.com/characters/...)."}
</DialogDescription>
</DialogHeader>
{isMetadataOpen ? (
<div className="space-y-4">
<Input
placeholder="id.webp"
value={avatarPath}
onChange={(e: React.ChangeEvent<HTMLInputElement>) =>
setAvatarPath(e.target.value)
}
/>
<Button onClick={handleFetchAvatar} className="w-full">
Fetch Avatar
</Button>
</div>
) : (
<div className="space-y-4">
<Input
placeholder="https://janitorai.com/characters/..."
value={characterUrl}
onChange={(e: React.ChangeEvent<HTMLInputElement>) =>
setCharacterUrl(e.target.value)
}
/>
<p className="text-sm text-muted-foreground">
Upon clicking this button, a new tab will open with the
character's metadata. Look for the avatar field and copy the
value before returning to this page.
</p>
<Button onClick={handleOpenMetadata} className="w-full">
Open Metadata
</Button>
</div>
)}
</DialogContent>
</Dialog>
<Dialog open={changesDialogOpen} onOpenChange={setChangesDialogOpen}>
<DialogContent className="max-w-4xl max-h-[80vh] overflow-y-auto">
<DialogHeader>
<DialogTitle>
Change History: {selectedChanges?.cardName}
</DialogTitle>
<DialogDescription className="flex items-center justify-between">
<span>Version history showing changes to description and scenario fields</span>
<Button
variant="outline"
size="sm"
onClick={() => setShowFullText(!showFullText)}
>
{showFullText ? 'Show Changes Only' : 'Show Full Text'}
</Button>
</DialogDescription>
</DialogHeader>
{selectedChanges && (
<div className="space-y-4">
<div className="grid grid-cols-2 gap-4 text-sm">
<div>
<strong>Total Versions:</strong> {selectedChanges.totalVersions}
</div>
<div>
<strong>Current Version:</strong> {selectedChanges.currentVersion}
</div>
<div>
<strong>Description Changes:</strong> {selectedChanges.summary.descriptionChanges}
</div>
<div>
<strong>Scenario Changes:</strong> {selectedChanges.summary.scenarioChanges}
</div>
</div>
<Separator />
<div className="space-y-4">
<h3 className="text-lg font-semibold">Version History</h3>
{selectedChanges.versions.map((version: any, index: number) => (
<div key={version.version} className="border rounded-lg p-4">
<div className="flex justify-between items-center mb-2">
<h4 className="font-semibold">
Version {version.version} ({version.changeType})
</h4>
<div className="text-sm text-muted-foreground">
{new Date(version.timestamp).toLocaleString()}
{version.messageCount && ` • Message ${version.messageCount}`}
</div>
</div>
{version.changes.description && (
<div className="mb-3">
<h5 className="font-medium text-sm mb-1">Description Change:</h5>
{version.changeType === 'initial' ? (
<div className="bg-blue-50 dark:bg-blue-950 p-2 rounded text-sm">
<strong>Initial Content:</strong> {version.changes.description.new}
</div>
) : (
<div className="space-y-2">
{version.addedText?.description && (
<div className="bg-green-50 dark:bg-green-950 p-2 rounded text-sm">
<div className="flex justify-between items-start">
<div>
<strong>Added:</strong> {version.addedText.description}
</div>
<Button
variant="ghost"
size="icon"
className="ml-2 h-6 w-6"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(version.addedText.description);
}}
>
<Copy className="h-3 w-3" />
</Button>
</div>
</div>
)}
{version.removedText?.description && (
<div className="bg-red-50 dark:bg-red-950 p-2 rounded text-sm">
<strong>Removed:</strong> {version.removedText.description}
</div>
)}
{showFullText && (
<div className="space-y-1 mt-2 pt-2 border-t">
<div className="bg-gray-50 dark:bg-gray-950 p-2 rounded text-xs">
<strong>Full Old:</strong> {version.changes.description.old}
</div>
<div className="bg-gray-50 dark:bg-gray-950 p-2 rounded text-xs">
<strong>Full New:</strong> {version.changes.description.new}
</div>
</div>
)}
</div>
)}
</div>
)}
{version.changes.scenario && (
<div>
<h5 className="font-medium text-sm mb-1">Scenario Change:</h5>
{version.changeType === 'initial' ? (
<div className="bg-blue-50 dark:bg-blue-950 p-2 rounded text-sm">
<strong>Initial Content:</strong> {version.changes.scenario.new}
</div>
) : (
<div className="space-y-2">
{version.addedText?.scenario && (
<div className="bg-green-50 dark:bg-green-950 p-2 rounded text-sm">
<div className="flex justify-between items-start">
<div>
<strong>Added:</strong> {version.addedText.scenario}
</div>
<Button
variant="ghost"
size="icon"
className="ml-2 h-6 w-6"
onClick={(e) => {
e.stopPropagation();
copyToClipboard(version.addedText.scenario);
}}
>
<Copy className="h-3 w-3" />
</Button>
</div>
</div>
)}
{version.removedText?.scenario && (
<div className="bg-red-50 dark:bg-red-950 p-2 rounded text-sm">
<strong>Removed:</strong> {version.removedText.scenario}
</div>
)}
{showFullText && (
<div className="space-y-1 mt-2 pt-2 border-t">
<div className="bg-gray-50 dark:bg-gray-950 p-2 rounded text-xs">
<strong>Full Old:</strong> {version.changes.scenario.old}
</div>
<div className="bg-gray-50 dark:bg-gray-950 p-2 rounded text-xs">
<strong>Full New:</strong> {version.changes.scenario.new}
</div>
</div>
)}
</div>
)}
</div>
)}
</div>
))}
</div>
</div>
)}
</DialogContent>
</Dialog>
</main>
);
}