fix: add missing hooks and fix type error

This commit is contained in:
2026-06-04 11:53:00 +02:00
parent f4cb318175
commit 8e36ec5ca5
3 changed files with 133 additions and 1 deletions
+71
View File
@@ -0,0 +1,71 @@
import { useState, useEffect, useCallback } from "react";
export type SystemPrompts = {
all: string;
partial: string;
style_normal: string;
style_crazy: string;
lyrics_partial: string;
};
export function useSystemPrompts() {
const [prompts, setPrompts] = useState<SystemPrompts | null>(null);
const [loading, setLoading] = useState(true);
const [error, setError] = useState<string | null>(null);
const fetchPrompts = useCallback(async () => {
try {
setLoading(true);
const res = await fetch("/api/system-prompts");
if (!res.ok) throw new Error("Failed to fetch system prompts");
const data = await res.json();
setPrompts(data);
setError(null);
} catch (err) {
setError(err instanceof Error ? err.message : "Unknown error");
} finally {
setLoading(false);
}
}, []);
const savePrompts = async (newPrompts: SystemPrompts) => {
try {
setLoading(true);
const res = await fetch("/api/system-prompts", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify(newPrompts),
});
if (!res.ok) throw new Error("Failed to save system prompts");
setPrompts(newPrompts);
setError(null);
return true;
} catch (err) {
setError(err instanceof Error ? err.message : "Unknown error");
return false;
} finally {
setLoading(false);
}
};
const resetPrompts = async () => {
try {
setLoading(true);
const res = await fetch("/api/system-prompts", { method: "DELETE" });
if (!res.ok) throw new Error("Failed to reset system prompts");
await fetchPrompts();
return true;
} catch (err) {
setError(err instanceof Error ? err.message : "Unknown error");
return false;
} finally {
setLoading(false);
}
};
useEffect(() => {
fetchPrompts();
}, [fetchPrompts]);
return { prompts, loading, error, savePrompts, resetPrompts };
}
+61
View File
@@ -0,0 +1,61 @@
import { useState, useEffect } from "react";
import { getConfig } from "./llm";
import type { VideoAiLink } from "./types";
const LOCAL_STORAGE_KEY = "melodymuse-video-links";
export function useVideoAiLinks() {
const [links, setLinks] = useState<VideoAiLink[]>([]);
const [loading, setLoading] = useState(true);
useEffect(() => {
let active = true;
async function fetchLinks() {
try {
const local = window.localStorage.getItem(LOCAL_STORAGE_KEY);
if (local) {
const parsed = JSON.parse(local);
if (Array.isArray(parsed)) {
if (active) {
setLinks(parsed);
setLoading(false);
}
return;
}
}
// Fetch from backend
const res = await getConfig();
if (active) {
setLinks(res.videoAiLinks || []);
setLoading(false);
}
} catch (err) {
if (active) {
setLoading(false);
}
}
}
fetchLinks();
return () => { active = false; };
}, []);
const saveLinks = (newLinks: VideoAiLink[]) => {
setLinks(newLinks);
window.localStorage.setItem(LOCAL_STORAGE_KEY, JSON.stringify(newLinks));
};
const clearLocalLinks = () => {
window.localStorage.removeItem(LOCAL_STORAGE_KEY);
// Refresh to fetch defaults again
setLoading(true);
getConfig().then(res => {
setLinks(res.videoAiLinks || []);
setLoading(false);
}).catch(() => setLoading(false));
};
return { links, loading, saveLinks, clearLocalLinks, hasLocal: window.localStorage.getItem(LOCAL_STORAGE_KEY) !== null };
}
+1 -1
View File
@@ -364,7 +364,7 @@ export function SettingsPage() {
) : ( ) : (
<div className="space-y-3"> <div className="space-y-3">
<select <select
value={activePromptTab} value={activePromptTab as string}
onChange={(e) => setActivePromptTab(e.target.value as keyof SystemPrompts)} onChange={(e) => setActivePromptTab(e.target.value as keyof SystemPrompts)}
className="input py-1.5 text-xs w-full" className="input py-1.5 text-xs w-full"
> >