Update app.py

app.py

@@ -2,6 +2,8 @@ import os, shutil, base64, uuid, mimetypes, json
 from pydub import AudioSegment
 from openai import OpenAI
 import gradio as gr
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.responses import JSONResponse
 
 # ====== 基本設定 ======
 PASSWORD = os.getenv("APP_PASSWORD", "chou")
@@ -26,87 +28,75 @@ MIME_EXT = {
 }
 
 def _dataurl_to_file(data_url: str, orig_name: str | None = None) -> str:
-    # data_url: "data:audio/mp4;base64,AAAA..."
     try:
         header, b64 = data_url.split(",", 1)
     except ValueError:
-        raise ValueError("data URL 格式錯誤(缺少逗號)。")
-    # 取 MIME
+        raise ValueError("data URL format error (missing comma).")
     mime = header.split(";")[0].split(":", 1)[-1].strip()
     ext = MIME_EXT.get(mime) or (mimetypes.guess_extension(mime) or "m4a").lstrip(".")
-    # 臨時檔名
     fname = orig_name if (orig_name and "." in orig_name) else f"upload_{uuid.uuid4().hex}.{ext}"
     with open(fname, "wb") as f:
         f.write(base64.b64decode(b64))
-    print(f"✅ 已從 data URL 生成檔案:{fname}, 大小:{os.path.getsize(fname)} bytes")
+    print(f"✅ Created file from data URL: {fname}, size: {os.path.getsize(fname)} bytes")
     return fname
 
 def _extract_effective_path(file_obj) -> str:
-    """
-    從 Gradio 的 File 輸入中,得到真正存在的檔案路徑。
-    🔴 加強版:詳細記錄所有嘗試過程
-    """
-    print(f"\n🔍 開始解析檔案...")
-    print(f"📦 收到的類型: {type(file_obj)}")
-    print(f"📦 收到的內容: {file_obj}")
+    """從各種格式中提取有效檔案路徑"""
+    print(f"\n🔍 Starting file parsing...")
+    print(f"📦 Received type: {type(file_obj)}")
+    print(f"📦 Content: {str(file_obj)[:200]}...")
     
-    # 情況 A:字串(可能是路徑或 data:URL)
+    # 字串模式
     if isinstance(file_obj, str):
         s = file_obj.strip().strip('"')
-        print(f"  → 情況A:字串模式")
+        print(f"  → Mode A: String")
         if s.startswith("data:"):
-            print(f"  → 偵測到 data URL,長度:{len(s)}")
+            print(f"  → Detected data URL, length: {len(s)}")
             return _dataurl_to_file(s, None)
         if os.path.isfile(s):
-            print(f"  → 找到有效路徑:{s}")
+            print(f"  → Found valid path: {s}")
             return s
-        print(f"  → 字串無效,繼續嘗試...")
     
-    # 情況 B:dict
+    # 字典模式
     if isinstance(file_obj, dict):
-        print(f"  → 情況B:字典模式")
-        print(f"  → 字典的 keys: {list(file_obj.keys())}")
+        print(f"  → Mode B: Dictionary")
+        print(f"  → Keys: {list(file_obj.keys())}")
         
-        # 🔴 優先檢查 data(Base64 模式)
+        # 優先 data URL
         data = file_obj.get("data")
         if isinstance(data, str) and data.startswith("data:"):
-            print(f"  → ✅ 找到 data URL! 長度:{len(data)}")
+            print(f"  → ✅ Found data URL! Length: {len(data)}")
             orig_name = file_obj.get("orig_name")
-            print(f"  → 檔名:{orig_name}")
             return _dataurl_to_file(data, orig_name)
         
-        # 再檢查 path
+        # path
         p = str(file_obj.get("path") or "").strip().strip('"')
         if p and os.path.isfile(p):
-            print(f"  → 找到 path:{p}")
+            print(f"  → Found path: {p}")
             return p
         
-        # 檢查 url
+        # url
         u = str(file_obj.get("url") or "").strip().strip('"')
         if u and os.path.isfile(u):
-            print(f"  → 找到 url:{u}")
+            print(f"  → Found url: {u}")
             return u
-        
-        print(f"  → ❌ 字典內沒有有效的 data/path/url")
     
-    # 情況 C:物件
-    print(f"  → 情況C:物件模式")
+    # 物件模式
+    print(f"  → Mode C: Object")
     for attr in ("name", "path"):
         p = getattr(file_obj, attr, None)
         if isinstance(p, str):
             s = p.strip().strip('"')
             if os.path.isfile(s):
-                print(f"  → 找到物件屬性 {attr}:{s}")
+                print(f"  → Found object attr {attr}: {s}")
                 return s
     
-    # 物件上的 data:URL
     data = getattr(file_obj, "data", None)
     if isinstance(data, str) and data.startswith("data:"):
-        print(f"  → 找到物件的 data URL")
+        print(f"  → Found object data URL")
         return _dataurl_to_file(data, getattr(file_obj, "orig_name", None))
     
-    print(f"  → ❌ 所有方法都失敗!")
-    raise FileNotFoundError(f"無法解析上傳檔案。收到的類型:{type(file_obj)}, 內容:{str(file_obj)[:200]}")
+    raise FileNotFoundError(f"Cannot parse uploaded file. Type: {type(file_obj)}")
 
 # ====== 分段處理 ======
 def split_audio(path):
