Enhance extensive ad generation with custom niche support

- Updated the ExtensiveGenerateRequest model to include an optional custom_niche field for enhanced flexibility in ad generation.
- Modified the generate_extensive function to validate custom niche input when 'others' is selected as the niche.
- Adjusted frontend components to accommodate the new custom_niche field, including validation in the ExtensiveForm and updates to the GeneratePage.
- Updated API endpoints to handle the new custom_niche parameter, ensuring seamless integration across the application.

frontend/app/generate/page.tsx

@@ -365,6 +365,7 @@ export default function GeneratePage() {
 
   const handleExtensiveGenerate = async (data: {
     niche: Niche;
+    custom_niche?: string;
     target_audience?: string;
     offer?: string;
     num_images: number;

frontend/components/generation/ExtensiveForm.tsx

@@ -10,19 +10,32 @@ import { IMAGE_MODELS } from "@/lib/constants/models";
 import type { Niche } from "@/types/api";
 
 const extensiveSchema = z.object({
-  niche: z.enum(["home_insurance", "glp1"]),
+  niche: z.enum(["home_insurance", "glp1", "others"]),
+  custom_niche: z.string().optional(),
   target_audience: z.string().optional(),
   offer: z.string().optional(),
   num_images: z.number().min(1).max(3),
   num_strategies: z.number().min(1).max(10),
   image_model: z.string().nullable().optional(),
-});
+}).refine(
+  (data) => {
+    if (data.niche === "others") {
+      return data.custom_niche && data.custom_niche.trim().length > 0;
+    }
+    return true;
+  },
+  {
+    message: "Please enter your custom niche",
+    path: ["custom_niche"],
+  }
+);
 
 type ExtensiveFormData = z.infer<typeof extensiveSchema>;
 
 interface ExtensiveFormProps {
   onSubmit: (data: {
     niche: Niche;
+    custom_niche?: string;
     target_audience?: string;
     offer?: string;
     num_images: number;
@@ -45,6 +58,7 @@ export const ExtensiveForm: React.FC<ExtensiveFormProps> = ({
     resolver: zodResolver(extensiveSchema),
     defaultValues: {
       niche: "home_insurance" as Niche,
+      custom_niche: "",
       target_audience: "",
       offer: "",
       num_images: 1,
@@ -55,6 +69,7 @@ export const ExtensiveForm: React.FC<ExtensiveFormProps> = ({
 
   const numImages = watch("num_images");
   const numStrategies = watch("num_strategies");
+  const selectedNiche = watch("niche");
 
   return (
     <Card variant="glass">
@@ -71,11 +86,29 @@ export const ExtensiveForm: React.FC<ExtensiveFormProps> = ({
             options={[
               { value: "home_insurance", label: "Home Insurance" },
               { value: "glp1", label: "GLP-1" },
+              { value: "others", label: "Others" },
             ]}
             error={errors.niche?.message}
             {...register("niche")}
           />
 
+          {selectedNiche === "others" && (
+            <div>
+              <label className="block text-sm font-semibold text-gray-700 mb-2">
+                Custom Niche <span className="text-red-500">*</span>
+              </label>
+              <input
+                type="text"
+                className="w-full px-4 py-3 rounded-xl border-2 border-gray-300 bg-white/80 backdrop-blur-sm focus:outline-none focus:ring-2 focus:ring-blue-500 focus:border-blue-500 transition-all duration-250"
+                placeholder="e.g., Pet Insurance, Solar Panels, Fitness Supplements"
+                {...register("custom_niche")}
+              />
+              {errors.custom_niche && (
+                <p className="text-red-500 text-xs mt-1">{errors.custom_niche.message}</p>
+              )}
+            </div>
+          )}
+
           <div>
             <label className="block text-sm font-semibold text-gray-700 mb-2">
               Target Audience

frontend/lib/api/endpoints.ts

@@ -72,6 +72,7 @@ export const generateMatrixAd = async (params: {
 // Extensive Endpoint
 export const generateExtensiveAd = async (params: {
   niche: Niche;
+  custom_niche?: string;
   target_audience?: string;
   offer?: string;
   num_images: number;
@@ -83,6 +84,7 @@ export const generateExtensiveAd = async (params: {
     niche: params.niche,
     num_images: params.num_images || 1,
     num_strategies: params.num_strategies || 1,
+    ...(params.custom_niche && { custom_niche: params.custom_niche }),
     ...(params.target_audience && { target_audience: params.target_audience }),
     ...(params.offer && { offer: params.offer }),
     ...(params.image_model && { image_model: params.image_model }),

frontend/types/api.ts

@@ -1,6 +1,6 @@
 // API Response Types based on FastAPI backend
 
-export type Niche = "home_insurance" | "glp1";
+export type Niche = "home_insurance" | "glp1" | "others";
 
 export interface ImageResult {
   filename?: string | null;

main.py

@@ -1104,8 +1104,12 @@ async def get_compatible_concepts(angle_key: str):
 
 class ExtensiveGenerateRequest(BaseModel):
     """Request for extensive generation."""
-    niche: Literal["home_insurance", "glp1"] = Field(
-        description="Target niche"
+    niche: str = Field(
+        description="Target niche: home_insurance, glp1, or others"
+    )
+    custom_niche: Optional[str] = Field(
+        default=None,
+        description="Custom niche name when 'others' is selected"
     )
     target_audience: Optional[str] = Field(
         default=None,
@@ -1151,10 +1155,24 @@ async def generate_extensive(
     5. Generates images for each strategy
     
     Returns all generated ads from all strategies (like batch generation).
+    
+    Supports custom niches via the 'others' option - when niche is 'others',
+    custom_niche field must be provided with the custom niche name.
     """
     try:
+        # Determine effective niche
+        if request.niche == "others":
+            if not request.custom_niche or not request.custom_niche.strip():
+                raise HTTPException(
+                    status_code=400,
+                    detail="custom_niche is required when niche is 'others'"
+                )
+            effective_niche = request.custom_niche.strip()
+        else:
+            effective_niche = request.niche
+        
         results = await ad_generator.generate_ad_extensive(
-            niche=request.niche,
+            niche=effective_niche,
             target_audience=request.target_audience,
             offer=request.offer,
             num_images=request.num_images,
@@ -1167,6 +1185,8 @@ async def generate_extensive(
             count=len(results),
             ads=results
         )
+    except HTTPException:
+        raise
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))