Update app.py

app.py

@@ -1,11 +1,11 @@
 import os
 import time
 import shutil
+import threading
 from pydub import AudioSegment
 from openai import OpenAI
 import gradio as gr
 from fastapi import FastAPI, UploadFile, File, Form
-import threading
 import uvicorn
 
 # ======================================================
@@ -27,67 +27,65 @@ def split_audio_if_needed(path):
     if size <= MAX_SIZE:
         return [path]
     audio = AudioSegment.from_file(path)
-    num = int(size / MAX_SIZE) + 1
-    chunk_ms = len(audio) / num
-    files = []
-    for i in range(num):
-        start, end = int(i * chunk_ms), int((i + 1) * chunk_ms)
-        chunk = audio[start:end]
+    n = int(size / MAX_SIZE) + 1
+    chunk_ms = len(audio) / n
+    parts = []
+    for i in range(n):
         fn = f"chunk_{i+1}.wav"
-        chunk.export(fn, format="wav")
-        files.append(fn)
-    return files
+        audio[int(i * chunk_ms):int((i + 1) * chunk_ms)].export(fn, format="wav")
+        parts.append(fn)
+    return parts
 
 def transcribe_core(path, model="whisper-1"):
-    if path and path.lower().endswith(".mp4"):
-        fixed_path = path[:-4] + ".m4a"
+    if path.lower().endswith(".mp4"):
+        fixed = path[:-4] + ".m4a"
         try:
-            shutil.copy(path, fixed_path)
-            path = fixed_path
+            shutil.copy(path, fixed)
+            path = fixed
             print("🔧 已自動修正 mp4 → m4a")
         except Exception as e:
             print(f"⚠️ mp4→m4a 轉檔失敗：{e}")
 
+    # Whisper 轉錄
     chunks = split_audio_if_needed(path)
-    raw_parts = []
+    txts = []
     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)
+            txts.append(res)
+    raw = "\n".join(txts)
 
+    # 簡→繁
     conv_prompt = (
         "請將以下內容完整轉換為「繁體中文（台灣用語）」：\n"
-        "規則：1) 僅做簡→繁字形轉換；2) 不要意譯或改寫；3) 不要添加任何前後綴。\n-----\n" + full_raw
+        "規則：1) 僅做簡→繁字形轉換；2) 不要意譯或改寫；3) 不要添加任何前後綴。\n-----\n" + raw
     )
-    trad_resp = client.chat.completions.create(
+    trad = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "system", "content": "你是嚴格的繁體中文轉換器。"},
             {"role": "user", "content": conv_prompt}
         ],
         temperature=0.0,
-    )
-    full_trad = trad_resp.choices[0].message.content.strip()
+    ).choices[0].message.content.strip()
 
+    # 摘要
     sum_prompt = (
-        "請用台灣繁體中文撰寫摘要。若內容資訊多，可條列出重點；若內容簡短，請用一句話概述即可。\n\n"
-        + full_trad
+        "請用台灣繁體中文撰寫摘要。若內容資訊多，可條列出重點；"
+        "若內容簡短，請用一句話概述即可。\n\n" + trad
     )
-    sum_resp = client.chat.completions.create(
+    summ = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "system", "content": "你是一位精準且嚴格使用台灣繁體中文的摘要助手。"},
             {"role": "user", "content": sum_prompt}
         ],
         temperature=0.2,
-    )
-    summ = sum_resp.choices[0].message.content.strip()
-    return full_trad, summ
+    ).choices[0].message.content.strip()
+
+    return trad, summ
 
 # ======================================================
 # 💬 Gradio 介面
@@ -111,7 +109,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     run.click(transcribe_with_password, [pw, f], [s, t, su])
 
 # ======================================================
-# 🌐 FastAPI for iPhone 捷徑
+# 🌐 FastAPI for 捷徑
 # ======================================================
 api_app = FastAPI(title="LINE Transcription API")
 
@@ -126,25 +124,18 @@ async def api_transcribe(file: UploadFile = File(...), token: str = Form(...)):
     os.remove(temp)
     return {"text": text, "summary": summary}
 
-def run_api():
-    uvicorn.run(api_app, host="0.0.0.0", port=7861)
+@api_app.get("/api/health")
+def health():
+    return {"status": "ok", "time": int(time.time())}
 
 # ======================================================
-# 🚀 啟動（同時啟 Gradio + FastAPI）
+# 🚀 啟動設定（Gradio 為主入口，FastAPI 背景執行）
 # ======================================================
-
 def run_api():
     uvicorn.run(api_app, host="0.0.0.0", port=7861, log_level="info")
 
-# --- 關鍵修改區 ---
-import threading
-
-# 1️⃣ 讓 FastAPI 在背景執行
 threading.Thread(target=run_api, daemon=True).start()
+app = demo  # 讓 Hugging Face 正確掛載主頁
 
-# 2️⃣ 讓 Hugging Face Spaces 主入口指向 Gradio
-app = demo  # �� 這一行是關鍵，讓 / 顯示你的網頁 UI
-
-# 3️⃣ 如果本地測試要同時啟動 UI（例如你用 python app.py 執行）
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)