d
This commit is contained in:
@@ -73,6 +73,8 @@ export default function Images() {
|
||||
const [imageQueue, setImageQueue] = useState<ImageQueueItem[]>([]);
|
||||
const [currentContentId, setCurrentContentId] = useState<number | null>(null);
|
||||
const [taskId, setTaskId] = useState<string | null>(null);
|
||||
const [imageModel, setImageModel] = useState<string | null>(null);
|
||||
const [imageProvider, setImageProvider] = useState<string | null>(null);
|
||||
|
||||
// Load images - wrapped in useCallback
|
||||
const loadImages = useCallback(async () => {
|
||||
@@ -231,17 +233,25 @@ export default function Images() {
|
||||
return;
|
||||
}
|
||||
|
||||
// Fetch image generation settings to get max_in_article_images
|
||||
// Fetch image generation settings to get max_in_article_images, model, and provider
|
||||
let maxInArticleImages = 2; // Default
|
||||
let model = null;
|
||||
let provider = null;
|
||||
try {
|
||||
const settings = await fetchImageGenerationSettings();
|
||||
if (settings.success && settings.config) {
|
||||
maxInArticleImages = settings.config.max_in_article_images || 2;
|
||||
model = settings.config.model || null;
|
||||
provider = settings.config.provider || null;
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('Failed to fetch image settings, using default:', error);
|
||||
}
|
||||
|
||||
// Store model and provider for modal display
|
||||
setImageModel(model);
|
||||
setImageProvider(provider);
|
||||
|
||||
// Build image queue
|
||||
const queue = buildImageQueue(contentId, maxInArticleImages);
|
||||
|
||||
@@ -394,12 +404,16 @@ export default function Images() {
|
||||
setImageQueue([]);
|
||||
setCurrentContentId(null);
|
||||
setTaskId(null);
|
||||
setImageModel(null);
|
||||
setImageProvider(null);
|
||||
// Reload images after closing if generation completed
|
||||
loadImages();
|
||||
}}
|
||||
queue={imageQueue}
|
||||
totalImages={imageQueue.length}
|
||||
taskId={taskId}
|
||||
model={imageModel || undefined}
|
||||
provider={imageProvider || undefined}
|
||||
onUpdateQueue={setImageQueue}
|
||||
onLog={addAiLog}
|
||||
/>
|
||||
|
||||
Reference in New Issue
Block a user