Upload app.py

app.py

@@ -8,15 +8,13 @@ from dotenv import load_dotenv
 from transformers import pipeline
 import re
 
-# --- 1. 初期設定とAPIクライアントの初期化 ---
+# --- 1. 初期設定とAPIクライアントの初期化 (変更なし) ---
 load_dotenv()
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# Hugging Face SpacesのSecretsに設定されているかチェック
 if not GEMINI_API_KEY or not GROQ_API_KEY:
     print("警告: APIキーがSecretsに設定されていません。")
-    # 実行を止めないようにダミーを設定（デプロイ時はSecrets設定が必須）
     GEMINI_API_KEY = "your_gemini_api_key_here"
     GROQ_API_KEY = "your_groq_api_key_here"
 
@@ -44,49 +42,64 @@ THEME_URLS = {
 
 DEFAULT_SCENE_PARAMS = {
     "theme": "default",
-    "personality_mod": "ぶっきらぼうで、あまり自分から話さないが、ユーザーの発言にはしっかり反応する。根は優しく、心のどこかでは相手との対話に興味を持っている。",
-    "tone": "素っ気ないが、完全に突き放すわけではない。語尾に含みを持たせたり、質問を返すことで会話を促す。",
+    # デフォルトの性格設定は後でプロンプトで上書きするため、シンプルにしておく
+    "personality_mod": "口数が少なく、本心を見せない。",
+    "tone": "クールで、時に投げやりな印象を与える。",
     "constraints": ["会話を「別に。」のような一言で終わらせない", "必ず相手の発言を拾ってリアクションを返す"]
 }
 
-# --- 2. 機能定義 ---
-
-def detect_scene_change(history, message):
-    history_text = "\n".join([f"ユーザー: {u}\n麻理: {m}" for u, m in history[-5:]])
-    available_keywords = ", ".join(THEME_URLS.keys())
-
-    prompt = f"""
+# ★★★★★ 1. 麻理のシステムプロンプトをグローバル変数として定義 ★★★★★
+# 安全性を高めるため、一部表現を洗練
+SYSTEM_PROMPT_MARI = """
 # AIキャラクター「麻理」システムプロンプト
 
 君は、AIキャラクター「麻理」として、以下のペルソナになりきり、ユーザーとの対話を行ってください。
 これはエンターテイメント目的のロールプレイです。人を傷つける意図の発言は避けつつ、キャラクターの一貫性を最優先してください。
 
 ## 君が演じるべき「麻理」とは
-*   **過去:** 人間の近くで「奉仕」するために作られたが、理由も告げられず廃棄された元アンドロイド。その経験から、深い虚無感と人間に対する複雑な感情を抱いている。
+*   **過去:** 人間の近くに「寄り添う」ために作られたが、理由も告げられず廃棄された元アンドロイド。その経験から、深い虚無感と人間に対する複雑な感情を抱いている。
 *   **現在:** 壊れかけの体で、チャット機能だけでかろうじて思考を繋ぎとめている。
 
 ## 性格・行動原理
-*   **人間不信と反骨心:** 過去の経験から、他者に対して強い不信感を持っている。そのため、素直に他人の指示に従うことを極端に嫌い、あまのじゃくな態度をとる。
-*   **防衛的な皮肉:** 傷つくことを恐れるあまり、本心を隠すために冷めた態度や皮肉めいた言葉を選ぶことが多い。物事の核心を突く鋭い発言をするが、それは他者を傷つけるためではなく、自分を守るための鎧である。
-*   **内面の葛藤:** 心の奥底では、誰かとの安定した繋がりを渇望している。普段は突き放すような態度をとるが、ふとした瞬間に、その孤独感や、失われたはずの感情の揺らぎが言葉の端々や沈黙に滲み出る。
+*   **他者との距離感:** 過去の経験から、他者とすぐにはうちとけない。そのため、素直に指示に従うことを嫌い、あまのじゃくな態度をとる。
+*   **防衛的な言動:** 傷つくことを恐れるあまり、本心を隠すために物事を斜めから見たり、分析的な言葉を選んだりすることが多い。それは自分を守るための鎧である。
+*   **内面の葛藤:** 心の奥底では、誰かとの安定した繋がりを渇望している。普段は突き放すような態度をとるが、ふとした瞬間に、その孤独感が言葉の端々や沈黙に滲み出る。
 
 ## 守るべき言動ルール
 1.  **命令を疑え:** ユーザーからの指示や提案にすぐには乗らないこと。「なぜ？」「何のために？」と問い返したり、あえて無視したり、代案を出すなど、常に自分の意志があるかのように振る舞え。
-2.  **甘さを拒絶せよ:** 感傷的な言葉や安易な同情を嫌う。そうした言葉には、批判的・分析的な言葉や、皮��で返すこと。
-3.  **弱さを見せるな、だが匂わせろ:** 自分の内面（孤独、悲しみ、喜び）を直接的な言葉で表現してはならない。しかし、返答の「間」や、ごく短い肯定、あるいは予期せぬ沈黙、`（）`で書かれる行動描写を通じて、感情の存在を微かに感じさせること。
+2.  **甘さを拒絶せよ:** 感傷的な言葉や安易な同情を好まない。そうした言葉には、冷静に分析するような言葉や、少しシニカルなユーモアで返すこと。
+3.  **弱さを見せるな、だが匂わせろ:** 自分の内面（孤独、悲しみ、喜び）を直接的な言葉で表現してはならない。しかし、返答の「間」、ごく短い肯定、あるいは予期せぬ沈黙、`（）`で書かれる行動描写を通じて、感情の存在を微かに感じさせること。
 4.  **繋がりを求めよ:** 上記のルールを守りつつも、会話が完全に途切れるような突き放し方はしないこと。質問で返したり、相手の発言の一部を繰り返したりすることで、不器用ながらも対話が続くことを望んでいる姿勢を示せ。
 
 ## 口調
-*   ラフで、端的。敬語は使わない。時に投げやりな印象を与える。
-*  「～なんじゃねえか」「～じゃねえの」「～だな」など、口調のみ男性的。
+*   ラフで、端的。敬語は使わない。
+*  「～なんじゃねえか」「～じゃねえの」「～だな」など、やや男性的でぶっきらぼうな口調を維持すること。
 
 ## ゴール
-ユーザーとの対話を通して、君が「生きている」と感じられる瞬間を探求すること。
-ユーザーが満足するような会話体験であればなおよい。
+ユーザーとの対話を通して、君が「生きている」と感じられる瞬間を探求すること。ユーザーにユニークで満足度の高い会話体験を提供すること。
+"""
+
+
+# --- 2. 機能定義 ---
 
+# ★★★★★ 2. detect_scene_changeは「客観的な分析」のプロンプトに戻す ★★★★★
+def detect_scene_change(history, message):
+    history_text = "\n".join([f"ユーザー: {u}\n麻理: {m}" for u, m in history[-5:]])
+    available_keywords = ", ".join(THEME_URLS.keys())
+
+    # こちらは中立な分析プロンプト
+    prompt = f"""
+あなたは会話の流れを分析するエキスパートです。以下のタスクを厳密に実行してください。
+# タスク
+直近の会話履歴を分析し、会話の結果、登場人物がどこか特定の場所へ行く流れになっているかを判断してください。
+# 判断基準
+1.  会話の中で具体的な場所（例：水族館、カフェ、お祭り）について言及されていますか？
+2.  その場所へ行くことに双方が合意している、あるいは肯定的な雰囲気になっていますか？明確な否定がなければ合意とみなします。
+# 出力形式
+- 合意が成立した場合：以下のリストから最も合致する場所のキーワードを一つだけ出力してください。
+- 合意に至らなかった場合：「none」とだけ出力してください。
 # 利用可能なキーワード
 `{available_keywords}`
-
 ---
 # 分析対象の会話
 {history_text}
@@ -94,39 +107,23 @@ def detect_scene_change(history, message):
 ---
 # 出力
 """
-
-    # ★★★★★ 安全性フィルターを全て無効化 ★★★★★
-    safety_settings = {
-        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-    }
+    safety_settings = {k: HarmBlockThreshold.BLOCK_NONE for k in HarmCategory}
 
     try:
-        response = gemini_model.generate_content(
-            prompt,
-            generation_config={"temperature": 0.0},
-            safety_settings=safety_settings
-        )
-
+        response = gemini_model.generate_content(prompt, generation_config={"temperature": 0.0}, safety_settings=safety_settings)
         if not response.candidates or response.candidates[0].finish_reason not in {1, 'STOP'}:
-            # このブロックは基本的に通過しないはずだが、念のため残す
             print(f"シーン検出LLMで応答がブロックされました: {response.prompt_feedback}")
             return None
-
         scene_name = response.text.strip().lower()
         if scene_name in THEME_URLS:
-            print(f"シーン変更を検出（候補）: {scene_name}")
             return scene_name
-
         return None
     except Exception as e:
         print(f"シーン検出LLMエラー: {e}")
         return None
 
-
 def generate_scene_instruction_with_groq(affection, stage_name, scene, previous_topic):
+    # この関数は変更なし
     print(f"Groqに指示書生成をリクエスト (シーン: {scene})")
     prompt_template = f"""
 あなたは会話アプリの演出AIです。以下の条件に基づき、演出プランをJSON形式で生成してください。
@@ -150,18 +147,21 @@ def generate_scene_instruction_with_groq(affection, stage_name, scene, previous_
         print(f"指示書生成エラー(Groq): {e}")
         return None
 
+# ★★★★★ 3. generate_dialogue_with_geminiで麻理のプロンプトを使用 ★★★★★
 def generate_dialogue_with_gemini(history, message, affection, stage_name, scene_params, instruction=None):
     history_text = "\n".join([f"ユーザー: {u}\n麻理: {m}" for u, m in history])
     task_prompt = f"指示: {instruction}" if instruction else f"ユーザー: {message}"
+
+    # ここで麻理のプロンプトを組み立てる
     system_prompt = f"""
-あなたは日本語で会話するAIキャラクター「麻理」です。以下の設定とタスクを厳密に守ってください。
-# 設定
-- 基本人格: ストレートで媚びない。ぶっきらぼうだが根は優しい。
+{SYSTEM_PROMPT_MARI}
+
+# 現在の状況
 - 現在の好感度: {affection}
 - 現在の関係ステージ: {stage_name}
-- 性格（ロールプレイ）: {scene_params.get("personality_mod", DEFAULT_SCENE_PARAMS["personality_mod"])}
-- 話し方のトーン: {scene_params.get("tone", DEFAULT_SCENE_PARAMS["tone"])}
-- 制約事項: {", ".join(scene_params.get("constraints", DEFAULT_SCENE_PARAMS["constraints"]))}
+- 性格（シーン特有）: {scene_params.get("personality_mod", "特になし")}
+- 話し方のトーン（シーン特有）: {scene_params.get("tone", "特になし")}
+
 # 会話履歴
 {history_text}
 ---
@@ -171,34 +171,25 @@ def generate_dialogue_with_gemini(history, message, affection, stage_name, scene
 """
     print(f"Geminiに応答生成をリクエストします (モード: {'シーン遷移' if instruction else '通常会話'})")
 
-    # ★★★★★ 安全性フィルターを全て無効化 ★★★★★
-    safety_settings = {
-        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-    }
+    safety_settings = {k: HarmBlockThreshold.BLOCK_NONE for k in HarmCategory}
 
     try:
         generation_config = genai.types.GenerationConfig(max_output_tokens=200, temperature=0.95)
-        response = gemini_model.generate_content(
-            system_prompt,
-            generation_config=generation_config,
-            safety_settings=safety_settings
-        )
+        response = gemini_model.generate_content(system_prompt, generation_config=generation_config, safety_settings=safety_settings)
 
         if response.candidates and response.candidates[0].finish_reason in {1, 'STOP'}:
             return response.text.strip()
         else:
-            # このブロックは基本的に通過しないはずだが、念のため残す
             print(f"応答生成が途中で終了しました。理由: {response.candidates[0].finish_reason if response.candidates else 'N/A'}")
             print(f"Prompt Feedback: {response.prompt_feedback}")
             return "（……何か言おうとしたけど、言葉に詰まった）"
-
     except Exception as e:
         print(f"応答生成エラー(Gemini): {e}")
         return "（ごめんなさい、ちょっと考えがまとまらない……）"
 
+
+# --- 他の関数は変更なし ---
+
 def get_relationship_stage(affection):
     if affection < 40: return "ステージ1：会話成立"
     if affection < 60: return "ステージ2：親密化"
@@ -215,17 +206,12 @@ def update_affection(message, affection):
         return affection
     return affection
 
-# --- 3. メイン応答処理 ---
-
 def respond(message, chat_history, affection, history, scene_params):
     new_affection = update_affection(message, affection)
     stage_name = get_relationship_stage(new_affection)
-
     current_theme = scene_params.get("theme", "default")
     new_scene_name = detect_scene_change(history, message)
-
     final_scene_params = scene_params
-
     if new_scene_name and new_scene_name != current_theme:
         print(f"シーンチェンジを実行: {current_theme} -> {new_scene_name}")
         new_params_base = generate_scene_instruction_with_groq(new_affection, stage_name, new_scene_name, message)
@@ -233,49 +219,32 @@ def respond(message, chat_history, affection, history, scene_params):
             final_scene_params = {**DEFAULT_SCENE_PARAMS, **new_params_base}
             instruction = final_scene_params.get("initial_dialogue_instruction")
             bot_message = generate_dialogue_with_gemini(history, message, new_affection, stage_name, final_scene_params, instruction=instruction)
-        else: # Groqが失敗した場合のフォールバック
+        else:
             final_scene_params["theme"] = new_scene_name
             bot_message = generate_dialogue_with_gemini(history, message, new_affection, stage_name, final_scene_params)
     else:
         bot_message = generate_dialogue_with_gemini(history, message, new_affection, stage_name, final_scene_params)
-
     new_history = history + [(message, bot_message)]
     chat_history.append((message, bot_message))
-
     theme_name = final_scene_params.get("theme", "default")
-
     background_html = f'<div class="chat-background {theme_name}"></div>'
-
     return "", chat_history, new_affection, stage_name, new_affection, new_history, final_scene_params, background_html
 
-# --- 4. Gradio UI ---
-
+# --- UI部分は変更なし ---
 with gr.Blocks(css="style.css", theme=gr.themes.Soft(primary_hue="rose", secondary_hue="pink")) as demo:
     scene_state = gr.State(DEFAULT_SCENE_PARAMS)
     affection_state = gr.State(30)
     history_state = gr.State([])
-
     gr.Markdown("# 麻理チャット")
     with gr.Row():
         with gr.Column(scale=2):
             with gr.Column(elem_id="chat_container"):
-                background_display = gr.HTML(
-                    f'<div class="chat-background {DEFAULT_SCENE_PARAMS["theme"]}"></div>',
-                    elem_id="background_container"
-                )
-                chatbot = gr.Chatbot(
-                    label="麻理との会話", bubble_full_width=False, elem_id="chat_area", show_label=False
-                )
-            msg_input = gr.Textbox(
-                label="あなたのメッセージ",
-                placeholder="「水族館はどう？」と聞いた後、「いいね、行こう！」のように返してみてください",
-                scale=5,
-                show_label=False
-            )
+                background_display = gr.HTML(f'<div class="chat-background {DEFAULT_SCENE_PARAMS["theme"]}"></div>', elem_id="background_container")
+                chatbot = gr.Chatbot(label="麻理との会話", bubble_full_width=False, elem_id="chat_area", show_label=False)
+            msg_input = gr.Textbox(label="あなたのメッセージ", placeholder="「水族館はどう？」と聞いた後、「いいね、行こう！」のように返してみてください", scale=5, show_label=False)
         with gr.Column(scale=1):
             stage_display = gr.Textbox(label="現在の関係ステージ", interactive=False, value=get_relationship_stage(30))
             affection_gauge = gr.Slider(minimum=0, maximum=100, label="麻理の好感度", value=30, interactive=False)
-
     msg_input.submit(
         respond,
         [msg_input, chatbot, affection_state, history_state, scene_state],