import { useCallback, useState } from 'react';
import { apiClient } from '../../api/client';

export interface ImageGenerationRequest {
  prompt: string;
  negative_prompt?: string;
  provider?: 'gemini' | 'huggingface' | 'stability';
  model?: string;
  width?: number;
  height?: number;
  guidance_scale?: number;
  steps?: number;
  seed?: number;
}

export interface ImageGenerationResponse {
  success: boolean;
  image_base64: string;
  width: number;
  height: number;
  provider: string;
  model?: string;
  seed?: number;
}

export function useImageGeneration() {
  const [isGenerating, setIsGenerating] = useState(false);
  const [error, setError] = useState<string | null>(null);
  const [result, setResult] = useState<ImageGenerationResponse | null>(null);

  const generate = useCallback(async (req: ImageGenerationRequest) => {
    setIsGenerating(true);
    setError(null);
    try {
      const { data } = await apiClient.post<ImageGenerationResponse>('/api/images/generate', req);
      setResult(data);
      return data;
    } catch (e: any) {
      const message = e?.response?.data?.detail || e?.message || 'Image generation failed';
      setError(message);
      throw new Error(message);
    } finally {
      setIsGenerating(false);
    }
  }, []);

  return { isGenerating, error, result, generate };
}

export interface PromptSuggestion {
  prompt: string;
  negative_prompt?: string;
  width?: number;
  height?: number;
  overlay_text?: string;
}

export async function fetchPromptSuggestions(payload: 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);
  return response.data.suggestions || [];
}