import gradio as gr
from transformers import pipeline

# AIの準備（これだけで動くように設定しました）
try:
    generator = pipeline("text-generation", model="CyberAgent/open-calm-small")
except:
    generator = None

def senior_ai(message, history):
    if generator is None:
        return "ごめん、機材（AI）の調子が悪いみたい。もう一度試してみて！"

    # 魔法の命令文（1行にまとめてエラーを完全回避！）
    m = "あなたは軽音部のライト先輩。明るいタメ口で、機材を『この子』と呼びます。トラブルには1. 2. と箇条書きで答えて。アンプの音量はローディに相談して。ギターのコーラスマイクはハウりやすいよ。"
    q = f"{m}\n後輩：{message}\nライト先輩："
    
    # 怖い言葉（ホラー）にならないための厳重な設定
    res = generator(q, max_length=150, do_sample=True, temperature=0.7, repetition_penalty=1.5, pad_token_id=0)
    
    reply = res[0]['generated_text'].split("ライト先輩：")[-1].strip()
    return reply if reply else "パッチケーブルの接触が悪いみたい。もう一回言って？"

# 画面を作る
demo = gr.ChatInterface(fn=senior_ai, title="🎸 軽音PAサポートAI ライト")

if __name__ == "__main__":
    demo.launch()