Update app/openai_client.py

app/openai_client.py

@@ -1,9 +1,60 @@
 from __future__ import annotations
-import os, json
+import os
+import json
 import httpx
 
 OPENAI_BASE = os.environ.get("OPENAI_BASE", "https://api.openai.com/v1")
 OPENAI_MODEL = os.environ.get("OPENAI_MODEL", "gpt-4o-mini")
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 
-_headers =
+HEADERS = {
+    "Authorization": f"Bearer {OPENAI_API_KEY}" if OPENAI_API_KEY else "",
+    "Content-Type": "application/json",
+}
+
+
+async def openai_chat(messages, temperature: float = 0.7, max_tokens: int = 300) -> str:
+    """
+    非同期：通常のチャット生成（UIの広告生成で使用）
+    """
+    if not OPENAI_API_KEY:
+        raise RuntimeError("OPENAI_API_KEY is not set in environment variables.")
+
+    url = f"{OPENAI_BASE}/chat/completions"
+    payload = {
+        "model": OPENAI_MODEL,
+        "messages": messages,
+        "temperature": float(temperature),
+        "max_tokens": int(max_tokens),
+    }
+    async with httpx.AsyncClient(timeout=60) as client:
+        r = await client.post(url, headers=HEADERS, json=payload)
+        r.raise_for_status()
+        data = r.json()
+        return data["choices"][0]["message"]["content"].strip()
+
+
+def openai_judge(system: str, user: str) -> dict:
+    """
+    同期：審査用（JSONフォーマットで判定/修正案を返す）
+    """
+    if not OPENAI_API_KEY:
+        raise RuntimeError("OPENAI_API_KEY is not set in environment variables.")
+
+    url = f"{OPENAI_BASE}/chat/completions"
+    payload = {
+        "model": OPENAI_MODEL,
+        "messages": [
+            {"role": "system", "content": system},
+            {"role": "user", "content": user},
+        ],
+        "response_format": {"type": "json_object"},
+        "temperature": 0.0,
+        "max_tokens": 400,
+    }
+    with httpx.Client(timeout=60) as client:
+        r = client.post(url, headers=HEADERS, json=payload)
+        r.raise_for_status()
+        data = r.json()
+        content = data["choices"][0]["message"]["content"].strip()
+        return json.loads(content)