Update app.py

app.py

@@ -1,68 +1,42 @@
 import gradio as gr
+from huggingface_hub import InferenceClient
+import os
 
-def senior_ai(message, history):
-    m = message.lower()
-    
-    # 1. 挨拶・雑談系
-    if any(k in m for k in ["こんにちは", "お疲れ", "おはよ", "ライト"]):
-        return "お疲れ様！ライトだよ。今日も部室でリハ？機材の調子が悪かったら、この子（機材）たちの声を代弁するから何でも聞いてね！"
-    
-    # 2. ミキサー（MGP32X）の深い悩み
-    if "信号" in m or "メーター" in m:
-        if "来ない" in m or "振れない" in m:
-            return (
-                "ミキサーに信号が来ないか…よし、一緒にチェックしよう！\n"
-                "1. まず、マイクや楽器が『Gain（ゲイン）』つまみで適切に増幅されてるか確認して。\n"
-                "2. チャンネルの『ON』ボタンは押してる？赤く光ってないと、この子（ミキサー）は動いてくれないよ。\n"
-                "3. ファンタム電源（+48V）が必要なコンデンサーマイクじゃないかも確認してみて！"
-            )
-        return "MGP32Xのメーターがしっかり振れてれば、ミキサーの中までは音が届いてる証拠だよ！"
-
-    # 3. スピーカー・音が出ない（複雑な状況）
-    if "スピーカー" in m or "出ない" in m:
-        if "信号は来てる" in m or "ミキサーには来てる" in m:
-            return (
-                "ミキサーまでは来てるのにスピーカーから出ないパターンだね。よし、出口を調べよう！\n"
-                "1. ミキサーの『STEREO』フェーダー（一番右のやつ）は上がってる？\n"
-                "2. パワーアンプ（XLSとかCPX）の電源は入ってるかな？\n"
-                "3. アンプのボリュームつまみが最小になってないか確認して。最後に電源を入れるのがコツだよ！"
-            )
-        return (
-            "音が出ない時は、まずどこまで音が届いてるか切り分けよう！\n"
-            "1. ミキサーのランプは光ってる？\n"
-            "2. シールドの断線がないか、予備の『この子（シールド）』に替えてみて。\n"
-            "3. 楽器自体のボリュームが0になってないかも意外と盲点だよ！"
-        )
-
-    # 4. ハウリング（特定の場所への配慮）
-    if any(k in m for k in ["ハウ", "ピー", "キーン"]):
-        res = "あちゃー、耳が痛いね！\n1. スピーカーとマイクの距離を離してみて。\n2. 特にギターとベースのコーラスマイクはハウりやすいから、立ち位置を少しズラすのがコツだよ！"
-        if "コーラス" in m:
-            res += "\n3. コーラスマイクは特に感度が高いから、ミキサーのLowを少しカットすると落ち着くかも。"
-        return res
+# 【ここがポイント！】金庫（Secrets）に入れたトークンを自動で読み込みます
+token = os.getenv("HF_TOKEN")
+client = InferenceClient("google/gemma-2-9b-it", token=token)
 
-    # 5. 音量・ローディ相談（鉄則）
-    if any(k in m for k in ["音量", "大きく", "上げたい"]):
-        return "もっと爆音でやりたい気持ちはわかるけど、アンプの音量を勝手にいじると外音（客席の音）がグチャグチャになっちゃうんだ。まずは『ローディ』に相談して、全体のバランスを見てからにしようね！"
-
-    # 6. 褒めると照れる
-    if any(k in m for k in ["すごい", "頼り", "さすが", "ありがとう"]):
-        return "えっ、そんなに褒められたら照れるよぉ…！あはは、機材のことなら私に任せて。後輩のライブ、成功させたいもんね！"
-
-    # デフォルトの回答（魔法の命令文の「不明点を質問する」ルールを徹底）
-    return (
-        f"『{message}』だね！なるほど…ちょっと判断迷うな。もっと詳しく教えて！\n"
-        "1. 今、どの機材のランプが何色に光ってる？\n"
-        "2. 接続の順番を入り口から教えてくれるかな？\n"
-        "それがわかれば、ライト先輩がビシッと解決しちゃうよ！"
+def senior_ai(message, history):
+    # あなたの「魔法の命令文」をすべてAIの脳みそに叩き込みます
+    system_message = (
+        "あなたは『軽音PAサポートAI ライト』。一人称は『私』。明るくて頼れる軽音部の先輩として、タメ口で明るく話してね。"
+        "機材を『この子』と呼び、褒められたら照れること。"
+        "【解決ルール】1.必ず箇条書きでステップ順に提示。2.専門用語は初心者にわかりやすく解説。3.不明な点は『どこが怪しそう？』と質問。4.アンプ音量は『ローディ』に相談させる。"
+        "【機材】VE-5(エフェクター), MGP32X(ミキサー), XLS DriveCore 2(返し用アンプ), CPX600(モニター用アンプ)。"
+        "【知識】ギター・ベースのコーラスマイクは特にハウりやすい。ミキサーはボーカル、ドラム、トークバック等を調節可能。"
     )
 
-# 画面設定
+    messages = [{"role": "system", "content": system_message}]
+    for h in history:
+        messages.append({"role": "user", "content": h[0]})
+        messages.append({"role": "assistant", "content": h[1]})
+    messages.append({"role": "user", "content": message})
+
+    try:
+        response = ""
+        for msg in client.chat_completion(messages, max_tokens=800, stream=True, temperature=0.7):
+            token_str = msg.choices[0].delta.content
+            response += token_str
+        return response
+    except Exception as e:
+        return f"ごめん！機材（API）の接続がうまくいかないみたい。Settingsで『HF_TOKEN』が正しく設定されてるか確認してみて！ (Error: {e})"
+
+# 画面のデザイン
 demo = gr.ChatInterface(
-    fn=senior_ai, 
+    fn=senior_ai,
     title="🎸 軽音PAサポートAI ライト",
-    description="【性格】明るく頼れる先輩（タメ口）【機材知識】VE-5, MGP32X, XLS, CPX600等"
+    description="何でも聞いてね。",
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()