@umituz/react-native-ai-generation-content 1.17.16 → 1.17.17

package/src/features/text-to-image/presentation/hooks/useFormState.ts

@@ -0,0 +1,103 @@
+/**
+ * Text-to-Image Form State Hook
+ * Manages form state for text-to-image generation
+ */
+
+import { useState, useCallback, useMemo } from "react";
+import type {
+  AspectRatio,
+  ImageSize,
+  NumImages,
+  OutputFormat,
+  TextToImageFormState,
+  TextToImageFormActions,
+  TextToImageFormDefaults,
+} from "../../domain/types/form.types";
+import { DEFAULT_FORM_VALUES } from "../../domain/constants/options.constants";
+
+export interface UseFormStateOptions {
+  defaults?: TextToImageFormDefaults;
+}
+
+export interface UseFormStateReturn {
+  state: TextToImageFormState;
+  actions: TextToImageFormActions;
+}
+
+export function useFormState(options?: UseFormStateOptions): UseFormStateReturn {
+  const defaults = useMemo(
+    () => ({ ...DEFAULT_FORM_VALUES, ...options?.defaults }),
+    [options?.defaults]
+  );
+
+  const [prompt, setPrompt] = useState("");
+  const [aspectRatio, setAspectRatio] = useState<AspectRatio>(
+    defaults.aspectRatio ?? "9:16"
+  );
+  const [size, setSize] = useState<ImageSize>(defaults.size ?? "512x512");
+  const [numImages, setNumImages] = useState<NumImages>(defaults.numImages ?? 1);
+  const [negativePrompt, setNegativePrompt] = useState("");
+  const [guidanceScale, setGuidanceScale] = useState(defaults.guidanceScale ?? 7.5);
+  const [selectedModel, setSelectedModel] = useState<string | null>(null);
+  const [outputFormat, setOutputFormat] = useState<OutputFormat>(
+    defaults.outputFormat ?? "png"
+  );
+  const [selectedStyle, setSelectedStyle] = useState(
+    defaults.selectedStyle ?? "realistic"
+  );
+
+  const reset = useCallback(() => {
+    setPrompt("");
+    setAspectRatio(defaults.aspectRatio ?? "9:16");
+    setSize(defaults.size ?? "512x512");
+    setNumImages(defaults.numImages ?? 1);
+    setNegativePrompt("");
+    setGuidanceScale(defaults.guidanceScale ?? 7.5);
+    setSelectedModel(null);
+    setOutputFormat(defaults.outputFormat ?? "png");
+    setSelectedStyle(defaults.selectedStyle ?? "realistic");
+  }, [defaults]);
+
+  const state: TextToImageFormState = useMemo(
+    () => ({
+      prompt,
+      aspectRatio,
+      size,
+      numImages,
+      negativePrompt,
+      guidanceScale,
+      selectedModel,
+      outputFormat,
+      selectedStyle,
+    }),
+    [
+      prompt,
+      aspectRatio,
+      size,
+      numImages,
+      negativePrompt,
+      guidanceScale,
+      selectedModel,
+      outputFormat,
+      selectedStyle,
+    ]
+  );
+
+  const actions: TextToImageFormActions = useMemo(
+    () => ({
+      setPrompt,
+      setAspectRatio,
+      setSize,
+      setNumImages,
+      setNegativePrompt,
+      setGuidanceScale,
+      setSelectedModel,
+      setOutputFormat,
+      setSelectedStyle,
+      reset,
+    }),
+    [reset]
+  );
+
+  return { state, actions };
+}

package/src/features/text-to-image/presentation/hooks/useGeneration.ts

