kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AI Analysis and Content Strategy fixes. Enhanced Strategy Routes refactoring.

Browse Source
This commit is contained in:
ajaysi
2026-01-10 19:32:50 +05:30
parent 0b63ae7fc1
commit 8193cdba67
298 changed files with 45678 additions and 10952 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
frontend/src/components/ImageGen/useImageGeneration.ts
View File

@@ -4,7 +4,7 @@ import { apiClient } from '../../api/client';
export interface ImageGenerationRequest {
prompt: string;
negative_prompt?: string;
provider?: 'gemini' | 'huggingface' | 'stability';
provider?: 'gemini' | 'huggingface' | 'stability' | 'wavespeed';
model?: string;
width?: number;
height?: number;
@@ -31,12 +31,34 @@ export function useImageGeneration() {
const generate = useCallback(async (req: ImageGenerationRequest) => {
setIsGenerating(true);
setError(null);
setResult(null);
try {
const { data } = await apiClient.post<ImageGenerationResponse>('/api/images/generate', req);
setResult(data);
return data;
const response = await apiClient.post<ImageGenerationResponse>('/api/images/generate', req);
const data = response.data;
// Check if response has success field and image data
if (data && (data.success !== false) && data.image_base64) {
setResult(data);
setError(null);
return data;
} else {
// Response received but missing required data
const message = 'Image generation completed but response is incomplete';
setError(message);
throw new Error(message);
}
} catch (e: any) {
const message = e?.response?.data?.detail || e?.message || 'Image generation failed';
// Check if error response contains image data (partial success)
if (e?.response?.data?.image_base64) {
// Image was generated but there was an error in post-processing
const data = e.response.data;
console.warn('Image generation succeeded but post-processing had issues', data);
setResult(data);
setError(null);
return data;
}
const message = e?.response?.data?.detail || e?.response?.data?.message || e?.message || 'Image generation failed';
setError(message);
throw new Error(message);
} finally {
@@ -55,7 +77,15 @@ export interface PromptSuggestion {
overlay_text?: string;
}
export async function fetchPromptSuggestions(payload: any): Promise<PromptSuggestion[]> {
export async function fetchPromptSuggestions(payload: {
provider?: string;
model?: string;
image_type?: string;
title?: string;
section?: any;
research?: any;
persona?: any;
}): Promise<PromptSuggestion[]> {
// Use apiClient directly (same pattern as SEO analysis in SEOAnalysisModal.tsx)
// The apiClient interceptor will handle auth token injection automatically
const response = await apiClient.post('/api/images/suggest-prompts', payload);