@@ -125,14 +115,13 @@ def split_audio(path):
 
 # ====== 轉錄核心 ======
 def transcribe_core(path, model="whisper-1"):
-    # iPhone LINE 常見:mp4(其實是音訊容器)
     if path.lower().endswith(".mp4"):
         fixed = path[:-4] + ".m4a"
         try:
             shutil.copy(path, fixed)
             path = fixed
         except Exception as e:
-            print(f"⚠️ mp4→m4a 失敗: {e}")
+            print(f"⚠️ mp4→m4a failed: {e}")
 
     chunks = split_audio(path)
     raw = []
@@ -146,7 +135,6 @@ def transcribe_core(path, model="whisper-1"):
             raw.append(txt)
     raw_txt = "\n".join(raw)
 
-    # 簡轉繁(不意譯)
     conv = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
@@ -157,7 +145,6 @@ def transcribe_core(path, model="whisper-1"):
     )
     trad = conv.choices[0].message.content.strip()
 
-    # 摘要(內容多→條列;內容少→一句話)
     summ = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
@@ -168,55 +155,99 @@ def transcribe_core(path, model="whisper-1"):
     )
     return trad, summ.choices[0].message.content.strip()
 
-# ====== 對外函式(UI / API 共用) ======
-def transcribe(password, file):
+# ====== Gradio UI 函式 ======
+def transcribe_ui(password, file):
     print("\n" + "="*50)
-    print("🎯 新的轉錄請求")
-    print(f"🔑 密碼: {password[:2]}*** (長度:{len(password)})")
-    print(f"📁 檔案: {type(file)}")
+    print("🎯 New transcription request (UI)")
+    print(f"🔑 Password: {password[:2]}*** (length: {len(password)})")
+    print(f"📁 File: {type(file)}")
     print("="*50)
     
     if password.strip() != PASSWORD:
-        print("❌ 密碼驗證失敗")
         return "❌ Password incorrect", "", ""
-    
     if not file:
-        print("❌ 未收到檔案")
         return "⚠️ No file uploaded", "", ""
 
     try:
         path = _extract_effective_path(file)
-        print(f"✅ 成功解析檔案:{path}")
-        print(f"📊 檔案大小:{os.path.getsize(path)} bytes")
-    except Exception as e:
-        import traceback
-        error_msg = traceback.format_exc()
-        print(f"❌ 檔案解析失敗:\n{error_msg}")
-        return f"❌ File parsing failed: {e}", "", ""
-
-    try:
         text, summary = transcribe_core(path)
-        print("✅ 轉錄完成")
         return "✅ Transcription completed", text, summary
     except Exception as e:
         import traceback
-        error_msg = traceback.format_exc()
-        print(f"❌ 轉錄失敗:\n{error_msg}")
-        return f"❌ Transcription failed: {e}", "", ""
+        print(traceback.format_exc())
+        return f"❌ Error: {e}", "", ""
 
 # ====== Gradio UI ======
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🎧 LINE Audio Transcription & Summary (Enhanced Debug)")
+    gr.Markdown("## 🎧 LINE Audio Transcription & Summary")
     pw = gr.Textbox(label="Password", type="password")
     f = gr.File(label="Upload Audio File")
     run = gr.Button("Start Transcription 🚀")
     s = gr.Textbox(label="Status", interactive=False)
     t = gr.Textbox(label="Transcription Result", lines=10)
     su = gr.Textbox(label="AI Summary", lines=8)
-    # 關鍵:queue=False 讓 API 直接回應
-    run.click(transcribe, [pw, f], [s, t, su], queue=False)
+    run.click(transcribe_ui, [pw, f], [s, t, su], queue=False)
+
+# ====== 🔴 新增:自訂 API 端點(繞過 Gradio validation) ======
+app = FastAPI()
+
+@app.post("/api/transcribe")
+async def transcribe_api(request: Request):
+    """
+    直接接受 Base64 data URL 的自訂 API
+    
+    請求格式:
+    {
+      "password": "chou",
+      "file": {
+        "data": "data:audio/m4a;base64,UklGR...",
+        "orig_name": "recording.m4a"
+      }
+    }
+    """
+    try:
+        body = await request.json()
+        print("\n" + "="*50)
+        print("🎯 New transcription request (Custom API)")
+        print(f"📦 Request body keys: {list(body.keys())}")
+        print("="*50)
+        
+        password = body.get("password", "")
+        if password.strip() != PASSWORD:
+            return JSONResponse(
+                status_code=401,
+                content={"error": "Password incorrect"}
+            )
+        
+        file_obj = body.get("file")
+        if not file_obj:
+            return JSONResponse(
+                status_code=400,
+                content={"error": "No file provided"}
+            )
+        
+        # 直接處理 Base64
+        path = _extract_effective_path(file_obj)
+        text, summary = transcribe_core(path)
+        
+        return JSONResponse(content={
+            "status": "success",
+            "transcription": text,
+            "summary": summary
+        })
+        
+    except Exception as e:
+        import traceback
+        error_detail = traceback.format_exc()
+        print(f"❌ API Error:\n{error_detail}")
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e), "detail": error_detail}
+        )
 
-app = demo
+# 掛載 Gradio 到 FastAPI
+app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)