@@ -0,0 +1,99 @@
+/**
+ * Text-to-Image Generation Hook
+ * Orchestrates generation with app-provided callbacks
+ */
+
+import { useState, useCallback } from "react";
+import type {
+  TextToImageFormState,
+  TextToImageCallbacks,
+  GenerationResult,
+  GenerationRequest,
+} from "../../domain/types";
+
+export interface GenerationState {
+  isGenerating: boolean;
+  progress: number;
+  error: string | null;
+}
+
+export interface UseGenerationOptions {
+  formState: TextToImageFormState;
+  callbacks: TextToImageCallbacks;
+  onPromptCleared?: () => void;
+}
+
+export interface UseGenerationReturn {
+  generationState: GenerationState;
+  totalCost: number;
+  handleGenerate: () => Promise<GenerationResult | null>;
+}
+
+const initialState: GenerationState = {
+  isGenerating: false,
+  progress: 0,
+  error: null,
+};
+
+export function useGeneration(options: UseGenerationOptions): UseGenerationReturn {
+  const { formState, callbacks, onPromptCleared } = options;
+  const [generationState, setGenerationState] = useState<GenerationState>(initialState);
+
+  const totalCost = callbacks.calculateCost(formState.numImages, formState.selectedModel);
+
+  const handleGenerate = useCallback(async (): Promise<GenerationResult | null> => {
+    const trimmedPrompt = formState.prompt.trim();
+
+    if (!trimmedPrompt) {
+      setGenerationState((prev) => ({ ...prev, error: "Prompt is required" }));
+      return null;
+    }
+
+    if (!callbacks.isAuthenticated()) {
+      callbacks.onAuthRequired?.();
+      return null;
+    }
+
+    if (!callbacks.canAfford(totalCost)) {
+      callbacks.onCreditsRequired?.(totalCost);
+      return null;
+    }
+
+    setGenerationState({ isGenerating: true, progress: 0, error: null });
+
+    const request: GenerationRequest = {
+      prompt: trimmedPrompt,
+      model: formState.selectedModel ?? undefined,
+      aspectRatio: formState.aspectRatio,
+      size: formState.size,
+      negativePrompt: formState.negativePrompt.trim() || undefined,
+      guidanceScale: formState.guidanceScale,
+      numImages: formState.numImages,
+      style: formState.selectedStyle,
+      outputFormat: formState.outputFormat,
+    };
+
+    try {
+      const result = await callbacks.executeGeneration(request);
+
+      if (result.success) {
+        callbacks.onSuccess?.(result.imageUrls);
+        onPromptCleared?.();
+      } else {
+        const errorMessage = result.error;
+        setGenerationState((prev) => ({ ...prev, error: errorMessage }));
+        callbacks.onError?.(errorMessage);
+      }
+
+      setGenerationState({ isGenerating: false, progress: 100, error: null });
+      return result;
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      setGenerationState({ isGenerating: false, progress: 0, error: message });
+      callbacks.onError?.(message);
+      return null;
+    }
+  }, [formState, callbacks, totalCost, onPromptCleared]);
+
+  return { generationState, totalCost, handleGenerate };
+}

package/src/features/text-to-image/presentation/hooks/useTextToImageForm.ts

@@ -0,0 +1,58 @@
+/**
+ * Text-to-Image Form Hook
+ * Composes form state and generation for complete form management
+ */
+
+import { useMemo } from "react";
+import { useFormState, type UseFormStateOptions } from "./useFormState";
+import { useGeneration, type GenerationState } from "./useGeneration";
+import type {
+  TextToImageFormState,
+  TextToImageFormActions,
+  TextToImageCallbacks,
+} from "../../domain/types";
+
+export interface UseTextToImageFormOptions extends UseFormStateOptions {
+  callbacks: TextToImageCallbacks;
+}
+
+export interface UseTextToImageFormReturn {
+  state: TextToImageFormState;
+  actions: TextToImageFormActions;
+  generationState: GenerationState;
+  totalCost: number;
+  handleGenerate: () => Promise<void>;
+  isReady: boolean;
+}
+
+export function useTextToImageForm(
+  options: UseTextToImageFormOptions
+): UseTextToImageFormReturn {
+  const { callbacks, defaults } = options;
+
+  const { state, actions } = useFormState({ defaults });
+
+  const { generationState, totalCost, handleGenerate } = useGeneration({
+    formState: state,
+    callbacks,
+    onPromptCleared: actions.reset,
+  });
+
+  const isReady = useMemo(
+    () => state.prompt.trim().length > 0 && !generationState.isGenerating,
+    [state.prompt, generationState.isGenerating]
+  );
+
+  const wrappedHandleGenerate = async () => {
+    await handleGenerate();
+  };
+
+  return {
+    state,
+    actions,
+    generationState,
+    totalCost,
+    handleGenerate: wrappedHandleGenerate,
+    isReady,
+  };
+}

package/src/features/text-to-image/presentation/index.ts

@@ -1 +1,7 @@
+/**
+ * Text-to-Image Presentation Layer
+ * Hooks and components exports
+ */
+
 export * from "./hooks";
+export * from "./components";