Update app.py

app.py

@@ -4,7 +4,7 @@ from openai import OpenAI
 import gradio as gr
 
 # ========================
-# 🔐 使用者設定區（安全版本）
+# 🔐 使用者設定區（安全）
 # ========================
 PASSWORD = os.getenv("APP_PASSWORD", "defaultpass")
 MAX_SIZE = 25 * 1024 * 1024
@@ -31,56 +31,113 @@ def split_audio_if_needed(input_path):
     return chunk_files
 
 # ========================
-# 🎧 主功能：轉錄 + 摘要
+# 🎧 轉錄 + 摘要
 # ========================
-def transcribe_and_summarize(file):
+def transcribe_and_summarize(password, file, model_choice):
+    if password != PASSWORD:
+        return "❌ 密碼錯誤，請重新輸入。", "", ""
     if not file:
-        return "請上傳音訊檔。", ""
+        return "請上傳音訊檔。", "", ""
+
     chunks = split_audio_if_needed(file)
     transcripts = []
     for idx, f in enumerate(chunks, 1):
         with open(f, "rb") as audio_file:
             text = client.audio.transcriptions.create(
-                model="whisper-1",
+                model=model_choice,
                 file=audio_file,
                 response_format="text"
             )
         transcripts.append(text)
     full_text = "\n".join(transcripts)
+
     response = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "system", "content": "你是一位精準且擅長摘要的助手。"},
-            {"role": "user", "content": "請用繁體中文摘要以下內容：\n" + full_text}
+            {"role": "user", "content": f"請用繁體中文摘要以下內容：\n{full_text}"}
         ],
         temperature=0.4,
     )
     summary = response.choices[0].message.content.strip()
-    return full_text, summary
+    return full_text, summary, "✅ 已完成轉錄與摘要"
 
 # ========================
-# 🔑 密碼驗證包裝
+# 💬 進一步問 AI
 # ========================
-def transcribe_with_password(password, file):
-    if password != PASSWORD:
-        return "❌ 密碼錯誤，請重新輸入。", ""
-    return transcribe_and_summarize(file)
+def ask_about_transcript(full_text, user_question):
+    if not full_text.strip():
+        return "⚠️ 尚未有轉錄內容，請先上傳音訊。"
+    if not user_question.strip():
+        return "請輸入想詢問的問題。"
+    prompt = f"以下是語音轉錄的完整內容：\n\n{full_text}\n\n使用者的問題是：{user_question}\n\n請以繁體中文回答。"
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.6,
+    )
+    return response.choices[0].message.content.strip()
 
 # ========================
 # 🌐 Gradio 介面
 # ========================
-iface = gr.Interface(
-    fn=transcribe_with_password,
-    inputs=[
-        gr.Textbox(label="輸入密碼", type="password", placeholder="請輸入授權密碼"),
-        gr.Audio(type="filepath", label="上傳音訊檔 (.m4a, .aac, .wav)")
-    ],
-    outputs=[
-        gr.Textbox(label="完整轉錄文字", lines=10),
-        gr.Textbox(label="摘要結果", lines=10)
-    ],
-    title="語音轉錄與摘要工具 (受密碼保護)",
-    description="上傳音訊後，自動轉錄文字與產生摘要。請先輸入授權密碼以使用本服務。",
-)
-
-iface.launch()
+with gr.Blocks(theme=gr.themes.Soft(), css=".gr-button {font-size: 16px;} .copy-btn {background: #e0e0e0;}") as demo:
+    gr.Markdown("## 🎧 語音轉錄與摘要工具 (受密碼保護)")
+    gr.Markdown("上傳音訊後，自動轉錄文字並生成摘要。可進一步詢問 AI 關於對話內容。")
+
+    with gr.Row():
+        password_input = gr.Textbox(label="輸入密碼", type="password", placeholder="請輸入授權密碼")
+        model_choice = gr.Dropdown(
+            choices=["whisper-1", "gpt-4o-mini-transcribe"],
+            value="whisper-1",
+            label="選擇轉錄模型"
+        )
+
+    audio_input = gr.Audio(type="filepath", label="上傳音訊檔 (.m4a, .aac, .wav)")
+    transcribe_btn = gr.Button("開始轉錄與摘要 🚀")
+
+    status_box = gr.Textbox(label="狀態", interactive=False)
+
+    with gr.Row():
+        transcript_box = gr.Textbox(label="完整轉錄文字", lines=10)
+        copy_transcript = gr.Button("📋 複製", elem_classes="copy-btn")
+
+    with gr.Row():
+        summary_box = gr.Textbox(label="摘要結果", lines=10)
+        copy_summary = gr.Button("📋 複製", elem_classes="copy-btn")
+
+    with gr.Accordion("💬 進一步問 AI", open=False):
+        user_question = gr.Textbox(label="輸入你的問題（例如：我該如何回應？）", lines=2)
+        ask_btn = gr.Button("詢問 AI 🤔")
+        ai_reply = gr.Textbox(label="AI 回覆", lines=6)
+        copy_reply = gr.Button("📋 複製 AI 回覆", elem_classes="copy-btn")
+
+    # 事件綁定
+    transcribe_btn.click(
+        fn=transcribe_and_summarize,
+        inputs=[password_input, audio_input, model_choice],
+        outputs=[transcript_box, summary_box, status_box]
+    )
+
+    ask_btn.click(
+        fn=ask_about_transcript,
+        inputs=[transcript_box, user_question],
+        outputs=[ai_reply]
+    )
+
+    # 前端 JS 複製功能
+    copy_js = """
+    async (text) => {
+        try {
+            await navigator.clipboard.writeText(text);
+            alert("✅ 已複製到剪貼簿！");
+        } catch (err) {
+            alert("❌ 複製失敗：" + err);
+        }
+    }
+    """
+    copy_transcript.click(None, transcript_box, None, _js=copy_js)
+    copy_summary.click(None, summary_box, None, _js=copy_js)
+    copy_reply.click(None, ai_reply, None, _js=copy_js)
+
+demo.launch()