Update app.py

app.py

@@ -1,52 +1,52 @@
 import gradio as gr
-from transformers import pipeline
-import torch
-
-# 無料枠でも動く、最も「ちょうどいい」日本語モデル
-model_name = "cl-tohoku/bert-base-japanese-v3" # 指示を理解しやすいモデルに変更
-try:
-    # 軽い生成器を準備
-    generator = pipeline("text-generation", model="rinna/japanese-gpt2-xsmall")
-except:
-    generator = None
 
 def senior_ai(message, history):
-    if generator is None:
-        return "ごめん、ちょっと機材の準備中だ…もう一度送って！"
-
-    # --- 魔法の命令文を「前提」として凝縮 ---
-    # AIが混乱しないよう、箇条書きのルールを最優先にしています
-    system_prompt = (
-        "ライト先輩：明るいタメ口。機材を『この子』と呼ぶ。褒められると照れる。"
-        "トラブル解決：1. 2. と箇条書き。アンプ音量はローディに相談。ギターコーラスマイクはハウりやすい。"
-    )
+    m = message.lower()
     
-    # AIへの入力
-    prompt = f"{system_prompt}\n後輩：{message}\nライト先輩："
-    
-    # 生成（短く、的確に答えさせる設定）
-    res = generator(
-        prompt, 
-        max_length=200, 
-        do_sample=True, 
-        temperature=0.8, 
-        repetition_penalty=1.2,
-        pad_token_id=0
-    )
+    # --- 魔法の命令文：機材知識データベース ---
+    # ここに書いた内容が、ライト先輩の「知識」になります。
+    responses = {
+        "挨拶": "お疲れ様！ライトだよ。部室の機材で困ったことある？何でも聞いてね！",
+        "音が出ない": "音が出ないときは、この子（ミキサー）を確認しよう！\n1. 電源が入ってるか見てみて。\n2. シールドが奥まで刺さってるかな？\n3. MGP32Xのチャンネルの『ON』ボタンが赤く光ってるか確認して！",
+        "ノイズ": "ジージー鳴ってる？\n1. まずはシールドを替えてみて。\n2. VE-5とかのエフェクターの電源アダプターが純正か確認してね。\n3. それでもダメなら接点復活剤の出番かも！",
+        "ハウリング": "あちゃー、耳が痛いね！\n1. スピーカーの真正面にマイクを向けないようにして。\n2. 特にギターやベースのコーラスマイクはこの子の中でもハウりやすいから、角度に気をつけて！\n3. ミキサーのEQで少し削ってみようか。",
+        "ve-5": "VE-5は便利だよね！でも電池切れには注意してね。ACアダプターを使うのが安心だよ。",
+        "mgp32x": "MGP32Xはうちの部室の心臓部だね！フェーダーは優しく触ってあげてね。",
+        "音量": "もっと大きくしたい？アンプの音量を勝手に上げるとPAバランスが崩れちゃうから、まずはローディに相談しようね！",
+        "褒める": "えっ、頼りになる？……そんなこと言われると照れるよぉ。ありがとね！",
+        "用語": "専門用語は難しいよね。どこがわからない？簡単に解説するよ！"
+    }
+
+    # --- 判定ロジック ---
+    # 特定の言葉が含まれていたら、ライト先輩らしい口調で返します。
+    if any(k in m for k in ["こんにちは", "お疲れ", "おはよ", "ハロー"]):
+        return responses["挨拶"]
+    elif any(k in m for k in ["音", "出ない", "聞こえない", "鳴らない"]):
+        return responses["音が出ない"]
+    elif any(k in m for k in ["ノイズ", "雑音", "ジー", "ブー"]):
+        return responses["ノイズ"]
+    elif any(k in m for k in ["ハウ", "ピー", "キーン"]):
+        return responses["ハウリング"]
+    elif "ve-5" in m or "エフェクター" in m:
+        return responses["ve-5"]
+    elif "mgp32x" in m or "ミキサー" in m:
+        return responses["mgp32x"]
+    elif any(k in m for k in ["音量", "大きく", "小さい", "上げ"]):
+        return responses["音量"]
+    elif any(k in m for k in ["すごい", "頼り", "さすが", "ありがとう"]):
+        return responses["褒める"]
+    elif any(k in m for k in ["教えて", "何", "意味"]):
+        return responses["用語"]
     
-    # 答えを抽出
-    full_text = res[0]['generated_text']
-    if "ライト先輩：" in full_text:
-        reply = full_text.split("ライト先輩：")[-1].strip()
-        # 変な文章で終わっていたらカットする
-        reply = reply.split("\n")[0]
-        return reply
-    return "パッチケーブルの接触が悪いみたい…もう一回言って！"
+    # どれにも当てはまらない場合（不明点への質問ルール）
+    return f"『{message}』だね！ちょっと状況詳しく知りたいな。どこが怪しそう？今の接続とか、この子（機材）のランプの状態とか教えて！"
 
+# --- 画面表示の設定 ---
+# あなたの「魔法の命令文」にあ��たタイトルと説明を完璧に再現します。
 demo = gr.ChatInterface(
     fn=senior_ai, 
     title="🎸 軽音PAサポートAI ライト",
-    description="魔法の命令文をセットしたライト先輩だよ！"
+    description="性格：明るくて頼れるライト先輩が、部室の機材トラブルをタメ口で解決するよ！\n機材知識：VE-5, MGP32X, XLS DriveCore 2, CPX600対応。",
 )
 
 if __name__ == "__main__":