MVP0

backend_django/adsassistant_backend/api/services.py

@@ -0,0 +1,77 @@
+import requests
+
+
+def _raise_with_detail(r: requests.Response):
+    """Raise an HTTPError but include upstream error body for easier debugging."""
+    try:
+        detail = r.json()
+    except Exception:
+        detail = r.text
+    http_error_msg = f"{r.status_code} {r.url}: {detail}"
+    raise requests.HTTPError(http_error_msg, response=r)
+from django.conf import settings
+from django.db.models import Sum
+from .models import Test, ResultEntry, TextVariant
+
+def agents_generate_texts(payload: dict) -> dict:
+    url = f"{settings.AGENTS_SERVICE_URL}/api/v1/texts/generate"
+    r = requests.post(url, json=payload, timeout=180)
+    
+    if r.status_code >= 400:
+        _raise_with_detail(r)
+
+    return r.json()
+
+def agents_analyze(rows: list[dict], objective: str, policy: dict | None = None) -> dict:
+    url = f"{settings.AGENTS_SERVICE_URL}/api/v1/tests/analyze"
+    payload = {"rows": rows, "objective": objective}
+    if policy:
+        payload["policy"] = policy
+    r = requests.post(url, json=payload, timeout=90)
+    
+    if r.status_code >= 400:
+        _raise_with_detail(r)
+
+    return r.json()
+
+def aggregate_test_rows(test: Test) -> list[dict]:
+    qs = (ResultEntry.objects.filter(test=test).values("variant_id").annotate(
+        impressions=Sum("impressions"), clicks=Sum("clicks"), conversions=Sum("conversions"), leads=Sum("leads"), spend=Sum("spend")
+    ))
+    rows=[]
+    for row in qs:
+        v = TextVariant.objects.get(id=row["variant_id"])
+        rows.append({
+            "variant_id": v.id, "format": v.format,
+            "impressions": int(row["impressions"] or 0),
+            "clicks": int(row["clicks"] or 0),
+            "conversions": int(row["conversions"] or 0),
+            "leads": int(row["leads"] or 0),
+            "spend": float(row["spend"] or 0.0),
+        })
+    return rows
+
+
+def aggregate_test_rows_by_segment(test: Test) -> list[dict]:
+    """Aggregate stats per (segment, variant)."""
+    qs = (ResultEntry.objects.filter(test=test)
+          .values("segment_id", "variant_id")
+          .annotate(impressions=Sum("impressions"), clicks=Sum("clicks"),
+                    conversions=Sum("conversions"), leads=Sum("leads"), spend=Sum("spend")))
+    rows = []
+    # preload formats
+    fmt_map = {tv.id: tv.format for tv in TextVariant.objects.filter(brief=test.brief)}
+    seg_map = {s.id: s.name for s in test.segments.all()}
+    for r in qs:
+        rows.append({
+            "variant_id": r["variant_id"],
+            "format": fmt_map.get(r["variant_id"]),
+            "segment_id": r["segment_id"],
+            "segment_name": seg_map.get(r["segment_id"]),
+            "impressions": int(r["impressions"] or 0),
+            "clicks": int(r["clicks"] or 0),
+            "conversions": int(r["conversions"] or 0),
+            "leads": int(r["leads"] or 0),
+            "spend": float(r["spend"] or 0.0),
+        })
+    return rows