Upload app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import google.generativeai as genai
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
 from groq import Groq
 import os
 import json
@@ -14,14 +15,13 @@ 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"
 
 genai.configure(api_key=GEMINI_API_KEY)
-gemini_model = genai.GenerativeModel('gemini-2.5-flash')
+gemini_model = genai.GenerativeModel('gemini-1.5-flash')
 groq_client = Groq(api_key=GROQ_API_KEY)
 
 print("日本語感情分析モデルをロード中...")
@@ -72,6 +72,10 @@ def detect_scene_change(history, message):
 """
     try:
         response = gemini_model.generate_content(prompt, generation_config={"temperature": 0.0})
+        # ★ 安全性チェックを追加
+        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 != "none" and re.match(r'^[a-z0-9_]+$', scene_name):
             return scene_name
@@ -104,6 +108,7 @@ def generate_scene_instruction_with_groq(affection, stage_name, scene, previous_
         print(f"指示書生成エラー(Groq): {e}")
         return None
 
+# ★★★★★ ここが重要な修正点 ★★★★★
 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}"
@@ -124,10 +129,32 @@ 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_MEDIUM_AND_ABOVE,
+        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+    }
+    
     try:
         generation_config = genai.types.GenerationConfig(max_output_tokens=200, temperature=0.95)
-        response = gemini_model.generate_content(system_prompt, generation_config=generation_config)
-        return response.text.strip()
+        response = gemini_model.generate_content(
+            system_prompt,
+            generation_config=generation_config,
+            safety_settings=safety_settings  # ★ 安全設定を適用
+        )
+
+        # ★ 応答が正常に生成されたかを確認してから .text にアクセスする
+        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 "（ごめんなさい、ちょっと考えがまとまらない……）"
@@ -175,7 +202,6 @@ def respond(message, chat_history, affection, history, scene_params):
     
     theme_name = final_scene_params.get("theme", "default")
     
-    # 背景レイヤーのHTMLコンテンツをクラス名付きで生成
     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
@@ -190,18 +216,14 @@ with gr.Blocks(css="style.css", theme=gr.themes.Soft(primary_hue="rose", seconda
     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="「水族館はどう？」と聞いた後、「いいね、行こう！」のように返してみてください",
@@ -212,13 +234,11 @@ with gr.Blocks(css="style.css", theme=gr.themes.Soft(primary_hue="rose", seconda
             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)
 
-    # 応答関数とUIコンポーネントの接続を更新
     msg_input.submit(
         respond,
         [msg_input, chatbot, affection_state, history_state, scene_state],
         [msg_input, chatbot, affection_gauge, stage_display, affection_state, history_state, scene_state, background_display]
     )
-    # ページロード時に初期ステージを表示
     demo.load(lambda affection: get_relationship_stage(affection), affection_state, stage_display)
 
 if __name__ == "__main__":