Upload 3 files

app.py

@@ -0,0 +1,198 @@
+import os
+import shutil
+import sqlite3
+import google.generativeai as genai
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.staticfiles import StaticFiles
+from linebot import LineBotApi, WebhookHandler
+from linebot.exceptions import InvalidSignatureError
+from linebot.models import *
+from pydub import AudioSegment
+from gradio_client import Client, handle_file
+import uvicorn
+
+# --- 1. 配置與環境變數 (雲端部署必備) ---
+app = FastAPI()
+if not os.path.exists("static"): os.makedirs("static")
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+# HF 部署時請在 Settings 設定這些 Secret
+LINE_CHANNEL_ACCESS_TOKEN = os.getenv('LINE_TOKEN', '您的TOKEN')
+LINE_CHANNEL_SECRET = os.getenv('LINE_SECRET', '您的SECRET')
+NGROK_URL = os.getenv('BASE_URL', 'https://您的HF網址.hf.space')
+GOOGLE_API_KEY = os.getenv('GEMINI_KEY', '您的GEMINI_KEY')
+
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-3.1-pro-preview')
+
+line_bot_api = LineBotApi(LINE_CHANNEL_ACCESS_TOKEN)
+handler = WebhookHandler(LINE_CHANNEL_SECRET)
+
+# --- 2. SQLite 資料庫初始化 ---
+DB_FILE = "user_memory.db"
+
+def init_db():
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS users 
+                 (user_id TEXT PRIMARY KEY, tribe TEXT, count INTEGER DEFAULT 0)''')
+    c.execute('''CREATE TABLE IF NOT EXISTS phrases 
+                 (id INTEGER PRIMARY KEY AUTOINCREMENT, user_id TEXT, native TEXT, zh TEXT)''')
+    conn.commit()
+    conn.close()
+
+init_db()
+
+# --- 3. 族語配置與 API 客戶端 ---
+TRIBE_CONFIG = {
+    "阿美": {"asr": "formosan_ami", "mt": "阿美"}, "泰雅": {"asr": "formosan_tay", "mt": "泰雅"},
+    "排灣": {"asr": "formosan_pwn", "mt": "排灣"}, "布農": {"asr": "formosan_bnn", "mt": "布農"},
+    "卑南": {"asr": "formosan_pyu", "mt": "卑南"}, "魯凱": {"asr": "formosan_dru", "mt": "魯凱"},
+    "鄒": {"asr": "formosan_tsu", "mt": "鄒"}, "賽夏": {"asr": "formosan_xsy", "mt": "賽夏"},
+    "雅美": {"asr": "formosan_tao", "mt": "雅美"}, "邵": {"asr": "formosan_ssf", "mt": "邵"},
+    "噶瑪蘭": {"asr": "formosan_ckv", "mt": "噶瑪蘭"}, "太魯閣": {"asr": "formosan_trv", "mt": "太魯閣"},
+    "撒奇萊雅": {"asr": "formosan_szy", "mt": "撒奇萊雅"}, "賽德克": {"asr": "formosan_sdq", "mt": "賽德克"},
+    "拉阿魯哇": {"asr": "formosan_sxr", "mt": "拉阿魯哇"}, "卡那卡那富": {"asr": "formosan_xnb", "mt": "卡那卡那富"}
+}
+
+asr_client = Client("https://ai-labs.ilrdf.org.tw/sapolita-kaldi/")
+tts_client = Client("https://ai-labs.ilrdf.org.tw/hnang-kari-ai-asi-sluhay/")
+mt_client  = Client("https://ai-labs.ilrdf.org.tw/kari-seejiq-tnpusu-ai-hmjil/")
+
+chat_sessions = {}
+
+def get_clean_value(res):
+    if isinstance(res, dict) and 'value' in res: return res['value']
+    if isinstance(res, list) and len(res) > 0: return res[0]
+    return res
+
+def get_ai_response(user_id, user_text, tribe_name):
+    if user_id not in chat_sessions:
+        chat_sessions[user_id] = model.start_chat(history=[])
+        chat_sessions[user_id].send_message(f"你現在是與我對話的{tribe_name}族朋友。請用中文聊天。規則：1.禁止教學。2.直接回答。3.限一短句。")
+    return chat_sessions[user_id].send_message(user_text).text
+
+def create_chat_card(tribe, user_text, ai_text_zh, ai_text_native):
+    return {
+        "type": "bubble",
+        "header": {
+            "type": "box", "layout": "vertical", "contents": [
+                {"type": "text", "text": f"💬 {tribe}語 AI 導師", "weight": "bold", "color": "#FFFFFF", "size": "sm"}
+            ], "backgroundColor": "#8B0000"
+        },
+        "body": {
+            "type": "box", "layout": "vertical", "spacing": "sm", "contents": [
+                {"type": "text", "text": f"你說：{user_text}", "size": "xs", "color": "#888888", "wrap": True},
+                {"type": "text", "text": ai_text_native, "weight": "bold", "size": "xl", "wrap": True, "color": "#000000"},
+                {"type": "text", "text": f"(翻譯：{ai_text_zh})", "size": "sm", "color": "#555555", "wrap": True}
+            ]
+        },
+        "footer": {
+            "type": "box", "layout": "vertical", "contents": [
+                {"type": "button", "action": {"type": "message", "label": "🔄 切換語別", "text": "選單"}, "style": "secondary", "height": "sm"}
+            ]
+        }
+    }
+
+def show_tribe_menu(event, start_index=0):
+    all_keys = list(TRIBE_CONFIG.keys())
+    if start_index == 0:
+        display_keys = all_keys[:10]
+        buttons = [QuickReplyButton(action=MessageAction(label=k, text=k)) for k in display_keys]
+        buttons.append(QuickReplyButton(action=MessageAction(label="更多族別", text="更多族別")))
+        msg = "請選擇練習語別："
+    else:
+        display_keys = all_keys[10:]
+        buttons = [QuickReplyButton(action=MessageAction(label=k, text=k)) for k in display_keys]
+        buttons.append(QuickReplyButton(action=MessageAction(label="上一頁", text="選單")))
+        msg = "請選擇其他族別："
+    line_bot_api.reply_message(event.reply_token, TextSendMessage(text=msg, quick_reply=QuickReply(items=buttons)))
+
+# --- 4. 路由與訊息處理 ---
+@app.post("/callback")
+async def callback(request: Request):
+    signature = request.headers.get('X-Line-Signature', '')
+    body = await request.body()
+    try: handler.handle(body.decode(), signature)
+    except InvalidSignatureError: raise HTTPException(status_code=400)
+    return 'OK'
+
+@handler.add(MessageEvent, message=TextMessage)
+def handle_text(event):
+    text = event.message.text.strip()
+    user_id = event.source.user_id
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+
+    if text in TRIBE_CONFIG:
+        c.execute("INSERT OR REPLACE INTO users (user_id, tribe, count) VALUES (?, ?, (SELECT count FROM users WHERE user_id=?))", (user_id, text, user_id))
+        conn.commit()
+        chat_sessions.pop(user_id, None)
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=f"✅ 已切換至【{text}語】模式！"))
+    elif text == "選單" or text == "🔄 切換語別":
+        show_tribe_menu(event, 0)
+    elif text == "更多族別":
+        show_tribe_menu(event, 10)
+    elif text in ["學習記錄", "查詢記錄", "學習紀錄"]:
+        c.execute("SELECT count FROM users WHERE user_id=?", (user_id,))
+        row = c.fetchone()
+        count = row[0] if row else 0
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=f"📈 【紀錄】累計對話：{count} 次"))
+    elif text in ["我的金句", "複習筆記"]:
+        c.execute("SELECT native, zh FROM phrases WHERE user_id=? ORDER BY id DESC LIMIT 5", (user_id,))
+        rows = c.fetchall()
+        msg = "📁 【最近金句】\n" + "\n".join([f"• {r[0]} ({r[1]})" for r in rows]) if rows else "📁 還沒有金句喔！"
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=msg))
+    else:
+        show_tribe_menu(event, 0)
+    conn.close()
+
+@handler.add(MessageEvent, message=AudioMessage)
+def handle_audio(event):
+    user_id = event.source.user_id
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("SELECT tribe FROM users WHERE user_id=?", (user_id,))
+    row = c.fetchone()
+    tribe = row[0] if row else None
+    
+    if not tribe:
+        show_tribe_menu(event, 0)
+        return
+
+    config = TRIBE_CONFIG[tribe]
+    msg_id = event.message.id
+    reply_wav = f"static/ai_{msg_id}.wav"
+    
+    try:
+        content = line_bot_api.get_message_content(msg_id)
+        with open(f"static/{msg_id}.m4a", "wb") as f: f.write(content.content)
+        audio = AudioSegment.from_file(f"static/{msg_id}.m4a")
+        audio.export(f"static/{msg_id}.wav", format="wav")
+
+        # ASR -> MT -> Gemini -> MT -> TTS
+        native_in = asr_client.predict(dialect_id=config["asr"], audio_data=handle_file(f"static/{msg_id}.wav"), api_name="/automatic_speech_recognition")
+        go_code = get_clean_value(mt_client.predict(ethnicity=config["mt"], api_name="/lambda"))
+        zh_in = mt_client.predict(text=native_in, src_lang=go_code, tgt_lang="zho_Hant", api_name="/translate")
+        ai_zh = get_ai_response(user_id, zh_in, tribe)
+        back_code = get_clean_value(mt_client.predict(ethnicity=config["mt"], api_name="/lambda_1"))
+        ai_native = get_clean_value(mt_client.predict(text=ai_zh, src_lang="zho_Hant", tgt_lang=back_code, api_name="/translate_1"))
+        speaker = get_clean_value(tts_client.predict(ethnicity=config["mt"], api_name="/lambda"))
+        temp_tts = tts_client.predict(ref=speaker, gen_text_input=ai_native, api_name="/default_speaker_tts")
+        shutil.move(temp_tts, reply_wav)
+
+        # SQL 存檔
+        c.execute("UPDATE users SET count = count + 1 WHERE user_id=?", (user_id,))
+        c.execute("INSERT INTO phrases (user_id, native, zh) VALUES (?, ?, ?)", (user_id, ai_native, ai_zh))
+        conn.commit()
+
+        line_bot_api.reply_message(event.reply_token, [
+            FlexSendMessage(alt_text="AI 對話", contents=create_chat_card(tribe, native_in, ai_zh, ai_native)),
+            AudioSendMessage(original_content_url=f"{NGROK_URL}/{reply_wav}", duration=len(audio))
+        ])
+    except Exception as e:
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=f"AI 稍後回來: {str(e)}"))
+    conn.close()
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)

packages.txt

@@ -0,0 +1 @@
+ffmpeg

requirements.txt

@@ -0,0 +1,7 @@
+fastapi
+uvicorn
+line-bot-sdk
+pydub
+gradio_client
+google-generativeai
+python-multipart