Update app.py

app.py

@@ -4,64 +4,18 @@ import shutil
 from pydub import AudioSegment
 from openai import OpenAI
 import gradio as gr
-from fastapi import FastAPI, File, UploadFile, Form, HTTPException
+from fastapi import HTTPException
 
 # ======================================================
 # 🔐 設定區
 # ======================================================
-PASSWORD = os.getenv("APP_PASSWORD", "chou")
+PASSWORD = os.getenv("APP_PASSWORD", "defaultpass")
 MAX_SIZE = 25 * 1024 * 1024
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 print("===== 🚀 啟動中 =====")
 print(f"APP_PASSWORD: {'✅ 已載入' if PASSWORD else '❌ 未載入'}")
-
-# ======================================================
-# ⚔️ 防暴力破解
-# ======================================================
-MAX_FAILED_IN_WINDOW = 10
-WINDOW_SECONDS = 24 * 3600
-LOCK_DURATION_SECONDS = 24 * 3600
-SHORT_BURST_LIMIT = 5
-SHORT_BURST_SECONDS = 60
-
-attempts = {}
-locked = {}
-
-def _now(): return int(time.time())
-def prune_old_attempts(sid):
-    cutoff = _now() - WINDOW_SECONDS
-    if sid in attempts:
-        attempts[sid] = [t for t in attempts[sid] if t >= cutoff]
-        if not attempts[sid]:
-            del attempts[sid]
-def check_lock(sid):
-    if sid in locked:
-        if _now() < locked[sid]:
-            remain = locked[sid] - _now()
-            return True, f"🔒 已被鎖定，請 {remain // 60} 分鐘後再試。"
-        else:
-            locked.pop(sid, None)
-            attempts.pop(sid, None)
-    prune_old_attempts(sid)
-    cnt = len(attempts.get(sid, []))
-    if cnt >= MAX_FAILED_IN_WINDOW:
-        locked[sid] = _now() + LOCK_DURATION_SECONDS
-        return True, f"🔒 嘗試過多，已鎖定 24 小時。"
-    return False, ""
-def record_failed_attempt(sid):
-    now = _now()
-    attempts.setdefault(sid, []).append(now)
-    prune_old_attempts(sid)
-    recent_cutoff = now - SHORT_BURST_SECONDS
-    recent = [t for t in attempts[sid] if t >= recent_cutoff]
-    if len(recent) >= SHORT_BURST_LIMIT:
-        locked[sid] = now + 300
-        return len(attempts[sid]), "⚠️ 多次快速嘗試，暫時鎖定5分鐘。"
-    return len(attempts[sid]), ""
-def clear_attempts(sid):
-    attempts.pop(sid, None)
-    locked.pop(sid, None)
+print(f"目前密碼內容：{PASSWORD}")
 
 # ======================================================
 # 🎧 音訊轉錄核心
@@ -82,23 +36,24 @@ def split_audio_if_needed(path):
         files.append(fn)
     return files
 
-def transcribe_core(path, model):
-    if path.lower().endswith(".mp4"):
+def transcribe_core(path, model="whisper-1"):
+    if path and path.lower().endswith(".mp4"):
         fixed_path = path[:-4] + ".m4a"
         try:
             shutil.copy(path, fixed_path)
             path = fixed_path
             print("🔧 已自動修正 mp4 → m4a")
         except Exception as e:
-            print(f"⚠️ mp4→m4a 複製失敗：{e}")
+            print(f"⚠️ mp4→m4a 轉檔失敗：{e}")
 
-    # 轉錄
     chunks = split_audio_if_needed(path)
     raw_parts = []
     for f in chunks:
         with open(f, "rb") as af:
             res = client.audio.transcriptions.create(
-                model=model, file=af, response_format="text"
+                model=model,
+                file=af,
+                response_format="text"
             )
             raw_parts.append(res)
     full_raw = "\n".join(raw_parts)
@@ -106,12 +61,13 @@ def transcribe_core(path, model):
     # 簡轉繁
     conv_prompt = (
         "請將以下內容完整轉換為「繁體中文（台灣用語）」：\n"
-        "規則：1) 僅做簡→繁字形轉換；2) 不意譯；3) 不添加文字。\n-----\n" + full_raw
+        "規則：1) 僅做簡→繁字形轉換；2) 不要意譯或改寫；3) 不要添加任何前後綴。\n"
+        "-----\n" + full_raw
     )
     trad_resp = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "system", "content": "你是嚴格的繁體中文轉換器。"},
