794 lines
20 KiB
TypeScript
794 lines
20 KiB
TypeScript
import { FolkShape } from "./folk-shape";
|
||
import { css, html } from "./tags";
|
||
|
||
const styles = css`
|
||
:host {
|
||
background: white;
|
||
border-radius: 8px;
|
||
box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
|
||
min-width: 420px;
|
||
min-height: 520px;
|
||
}
|
||
|
||
.header {
|
||
display: flex;
|
||
align-items: center;
|
||
justify-content: space-between;
|
||
padding: 8px 12px;
|
||
background: linear-gradient(135deg, #f59e0b, #ec4899);
|
||
color: white;
|
||
border-radius: 8px 8px 0 0;
|
||
font-size: 12px;
|
||
font-weight: 600;
|
||
cursor: move;
|
||
}
|
||
|
||
.header-title {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 6px;
|
||
}
|
||
|
||
.header-actions {
|
||
display: flex;
|
||
gap: 4px;
|
||
}
|
||
|
||
.header-actions button {
|
||
background: transparent;
|
||
border: none;
|
||
color: white;
|
||
cursor: pointer;
|
||
padding: 2px 6px;
|
||
border-radius: 4px;
|
||
font-size: 14px;
|
||
}
|
||
|
||
.header-actions button:hover {
|
||
background: rgba(255, 255, 255, 0.2);
|
||
}
|
||
|
||
.content {
|
||
display: flex;
|
||
flex-direction: column;
|
||
height: calc(100% - 36px);
|
||
overflow-y: auto;
|
||
}
|
||
|
||
.section {
|
||
padding: 12px;
|
||
border-bottom: 1px solid #e2e8f0;
|
||
}
|
||
|
||
.section-label {
|
||
font-size: 11px;
|
||
font-weight: 600;
|
||
color: #64748b;
|
||
text-transform: uppercase;
|
||
letter-spacing: 0.5px;
|
||
margin-bottom: 8px;
|
||
}
|
||
|
||
/* Brand references */
|
||
.upload-zone {
|
||
border: 2px dashed #cbd5e1;
|
||
border-radius: 8px;
|
||
padding: 16px;
|
||
text-align: center;
|
||
cursor: pointer;
|
||
font-size: 12px;
|
||
color: #94a3b8;
|
||
transition: border-color 0.2s;
|
||
}
|
||
|
||
.upload-zone:hover, .upload-zone.drag-over {
|
||
border-color: #f59e0b;
|
||
color: #f59e0b;
|
||
}
|
||
|
||
.thumb-grid {
|
||
display: flex;
|
||
flex-wrap: wrap;
|
||
gap: 6px;
|
||
margin-top: 8px;
|
||
}
|
||
|
||
.thumb {
|
||
position: relative;
|
||
width: 60px;
|
||
height: 60px;
|
||
border-radius: 6px;
|
||
overflow: hidden;
|
||
}
|
||
|
||
.thumb img {
|
||
width: 100%;
|
||
height: 100%;
|
||
object-fit: cover;
|
||
}
|
||
|
||
.thumb .remove-btn {
|
||
position: absolute;
|
||
top: 2px;
|
||
right: 2px;
|
||
width: 16px;
|
||
height: 16px;
|
||
background: rgba(0,0,0,0.6);
|
||
color: white;
|
||
border: none;
|
||
border-radius: 50%;
|
||
font-size: 10px;
|
||
line-height: 16px;
|
||
text-align: center;
|
||
cursor: pointer;
|
||
padding: 0;
|
||
}
|
||
|
||
.analyze-btn, .generate-btn {
|
||
padding: 8px 16px;
|
||
background: linear-gradient(135deg, #f59e0b, #ec4899);
|
||
color: white;
|
||
border: none;
|
||
border-radius: 6px;
|
||
font-size: 13px;
|
||
font-weight: 600;
|
||
cursor: pointer;
|
||
transition: opacity 0.2s;
|
||
}
|
||
|
||
.analyze-btn:hover, .generate-btn:hover {
|
||
opacity: 0.9;
|
||
}
|
||
|
||
.analyze-btn:disabled, .generate-btn:disabled {
|
||
opacity: 0.5;
|
||
cursor: not-allowed;
|
||
}
|
||
|
||
/* Style analysis display */
|
||
.style-display {
|
||
margin-top: 8px;
|
||
padding: 10px;
|
||
background: #fefce8;
|
||
border-radius: 6px;
|
||
font-size: 12px;
|
||
}
|
||
|
||
.palette {
|
||
display: flex;
|
||
gap: 4px;
|
||
margin: 6px 0;
|
||
}
|
||
|
||
.swatch {
|
||
width: 24px;
|
||
height: 24px;
|
||
border-radius: 4px;
|
||
border: 1px solid #e2e8f0;
|
||
}
|
||
|
||
.keywords {
|
||
display: flex;
|
||
flex-wrap: wrap;
|
||
gap: 4px;
|
||
margin: 6px 0;
|
||
}
|
||
|
||
.keyword-tag {
|
||
padding: 2px 8px;
|
||
background: #fef3c7;
|
||
border-radius: 12px;
|
||
font-size: 11px;
|
||
color: #92400e;
|
||
}
|
||
|
||
.brand-prefix {
|
||
width: 100%;
|
||
padding: 6px 8px;
|
||
border: 1px solid #e2e8f0;
|
||
border-radius: 4px;
|
||
font-size: 11px;
|
||
resize: vertical;
|
||
font-family: inherit;
|
||
min-height: 40px;
|
||
}
|
||
|
||
/* Source + Controls */
|
||
.source-preview {
|
||
margin-top: 8px;
|
||
}
|
||
|
||
.source-preview img {
|
||
max-width: 100%;
|
||
max-height: 150px;
|
||
border-radius: 6px;
|
||
}
|
||
|
||
.prompt-input {
|
||
width: 100%;
|
||
padding: 10px 12px;
|
||
border: 2px solid #e2e8f0;
|
||
border-radius: 8px;
|
||
font-size: 13px;
|
||
resize: none;
|
||
outline: none;
|
||
font-family: inherit;
|
||
margin-top: 8px;
|
||
}
|
||
|
||
.prompt-input:focus {
|
||
border-color: #ec4899;
|
||
}
|
||
|
||
.controls {
|
||
display: flex;
|
||
gap: 8px;
|
||
margin-top: 8px;
|
||
align-items: center;
|
||
flex-wrap: wrap;
|
||
}
|
||
|
||
.provider-select {
|
||
padding: 6px 10px;
|
||
border: 2px solid #e2e8f0;
|
||
border-radius: 6px;
|
||
font-size: 12px;
|
||
background: white;
|
||
cursor: pointer;
|
||
}
|
||
|
||
.strength-group {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 4px;
|
||
font-size: 11px;
|
||
color: #64748b;
|
||
}
|
||
|
||
.strength-slider {
|
||
width: 80px;
|
||
}
|
||
|
||
.brand-toggle {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 4px;
|
||
font-size: 11px;
|
||
color: #64748b;
|
||
}
|
||
|
||
/* Results */
|
||
.image-area {
|
||
flex: 1;
|
||
padding: 12px;
|
||
overflow-y: auto;
|
||
display: flex;
|
||
flex-direction: column;
|
||
gap: 12px;
|
||
}
|
||
|
||
.placeholder {
|
||
flex: 1;
|
||
display: flex;
|
||
flex-direction: column;
|
||
align-items: center;
|
||
justify-content: center;
|
||
color: #94a3b8;
|
||
text-align: center;
|
||
gap: 8px;
|
||
padding: 24px;
|
||
}
|
||
|
||
.placeholder-icon {
|
||
font-size: 48px;
|
||
opacity: 0.5;
|
||
}
|
||
|
||
.image-item {
|
||
position: relative;
|
||
}
|
||
|
||
.generated-image {
|
||
width: 100%;
|
||
border-radius: 8px;
|
||
box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
|
||
}
|
||
|
||
.image-meta {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 6px;
|
||
font-size: 11px;
|
||
color: #64748b;
|
||
margin-top: 4px;
|
||
padding: 4px 8px;
|
||
background: #f1f5f9;
|
||
border-radius: 4px;
|
||
}
|
||
|
||
.provider-badge {
|
||
padding: 1px 6px;
|
||
border-radius: 8px;
|
||
font-size: 10px;
|
||
font-weight: 600;
|
||
color: white;
|
||
}
|
||
|
||
.provider-badge.fal { background: #8b5cf6; }
|
||
.provider-badge.gemini { background: #0ea5e9; }
|
||
|
||
.loading {
|
||
display: flex;
|
||
flex-direction: column;
|
||
align-items: center;
|
||
justify-content: center;
|
||
padding: 24px;
|
||
gap: 12px;
|
||
}
|
||
|
||
.spinner {
|
||
width: 32px;
|
||
height: 32px;
|
||
border: 3px solid #e2e8f0;
|
||
border-top-color: #ec4899;
|
||
border-radius: 50%;
|
||
animation: spin 1s linear infinite;
|
||
}
|
||
|
||
@keyframes spin {
|
||
to { transform: rotate(360deg); }
|
||
}
|
||
|
||
.error {
|
||
color: #ef4444;
|
||
padding: 12px;
|
||
background: #fef2f2;
|
||
border-radius: 6px;
|
||
font-size: 13px;
|
||
}
|
||
`;
|
||
|
||
interface StyleAnalysis {
|
||
style_description: string;
|
||
color_palette: string[];
|
||
style_keywords: string[];
|
||
brand_prompt_prefix: string;
|
||
}
|
||
|
||
interface StudioResult {
|
||
id: string;
|
||
prompt: string;
|
||
url: string;
|
||
provider: string;
|
||
timestamp: string;
|
||
}
|
||
|
||
declare global {
|
||
interface HTMLElementTagNameMap {
|
||
"folk-image-studio": FolkImageStudio;
|
||
}
|
||
}
|
||
|
||
export class FolkImageStudio extends FolkShape {
|
||
static override tagName = "folk-image-studio";
|
||
|
||
static {
|
||
const sheet = new CSSStyleSheet();
|
||
const parentRules = Array.from(FolkShape.styles.cssRules)
|
||
.map((r) => r.cssText)
|
||
.join("\n");
|
||
const childRules = Array.from(styles.cssRules)
|
||
.map((r) => r.cssText)
|
||
.join("\n");
|
||
sheet.replaceSync(`${parentRules}\n${childRules}`);
|
||
this.styles = sheet;
|
||
}
|
||
|
||
#brandReferences: string[] = []; // server paths or data URLs
|
||
#styleAnalysis: StyleAnalysis | null = null;
|
||
#sourceImage: string | null = null;
|
||
#results: StudioResult[] = [];
|
||
#provider: string = "fal";
|
||
#strength = 0.85;
|
||
#applyBrandStyle = true;
|
||
#isAnalyzing = false;
|
||
#isGenerating = false;
|
||
#error: string | null = null;
|
||
|
||
// DOM refs
|
||
#wrapper: HTMLElement | null = null;
|
||
#refUploadInput: HTMLInputElement | null = null;
|
||
#sourceUploadInput: HTMLInputElement | null = null;
|
||
#promptInput: HTMLTextAreaElement | null = null;
|
||
#brandPrefixInput: HTMLTextAreaElement | null = null;
|
||
|
||
override createRenderRoot() {
|
||
const root = super.createRenderRoot();
|
||
|
||
this.#wrapper = document.createElement("div");
|
||
this.#wrapper.innerHTML = this.#renderAll();
|
||
|
||
const slot = root.querySelector("slot");
|
||
const containerDiv = slot?.parentElement as HTMLElement;
|
||
if (containerDiv) containerDiv.replaceWith(this.#wrapper);
|
||
|
||
this.#bindEvents();
|
||
return root;
|
||
}
|
||
|
||
#renderAll(): string {
|
||
return html`
|
||
<div class="header">
|
||
<span class="header-title">
|
||
<span>🖌️</span>
|
||
<span>Image Studio</span>
|
||
</span>
|
||
<div class="header-actions">
|
||
<button class="close-btn" title="Close">×</button>
|
||
</div>
|
||
</div>
|
||
<div class="content">
|
||
<!-- Brand References -->
|
||
<div class="section">
|
||
<div class="section-label">Brand References</div>
|
||
<div class="upload-zone ref-upload-zone">
|
||
Click or drag images to add brand references
|
||
</div>
|
||
<input type="file" class="ref-upload-input" accept="image/*" multiple hidden />
|
||
${this.#renderThumbs()}
|
||
${this.#brandReferences.length >= 1
|
||
? `<div style="margin-top:8px"><button class="analyze-btn" ${this.#isAnalyzing ? "disabled" : ""}>${this.#isAnalyzing ? "Analyzing..." : "Analyze Style"}</button></div>`
|
||
: ""}
|
||
${this.#renderStyleDisplay()}
|
||
</div>
|
||
|
||
<!-- Source + Controls -->
|
||
<div class="section">
|
||
<div class="section-label">Source Image + Prompt</div>
|
||
<div class="upload-zone source-upload-zone" style="padding:10px;">
|
||
${this.#sourceImage
|
||
? `<div class="source-preview"><img src="${this.#escapeHtml(this.#sourceImage)}" /></div>`
|
||
: "Click or drag a source image"}
|
||
</div>
|
||
<input type="file" class="source-upload-input" accept="image/*" hidden />
|
||
<textarea class="prompt-input" placeholder="Describe how to transform the image..." rows="2"></textarea>
|
||
<div class="controls">
|
||
<select class="provider-select">
|
||
<option value="fal" ${this.#provider === "fal" ? "selected" : ""}>fal.ai FLUX</option>
|
||
<option value="gemini" ${this.#provider === "gemini" ? "selected" : ""}>Nano Banana</option>
|
||
<option value="gemini-pro" ${this.#provider === "gemini-pro" ? "selected" : ""}>Nano Banana Pro</option>
|
||
<option value="gemini-flash-2" ${this.#provider === "gemini-flash-2" ? "selected" : ""}>Nano Banana 2</option>
|
||
</select>
|
||
<div class="strength-group" style="${this.#provider === "fal" ? "" : "display:none"}">
|
||
<label>Strength</label>
|
||
<input type="range" class="strength-slider" min="0" max="1" step="0.05" value="${this.#strength}" />
|
||
<span class="strength-val">${this.#strength}</span>
|
||
</div>
|
||
<label class="brand-toggle">
|
||
<input type="checkbox" class="brand-check" ${this.#applyBrandStyle ? "checked" : ""} />
|
||
Apply Brand Style
|
||
</label>
|
||
<button class="generate-btn" ${this.#isGenerating ? "disabled" : ""}>${this.#isGenerating ? "Generating..." : "Generate"}</button>
|
||
</div>
|
||
</div>
|
||
|
||
<!-- Results -->
|
||
<div class="image-area">
|
||
${this.#error ? `<div class="error">${this.#escapeHtml(this.#error)}</div>` : ""}
|
||
${this.#isGenerating ? '<div class="loading"><div class="spinner"></div><span>Generating image...</span></div>' : ""}
|
||
${this.#results.length > 0 ? this.#renderImageList() : (!this.#isGenerating ? '<div class="placeholder"><span class="placeholder-icon">🖌️</span><span>Upload a source image and generate</span></div>' : "")}
|
||
</div>
|
||
</div>
|
||
`;
|
||
}
|
||
|
||
#renderThumbs(): string {
|
||
if (!this.#brandReferences.length) return "";
|
||
return `<div class="thumb-grid">${this.#brandReferences
|
||
.map(
|
||
(ref, i) => `<div class="thumb">
|
||
<img src="${this.#escapeHtml(ref)}" />
|
||
<button class="remove-btn" data-ref-index="${i}">×</button>
|
||
</div>`
|
||
)
|
||
.join("")}</div>`;
|
||
}
|
||
|
||
#renderStyleDisplay(): string {
|
||
if (!this.#styleAnalysis) return "";
|
||
const s = this.#styleAnalysis;
|
||
return `
|
||
<div class="style-display">
|
||
<div style="margin-bottom:4px;font-weight:600;">Style Analysis</div>
|
||
<div style="margin-bottom:6px;">${this.#escapeHtml(s.style_description)}</div>
|
||
<div class="palette">${s.color_palette.map((c) => `<div class="swatch" style="background:${c}" title="${c}"></div>`).join("")}</div>
|
||
<div class="keywords">${s.style_keywords.map((k) => `<span class="keyword-tag">${this.#escapeHtml(k)}</span>`).join("")}</div>
|
||
<div class="section-label" style="margin-top:6px;">Brand Prompt Prefix</div>
|
||
<textarea class="brand-prefix">${this.#escapeHtml(s.brand_prompt_prefix)}</textarea>
|
||
</div>
|
||
`;
|
||
}
|
||
|
||
#renderImageList(): string {
|
||
return this.#results
|
||
.map(
|
||
(r) => `
|
||
<div class="image-item">
|
||
<img class="generated-image" src="${this.#escapeHtml(r.url)}" loading="lazy" />
|
||
<div class="image-meta">
|
||
<span class="provider-badge ${r.provider === "fal" ? "fal" : "gemini"}">${this.#escapeHtml(r.provider)}</span>
|
||
<span>${this.#escapeHtml(r.prompt)}</span>
|
||
</div>
|
||
</div>
|
||
`
|
||
)
|
||
.join("");
|
||
}
|
||
|
||
#rerender() {
|
||
if (!this.#wrapper) return;
|
||
this.#wrapper.innerHTML = this.#renderAll();
|
||
this.#bindEvents();
|
||
}
|
||
|
||
#bindEvents() {
|
||
if (!this.#wrapper) return;
|
||
|
||
const w = this.#wrapper;
|
||
this.#refUploadInput = w.querySelector(".ref-upload-input");
|
||
this.#sourceUploadInput = w.querySelector(".source-upload-input");
|
||
this.#promptInput = w.querySelector(".prompt-input");
|
||
this.#brandPrefixInput = w.querySelector(".brand-prefix");
|
||
|
||
// Close
|
||
w.querySelector(".close-btn")?.addEventListener("click", (e) => {
|
||
e.stopPropagation();
|
||
this.dispatchEvent(new CustomEvent("close"));
|
||
});
|
||
|
||
// Ref upload zone
|
||
const refZone = w.querySelector(".ref-upload-zone") as HTMLElement | null;
|
||
refZone?.addEventListener("click", (e) => {
|
||
e.stopPropagation();
|
||
this.#refUploadInput?.click();
|
||
});
|
||
refZone?.addEventListener("dragover", (e) => { e.preventDefault(); refZone.classList.add("drag-over"); });
|
||
refZone?.addEventListener("dragleave", () => refZone.classList.remove("drag-over"));
|
||
refZone?.addEventListener("drop", (e) => {
|
||
e.preventDefault();
|
||
refZone.classList.remove("drag-over");
|
||
if (e.dataTransfer?.files) this.#handleRefFiles(e.dataTransfer.files);
|
||
});
|
||
this.#refUploadInput?.addEventListener("change", () => {
|
||
if (this.#refUploadInput?.files) this.#handleRefFiles(this.#refUploadInput.files);
|
||
});
|
||
|
||
// Source upload zone
|
||
const srcZone = w.querySelector(".source-upload-zone") as HTMLElement | null;
|
||
srcZone?.addEventListener("click", (e) => {
|
||
e.stopPropagation();
|
||
this.#sourceUploadInput?.click();
|
||
});
|
||
srcZone?.addEventListener("dragover", (e) => { e.preventDefault(); srcZone.classList.add("drag-over"); });
|
||
srcZone?.addEventListener("dragleave", () => srcZone.classList.remove("drag-over"));
|
||
srcZone?.addEventListener("drop", (e) => {
|
||
e.preventDefault();
|
||
srcZone.classList.remove("drag-over");
|
||
if (e.dataTransfer?.files?.[0]) this.#handleSourceFile(e.dataTransfer.files[0]);
|
||
});
|
||
this.#sourceUploadInput?.addEventListener("change", () => {
|
||
if (this.#sourceUploadInput?.files?.[0]) this.#handleSourceFile(this.#sourceUploadInput.files[0]);
|
||
});
|
||
|
||
// Remove ref buttons
|
||
w.querySelectorAll(".remove-btn[data-ref-index]").forEach((btn) => {
|
||
btn.addEventListener("click", (e) => {
|
||
e.stopPropagation();
|
||
const idx = parseInt((btn as HTMLElement).dataset.refIndex || "0", 10);
|
||
this.#brandReferences.splice(idx, 1);
|
||
this.#rerender();
|
||
});
|
||
});
|
||
|
||
// Analyze button
|
||
w.querySelector(".analyze-btn")?.addEventListener("click", (e) => {
|
||
e.stopPropagation();
|
||
this.#analyzeStyle();
|
||
});
|
||
|
||
// Provider select
|
||
const provSelect = w.querySelector(".provider-select") as HTMLSelectElement | null;
|
||
provSelect?.addEventListener("change", () => {
|
||
this.#provider = provSelect.value;
|
||
this.#rerender();
|
||
});
|
||
|
||
// Strength slider
|
||
const slider = w.querySelector(".strength-slider") as HTMLInputElement | null;
|
||
slider?.addEventListener("input", () => {
|
||
this.#strength = parseFloat(slider.value);
|
||
const valEl = w.querySelector(".strength-val");
|
||
if (valEl) valEl.textContent = String(this.#strength);
|
||
});
|
||
|
||
// Brand toggle
|
||
const brandCheck = w.querySelector(".brand-check") as HTMLInputElement | null;
|
||
brandCheck?.addEventListener("change", () => {
|
||
this.#applyBrandStyle = brandCheck.checked;
|
||
});
|
||
|
||
// Brand prefix textarea
|
||
this.#brandPrefixInput?.addEventListener("input", () => {
|
||
if (this.#styleAnalysis && this.#brandPrefixInput) {
|
||
this.#styleAnalysis.brand_prompt_prefix = this.#brandPrefixInput.value;
|
||
}
|
||
});
|
||
|
||
// Generate button
|
||
w.querySelector(".generate-btn")?.addEventListener("click", (e) => {
|
||
e.stopPropagation();
|
||
this.#generate();
|
||
});
|
||
|
||
// Prevent drag on interactive elements
|
||
this.#promptInput?.addEventListener("pointerdown", (e) => e.stopPropagation());
|
||
this.#brandPrefixInput?.addEventListener("pointerdown", (e) => e.stopPropagation());
|
||
slider?.addEventListener("pointerdown", (e) => e.stopPropagation());
|
||
provSelect?.addEventListener("pointerdown", (e) => e.stopPropagation());
|
||
brandCheck?.addEventListener("pointerdown", (e) => e.stopPropagation());
|
||
}
|
||
|
||
#fileToDataUrl(file: File): Promise<string> {
|
||
return new Promise((resolve, reject) => {
|
||
const reader = new FileReader();
|
||
reader.onload = () => resolve(reader.result as string);
|
||
reader.onerror = reject;
|
||
reader.readAsDataURL(file);
|
||
});
|
||
}
|
||
|
||
async #handleRefFiles(files: FileList) {
|
||
for (const file of Array.from(files)) {
|
||
if (!file.type.startsWith("image/")) continue;
|
||
try {
|
||
// Upload to server for persistence
|
||
const dataUrl = await this.#fileToDataUrl(file);
|
||
const res = await fetch("/api/image-upload", {
|
||
method: "POST",
|
||
headers: { "Content-Type": "application/json" },
|
||
body: JSON.stringify({ image: dataUrl }),
|
||
});
|
||
if (res.ok) {
|
||
const { url } = await res.json();
|
||
this.#brandReferences.push(url);
|
||
} else {
|
||
this.#brandReferences.push(dataUrl);
|
||
}
|
||
} catch {
|
||
const dataUrl = await this.#fileToDataUrl(file);
|
||
this.#brandReferences.push(dataUrl);
|
||
}
|
||
}
|
||
this.#rerender();
|
||
}
|
||
|
||
async #handleSourceFile(file: File) {
|
||
if (!file.type.startsWith("image/")) return;
|
||
try {
|
||
const dataUrl = await this.#fileToDataUrl(file);
|
||
const res = await fetch("/api/image-upload", {
|
||
method: "POST",
|
||
headers: { "Content-Type": "application/json" },
|
||
body: JSON.stringify({ image: dataUrl }),
|
||
});
|
||
if (res.ok) {
|
||
const { url } = await res.json();
|
||
this.#sourceImage = url;
|
||
} else {
|
||
this.#sourceImage = dataUrl;
|
||
}
|
||
} catch {
|
||
this.#sourceImage = await this.#fileToDataUrl(file);
|
||
}
|
||
this.#rerender();
|
||
}
|
||
|
||
async #analyzeStyle() {
|
||
if (this.#isAnalyzing || !this.#brandReferences.length) return;
|
||
this.#isAnalyzing = true;
|
||
this.#error = null;
|
||
this.#rerender();
|
||
|
||
try {
|
||
const res = await fetch("/api/image-gen/analyze-style", {
|
||
method: "POST",
|
||
headers: { "Content-Type": "application/json" },
|
||
body: JSON.stringify({ images: this.#brandReferences }),
|
||
});
|
||
if (!res.ok) throw new Error("Style analysis failed");
|
||
this.#styleAnalysis = await res.json();
|
||
} catch (e: any) {
|
||
this.#error = e.message || "Analysis failed";
|
||
} finally {
|
||
this.#isAnalyzing = false;
|
||
this.#rerender();
|
||
}
|
||
}
|
||
|
||
async #generate() {
|
||
const prompt = this.#promptInput?.value.trim();
|
||
if (!prompt || !this.#sourceImage || this.#isGenerating) return;
|
||
|
||
this.#isGenerating = true;
|
||
this.#error = null;
|
||
this.#rerender();
|
||
|
||
try {
|
||
let fullPrompt = prompt;
|
||
if (this.#applyBrandStyle && this.#styleAnalysis?.brand_prompt_prefix) {
|
||
fullPrompt = this.#styleAnalysis.brand_prompt_prefix + " " + prompt;
|
||
}
|
||
|
||
const isGemini = this.#provider !== "fal";
|
||
const body: any = {
|
||
prompt: fullPrompt,
|
||
source_image: this.#sourceImage,
|
||
provider: isGemini ? "gemini" : "fal",
|
||
strength: this.#strength,
|
||
};
|
||
|
||
if (isGemini && this.#provider !== "gemini") {
|
||
body.model = this.#provider; // "gemini-pro" or "gemini-flash-2"
|
||
}
|
||
|
||
if (this.#brandReferences.length) {
|
||
body.reference_images = this.#brandReferences;
|
||
}
|
||
|
||
const res = await fetch("/api/image-gen/img2img", {
|
||
method: "POST",
|
||
headers: { "Content-Type": "application/json" },
|
||
body: JSON.stringify(body),
|
||
});
|
||
|
||
if (!res.ok) throw new Error("Generation failed");
|
||
const data = await res.json();
|
||
const url = data.url || data.image_url;
|
||
if (!url) throw new Error("No image returned");
|
||
|
||
this.#results.unshift({
|
||
id: crypto.randomUUID(),
|
||
prompt,
|
||
url,
|
||
provider: this.#provider,
|
||
timestamp: new Date().toISOString(),
|
||
});
|
||
|
||
if (this.#promptInput) this.#promptInput.value = "";
|
||
} catch (e: any) {
|
||
this.#error = e.message || "Generation failed";
|
||
} finally {
|
||
this.#isGenerating = false;
|
||
this.#rerender();
|
||
}
|
||
}
|
||
|
||
#escapeHtml(text: string): string {
|
||
const div = document.createElement("div");
|
||
div.textContent = text;
|
||
return div.innerHTML;
|
||
}
|
||
|
||
override toJSON() {
|
||
return {
|
||
...super.toJSON(),
|
||
type: "folk-image-studio",
|
||
provider: this.#provider,
|
||
strength: this.#strength,
|
||
styleAnalysis: this.#styleAnalysis,
|
||
brandReferences: this.#brandReferences,
|
||
results: this.#results,
|
||
};
|
||
}
|
||
}
|