Update prompt_engine.py

prompt_engine.py

@@ -1,4 +1,6 @@
 import json
+import os
+import requests
 
 class PromptGenerator:
     def __init__(self):
@@ -113,11 +115,11 @@ class PromptGenerator:
         except Exception as e:
             return f"分析エラー: {str(e)}"
 
-    def generate_prompt(self, groq_client, params):
+    def generate_prompt(self, client, params):
         """プロンプト生成のメインメソッド"""
         try:
             # 1. まずQwenモデルでユーザー入力を分析
-            analysis = self.analyze_user_input(groq_client, params["input"])
+            analysis = self.analyze_user_input(client, params["input"])
             
             # 2. 分析結果を含めたユーザーメッセージの構築
             user_message = self._build_user_message(params, analysis)
@@ -136,9 +138,12 @@ class PromptGenerator:
             elif "qwen" in model:
                 # 創造性に特化したモデルはさらに高い値
                 temperature = 0.85
+            elif "o4-mini" in model:
+                # OpenAI GPT APIの場合
+                return self._generate_with_openai(params, user_message)
             
             # Groq APIでプロンプト生成
-            response = groq_client.chat.completions.create(
+            response = client.chat.completions.create(
                 model=model,  # 選択されたモデルを使用
                 messages=[
                     {"role": "system", "content": self.system_prompt},
@@ -154,6 +159,45 @@ class PromptGenerator:
         except Exception as e:
             return f"エラーが発生しました: {str(e)}"
     
+    def _generate_with_openai(self, params, user_message):
+        """OpenAI GPT APIを使用してプロンプトを生成"""
+        try:
+            # OpenAI APIキーを環境変数から取得
+            api_key = os.environ.get("OPENAI_API_KEY")
+            if not api_key:
+                return "OpenAI APIキーが設定されていません。"
+            
+            # OpenAI APIのエンドポイント
+            url = "https://api.openai.com/v1/chat/completions"
+            
+            # ヘッダー情報
+            headers = {
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {api_key}"
+            }
+            
+            # リクエストボディ
+            data = {
+                "model": "o4-mini-2025-04-16",
+                "max_tokens": 2048,
+                "temperature": 0.75,
+                "messages": [
+                    {"role": "system", "content": self.system_prompt},
+                    {"role": "user", "content": user_message}
+                ]
+            }
+            
+            # APIリクエスト送信
+            response = requests.post(url, headers=headers, json=data)
+            response.raise_for_status()
+            
+            # レスポンスの解析
+            result = response.json()
+            return result["choices"][0]["message"]["content"]
+            
+        except Exception as e:
+            return f"OpenAI APIエラー: {str(e)}"
+    
     def _build_user_message(self, params, analysis=None):
         """ユーザーメッセージの構築"""
         # 基本情報
@@ -184,7 +228,7 @@ class PromptGenerator:
 この分析結果を考慮して、以下の重要なガイドラインに従ってプロンプトを生成してください：
 
 1. プロンプトは必ず英語で生成すること（システムタグやパラメータ名も含め）
-2. ユーザーの元の入力内容（{params["input"]}）の本質を必ず維持すること
+2. ユーザーの元の入力内容（{params["input"]}）を最優先し、その意図と本質を100%維持すること
 3. 主体となる対象や重要な要素は絶対に変更しないこと
    例：「イルカ」を「シャチ」に変えるなど種類/種別を変更しない
    例：「猫」を「虎」に変えるなど、近い種でも誤変換しない
@@ -197,11 +241,9 @@ class PromptGenerator:
 7. 代替解釈の可能性がある場合は、ユーザーの意図に最も近いと思われる解釈を選択すること
 8. システムタグやパラメータ内の言語も必ず英語にすること（例：「图像:」→「image:」）
 
-プロンプトの質を高めるため、以下の要素を含めてください：
-- 明確な主体と背景の説明
-- 適切な芸術的表現方法と技術的詳細
-- 重み付けシステム(::1.2など)で重要な要素を強調
-- MidJourney/nijiJourney固有のパラメータを効果的に活用
+【重要】ユーザーの元の入力（{params["input"]}）を最も重視してください。分析結果はあくまで補助情報として使用し、
+元の入力内容の意図を変えたり、要素を追加/除外したりしないでください。
+プロンプトは必ずユーザーの入力を正確に英語に変換し、その上で適切なテクニックを適用してください。
 """
         
         # 基本設定