+            {"role": "system", "content": "你是嚴格的繁體中文轉換器，只進行字形轉換。"},
             {"role": "user", "content": conv_prompt}
         ],
         temperature=0.0,
@@ -120,80 +76,45 @@ def transcribe_core(path, model):
 
     # 摘要
     sum_prompt = (
-        "請用台灣繁體中文撰寫摘要，若內容需要時可條列重點（不固定數量），"
-        "重點是讓人快速理解原文，無需贅述：\n" + full_trad
+        "請用台灣繁體中文撰寫摘要。若內容資訊多，可條列出重點；若內容簡短，請用一句話概述即可。\n\n"
+        + full_trad
     )
     sum_resp = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "system", "content": "你是台灣繁體中文摘要助手。"},
+            {"role": "system", "content": "你是一位精準且嚴格使用台灣繁體中文的摘要助手。"},
             {"role": "user", "content": sum_prompt}
         ],
         temperature=0.2,
     )
     summ = sum_resp.choices[0].message.content.strip()
-
     return full_trad, summ
 
-# ======================================================
-# 🌐 FastAPI API（供 iPhone 捷徑使用）
-# ======================================================
-app = FastAPI(title="LINE Transcription API")
-
-@app.post("/api/transcribe")
-async def api_transcribe(file: UploadFile = File(...), token: str = Form(default=None)):
-    """供捷徑上傳音訊並取得 JSON"""
-    if token != PASSWORD:
-        raise HTTPException(status_code=403, detail="Invalid token")
-
-    temp = file.filename
-    with open(temp, "wb") as f:
-        f.write(await file.read())
-
-    text, summary = transcribe_core(temp, "whisper-1")
-    os.remove(temp)
-    return {"text": text, "summary": summary}
-
-@app.get("/health")
-def health():
-    return {"status": "ok", "time": int(time.time())}
-
 # ======================================================
 # 💬 Gradio 介面
 # ======================================================
-def transcribe_with_password(session_id, password, file_input, model_choice):
+def transcribe_with_password(password, file):
     if password.strip() != PASSWORD:
-        return "❌ 密碼錯誤", "", ""
-    path = file_input.name if file_input else None
-    if not path:
-        return "⚠️ 找不到檔案", "", ""
-    full, summ = transcribe_core(path, model_choice)
-    return "✅ 完成", full, summ
+        raise HTTPException(status_code=403, detail="密碼錯誤 ❌")
+    if not file:
+        return "⚠️ 未選擇檔案", "", ""
+    text, summary = transcribe_core(file.name)
+    return "✅ 完成", text, summary
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🎧 LINE 語音轉錄與摘要工具（支援 iPhone LINE .mp4）")
-
-    with gr.Row():
-        password_input = gr.Textbox(label="輸入密碼", type="password", placeholder="請輸入密碼")
-        model_choice = gr.Dropdown(["whisper-1", "gpt-4o-mini-transcribe"], value="whisper-1", label="選擇模型")
-
-    file_input = gr.File(label="上傳音訊（.m4a / .mp4 / .aac / .wav）", file_count="single")
-    transcribe_btn = gr.Button("開始轉錄 🚀")
-    status_box = gr.Textbox(label="狀態")
-    transcript_box = gr.Textbox(label="完整轉錄文字", lines=10)
-    summary_box = gr.Textbox(label="摘要結果", lines=10)
-
-    transcribe_btn.click(
-        transcribe_with_password,
-        [gr.State(value="local"), password_input, file_input, model_choice],
-        [status_box, transcript_box, summary_box]
-    )
+    gr.Markdown("## 🎧 LINE 語音轉錄與摘要（支援 .m4a / .mp4）")
+    pw = gr.Textbox(label="輸入密碼", type="password")
+    f = gr.File(label="上傳音訊檔")
+    run = gr.Button("開始轉錄 🚀")
+    s = gr.Textbox(label="狀態", interactive=False)
+    t = gr.Textbox(label="轉錄結果", lines=10)
+    su = gr.Textbox(label="AI 摘要", lines=8)
+    run.click(transcribe_with_password, [pw, f], [s, t, su])
 
 # ======================================================
-# 🚀 啟動（Hugging Face）
+# 🚀 啟動
 # ======================================================
-gr.mount_gradio_app(app, demo, path="/")
-
 if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)
+else:
+    demo.launch()