Update backend/main.py

backend/main.py

@@ -33,10 +33,15 @@ class ResearchRequest(BaseModel):
     method: Literal["gpt", "claude"]
 
 
-class ResearchResponse(BaseModel):
+class AudienceResult(BaseModel):
+    target_audience: str
     output: list[ImageAdEssentials]
 
 
+class ResearchResponse(BaseModel):
+    results: list[AudienceResult]
+
+
 # --- Endpoints ---
 
 @app.get("/api/target-audiences")
@@ -48,25 +53,27 @@ def get_target_audiences():
 @app.post("/api/research", response_model=ResearchResponse)
 def run_research(req: ResearchRequest):
     """
-    Run the researcher using the selected method (GPT or Claude).
-    Returns a list of psychology triggers with angles and concepts.
+    Run the researcher for each target audience separately.
+    Returns grouped results — one set of triggers per audience.
     """
     try:
-        # Join multiple audiences into a comma-separated string for the prompt
-        audience_str = ", ".join(req.target_audience)
-
-        if req.method == "gpt":
-            result = researcher_gpt(
-                audience_str, req.product_category, req.product_description
-            )
-        elif req.method == "claude":
-            result = researcher_claude(
-                audience_str, req.product_category, req.product_description
-            )
-        else:
-            raise HTTPException(status_code=400, detail="Invalid method. Use 'gpt' or 'claude'.")
-
-        return ResearchResponse(output=result)
+        all_results = []
+
+        for audience in req.target_audience:
+            if req.method == "gpt":
+                result = researcher_gpt(
+                    audience, req.product_category, req.product_description
+                )
+            elif req.method == "claude":
+                result = researcher_claude(
+                    audience, req.product_category, req.product_description
+                )
+            else:
+                raise HTTPException(status_code=400, detail="Invalid method. Use 'gpt' or 'claude'.")
+
+            all_results.append(AudienceResult(target_audience=audience, output=result))
+
+        return ResearchResponse(results=all_results)
 
     except ValueError as e:
         raise HTTPException(status_code=500, detail=str(e))