Update core/ai_client.py

core/ai_client.py

@@ -1,125 +1,12 @@
-import os, base64, json
-from typing import List, Dict, Any, Optional
+import os
+from openai import OpenAI
 
-try:
-    from openai import OpenAI
-except Exception:
-    OpenAI = None  # 起動は継続、UIだけでも出す
+VISION_MODEL = os.environ.get("OPENAI_VISION_MODEL", "gpt-4o-mini")
+TEXT_MODEL   = os.environ.get("OPENAI_TEXT_MODEL",   "gpt-4o-mini")
 
-OPENAI_MODEL_VISION = os.environ.get("OPENAI_VISION_MODEL", "gpt-4o-mini")
-OPENAI_MODEL_TEXT   = os.environ.get("OPENAI_TEXT_MODEL",   "gpt-4o-mini")
-
-SYSTEM_JSON = """あなたは有能な財務アナリストです。
-与えられた決算書（画像またはテキスト）から、次の厳密な JSON 構造のみを日本語の単位なし・半角数値で返してください。分からない項目は null。
-{
-  "company": {"name": null},
-  "period": {"start_date": null, "end_date": null},
-  "balance_sheet": {
-    "total_assets": null, "total_liabilities": null, "total_equity": null,
-    "current_assets": null, "fixed_assets": null,
-    "current_liabilities": null, "long_term_liabilities": null
-  },
-  "income_statement": {
-    "sales": null, "cost_of_sales": null, "gross_profit": null,
-    "operating_expenses": null, "operating_income": null,
-    "ordinary_income": null, "net_income": null
-  },
-  "cash_flows": {
-    "operating_cash_flow": null, "investing_cash_flow": null, "financing_cash_flow": null
-  }
-}
-"""
-
-def _b64(img: bytes) -> str:
-    import base64 as _b
-    return _b.b64encode(img).decode("utf-8")
-
-def _client() -> Optional[OpenAI]:
-    if OpenAI is None:
-        return None
+def get_client() -> OpenAI:
     key = os.environ.get("OPENAI_API_KEY")
     if not key:
-        return None
-    return OpenAI(api_key=key, timeout=30)  # proxies は渡さない
-
-def extract_financials(images: Optional[List[bytes]], text_blob: Optional[str], company_hint: str) -> Dict[str, Any]:
-    """
-    OpenAI が使えない場合は「空の既定JSON」を返す（UIは必ず表示される）
-    """
-    base: Dict[str, Any] = {
-        "company": {"name": company_hint or None},
-        "period": {"start_date": None, "end_date": None},
-        "balance_sheet": {
-            "total_assets": None, "total_liabilities": None, "total_equity": None,
-            "current_assets": None, "fixed_assets": None,
-            "current_liabilities": None, "long_term_liabilities": None
-        },
-        "income_statement": {
-            "sales": None, "cost_of_sales": None, "gross_profit": None,
-            "operating_expenses": None, "operating_income": None,
-            "ordinary_income": None, "net_income": None
-        },
-        "cash_flows": {
-            "operating_cash_flow": None, "investing_cash_flow": None, "financing_cash_flow": None
-        }
-    }
-
-    client = _client()
-    if client is None:
-        return base  # キー未設定 or ライブラリ不在でも落とさない
-
-    if images:
-        content = [{"type": "text", "text": SYSTEM_JSON}]
-        if company_hint:
-            content.append({"type": "text", "text": f"会社名の候補: {company_hint}"})
-        for im in images:
-            content.append({"type": "input_image", "image_url": f"data:image/png;base64,{_b64(im)}"})
-        resp = client.chat.completions.create(
-            model=OPENAI_MODEL_VISION,
-            messages=[
-                {"role": "system", "content": "返答は必ず有効な JSON オブジェクトのみ。説明を含めない。"},
-                {"role": "user", "content": content},
-            ],
-            response_format={"type": "json_object"},
-            temperature=0.1,
-        )
-        return json.loads(resp.choices[0].message.content or "{}") or base
-
-    else:
-        prompt = f"{SYSTEM_JSON}\n\n以下は決算書のテキストです。上記の JSON だけを返してください。\n\n{text_blob or ''}"
-        resp = client.chat.completions.create(
-            model=OPENAI_MODEL_TEXT,
-            messages=[
-                {"role": "system", "content": "返答は必ず有効な JSON オブジェクトのみ。"},
-                {"role": "user", "content": prompt},
-            ],
-            response_format={"type": "json_object"},
-            temperature=0.1,
-        )
-        return json.loads(resp.choices[0].message.content or "{}") or base
-
-def short_insight(fin: Dict[str, Any], score: Dict[str, Any]) -> str:
-    client = _client()
-    if client is None:
-        return "（OpenAI キー未設定のため AI 所見は省略）"
-    prompt = f"""次の財務データとスコア結果から、箇条書きで短く日本語でコメントしてください。
-- 良い点 3つ
-- 懸念点 3つ
-- 総評（100字以内）
-
-[財務データ]
-{json.dumps(fin, ensure_ascii=False)}
-
-[スコア]
-{json.dumps(score, ensure_ascii=False)}
-"""
-    try:
-        resp = client.chat.completions.create(
-            model=OPENAI_MODEL_TEXT,
-            messages=[{"role": "system", "content": "簡潔で公正な財務アナリスト。"},
-                      {"role": "user", "content": prompt}],
-            temperature=0.3,
-        )
-        return resp.choices[0].message.content or "（AI 所見なし）"
-    except Exception as e:
-        return f"AI所見の生成に失敗: {e}"
+        raise RuntimeError("OPENAI_API_KEY が未設定です。環境変数に設定してください。")
+    # proxies は渡さない（OpenAI 1.x では不正引数になり得る）
+    return OpenAI(api_key=key, timeout=30)