Update app.py

app.py

@@ -1,8 +1,11 @@
 """
-多策略 RAG 文件問答系統 v2 — ChromaDB + PDF/DOCX 版本（含 Telegram 推送）
+多策略 RAG 文件問答系統 v2 — ChromaDB + PDF/DOCX + Telegram 推播版
 
 安裝依賴：
     pip install gradio groq pypdf python-docx sentence-transformers numpy chromadb scikit-learn requests
+
+執行：
+    python rag_telegram_app.py
 """
 
 from __future__ import annotations
@@ -29,31 +32,39 @@ from sklearn.feature_extraction.text import TfidfVectorizer
 
 
 # ══════════════════════════════════════════════════════════
-#  Telegram 推送設定
+#  Telegram 推播工具
 # ══════════════════════════════════════════════════════════
-DEFAULT_TELEGRAM_CHAT_ID = "5731474175"
-TELEGRAM_MAX_LEN = 4000  # Telegram 訊息長度限制 (4096)，留一些緩衝
-
-
-def send_telegram_message(text: str, chat_id: str, token: str) -> dict:
-    """將文字訊息送到 Telegram。若文字過長會自動分段傳送。"""
-    if not token:
-        return {"ok": False, "error": "尚未提供 Bot Token"}
-    if not chat_id:
-        return {"ok": False, "error": "尚未提供 Chat ID"}
-    if not text:
-        return {"ok": False, "error": "empty text"}
-
-    url = f"https://api.telegram.org/bot{token}/sendMessage"
-    results = []
-    for i in range(0, len(text), TELEGRAM_MAX_LEN):
-        chunk = text[i:i + TELEGRAM_MAX_LEN]
-        try:
-            r = requests.post(url, data={"chat_id": chat_id, "text": chunk}, timeout=15)
-            results.append(r.json())
-        except Exception as exc:
-            results.append({"ok": False, "error": f"{type(exc).__name__}: {exc}"})
-    return results[-1] if results else {"ok": False, "error": "no chunks sent"}
+TELEGRAM_MAX_LEN = 4000  # Telegram 訊息長度上限為 4096，留一點餘量
+
+
+def send_telegram_message(token: str, chat_id: str, text: str) -> tuple[bool, str]:
+    """將文字訊息發送至指定的 Telegram 聊天室。"""
+    token = (token or "").strip()
+    chat_id = (chat_id or "").strip()
+    if not token or not chat_id:
+        return False, "⚠ 尚未設定 Telegram Bot Token 或 Chat ID"
+
+    if len(text) > TELEGRAM_MAX_LEN:
+        text = text[:TELEGRAM_MAX_LEN] + "\n…（內容過長，已截斷）"
+
+    try:
+        resp = requests.post(
+            f"https://api.telegram.org/bot{token}/sendMessage",
+            data={"chat_id": chat_id, "text": text},
+            timeout=10,
+        )
+        data = resp.json()
+        if data.get("ok"):
+            return True, "✓ 已成功發送至 Telegram"
+        return False, f"✗ Telegram 發送失敗：{data.get('description', '未知錯誤')}"
+    except Exception as exc:
+        return False, f"✗ Telegram 發送失敗：{type(exc).__name__}: {exc}"
+
+
+def test_telegram_connection(token: str, chat_id: str) -> str:
+    """測試 Telegram Bot Token / Chat ID 是否正確（不會打擾使用者太多）。"""
+    ok, msg = send_telegram_message(token, chat_id, "✅ RAG 系統已連接，測試訊息成功！")
+    return msg if not ok else "✓ 連線成功，已發送測試訊息"
 
 
 # ══════════════════════════════════════════════════════════
@@ -381,7 +392,7 @@ class MultiStrategyRAG:
         if not self.client:
             return (
                 "⚠ 尚未設定 Groq API Key。\n"
-                "請在左欄「Step 01」輸入您的 Groq API Key 並點擊「套用」後再提問。\n\n"
+                "請在左欄「Step 00」輸入您的 Groq API Key 並點擊「套用」後再提問。\n\n"
                 "（檢索已完成，可在下方「查看檢索到的文本片段」確認結果）",
                 source_preview,
             )
@@ -414,21 +425,16 @@ class MultiStrategyRAG:
 #  Gradio UI
 # ══════════════════════════════════════════════════════════
 STRATEGY_INFO = [
-    ("semantic",     "🔍 語意搜尋",   "ChromaDB 向量相似度，最通用"),
-    ("tfidf",        "📊 TF-IDF",     "字元 n-gram 關鍵詞統計"),
-    ("hybrid",       "⚡ 混合搜尋",   "語意 + TF-IDF 結果合併去重"),
-    ("rerank",       "🎯 重新排序",   "LLM 對候選片段二次評分"),
-    ("multi_query",  "🔄 多查詢擴展", "自動生成多角度問題聯合搜尋"),
-    ("compress",     "✂️ 上下文壓縮", "LLM 提取最相關句子精簡上下文"),
-    ("parent_child", "📂 父子文檔",   "小片段定位 → 回傳對應大片段"),
-    ("hyde",         "💡 HyDE",       "先生成假設答案再語意搜尋"),
+    ("semantic",     "語意搜尋",   "ChromaDB 向量相似度，最通用",         "🔍"),
+    ("tfidf",        "TF-IDF",     "字元 n-gram 關鍵詞統計",              "📊"),
+    ("hybrid",       "混合搜尋",   "語意 + TF-IDF 結果合併去重",          "⚡"),
+    ("rerank",       "重新排序",   "LLM 對候選片段二次評分",              "🎯"),
+    ("multi_query",  "多查詢擴展", "自動生成多角度問題聯合搜尋",          "🔄"),
+    ("compress",     "上下文壓縮", "LLM 提取最相關句子精簡上下文",        "✂️"),
+    ("parent_child", "父子文檔",   "小片段定位 → 回傳對應大片段",        "📂"),
+    ("hyde",         "HyDE",       "先生成假設答案再語意搜尋",            "💡"),
 ]
 
-# 顯示用 label -> 內部 key 對照表
-STRATEGY_LABEL_TO_KEY = {label: key for key, label, _ in STRATEGY_INFO}
-STRATEGY_CHOICES = [label for _, label, _ in STRATEGY_INFO]
-STRATEGY_DESC_HTML = "<br>".join(f"<b>{label}</b> — {desc}" for _, label, desc in STRATEGY_INFO)
-
 CSS = """
 body, .gradio-container { background:#f5f4f1 !important; }
 
@@ -446,43 +452,34 @@ body, .gradio-container { background:#f5f4f1 !important; }
 .pill { display:inline-block; margin:10px 5px 0 0; padding:3px 10px; border-radius:16px;
         font-size:11px; background:#e8f4f0; color:#2d6a4f; border:1px solid rgba(45,106,79,.2); }
 .pill-amber { background:#fdf4e3; color:#b87a1a; border-color:rgba(184,122,26,.25); }
+.pill-blue { background:#e8f0fd; color:#1a5cb8; border-color:rgba(26,92,184,.25); }
 
 /* API Key 區塊 */
-#apikey-box {
+#apikey-box, #telegram-box {
     background: #fffbf2;
     border: 1.5px solid #f0c96a;
     border-radius: 10px;
     padding: 12px 14px;
     margin-bottom: 8px;
 }
+#telegram-box { background:#f0f6ff; border-color:#a9c8f0; }
 
-/* Telegram 區塊 */
-#telegram-box {
-    background: #eef6ff;
-    border: 1.5px solid #8ec4f0;
-    border-radius: 10px;
-    padding: 12px 14px;
-    margin-bottom: 8px;
-}
-
-/* 策略選擇區塊 */
-#strategy-box {
+.strat-grid { display:grid; grid-template-columns:repeat(4,1fr); gap:10px; margin:10px 0 16px; }
+.strat-card {
     background:#fff;
     border:1.5px solid #e5e0d8;
     border-radius:10px;
     padding:10px 12px;
-    margin-bottom: 8px;
-}
-#strategy-box .wrap { gap:6px !important; }
-#strategy-box label {
-    border:1.5px solid #e5e0d8 !important;
-    border-radius:8px !important;
-    padding:8px 10px !important;
-    margin:0 !important;
-    transition: border-color .15s, background .15s;
+    cursor:pointer;
+    transition:border-color .15s, box-shadow .15s;
+    text-align:left;
+    width:100%;
 }
-#strategy-box label:hover { border-color:#2d6a4f !important; }
-#strategy-desc { font-size:11px; color:#7a6e67; line-height:1.6; margin-top:6px; }
+.strat-card:hover { border-color:#2d6a4f; box-shadow:0 2px 8px rgba(45,106,79,.12); }
+.strat-card.active { border-color:#2d6a4f; background:#f0f9f5; box-shadow:0 2px 10px rgba(45,106,79,.18); }
+.strat-icon  { font-size:20px; margin-bottom:4px; }
+.strat-name  { font-size:13px; font-weight:700; color:#1a1714; margin:0 0 2px; }
+.strat-desc  { font-size:11px; color:#7a6e67; line-height:1.4; }
 
 .sec-label { font-size:11px; letter-spacing:1.5px; text-transform:uppercase;
              color:#7a6e67; font-weight:700; margin:16px 0 8px; }
@@ -490,25 +487,51 @@ body, .gradio-container { background:#f5f4f1 !important; }
             border-radius:12px !important; padding:16px !important; }
 #ask-btn { background:#2d6a4f !important; color:#fff !important; border:0 !important; border-radius:8px !important; }
 #apply-key-btn { background:#b87a1a !important; color:#fff !important; border:0 !important; border-radius:8px !important; }
-#send-tg-btn { background:#0088cc !important; color:#fff !important; border:0 !important; border-radius:8px !important; }
+#apply-tg-btn, #test-tg-btn { background:#1a5cb8 !important; color:#fff !important; border:0 !important; border-radius:8px !important; }
 """
 
 HEADER_HTML = """
 <div id="hdr">
-  <div class="hdr-eyebrow">Intelligent Document Analysis · v2</div>
+  <div class="hdr-eyebrow">Intelligent Document Analysis · v2 + Telegram</div>
   <div class="hdr-title">多策略 RAG 文件問答系統</div>
-  <div class="hdr-sub">支援 PDF / DOCX 上傳，採用 ChromaDB 持久化向量資料庫與 8 種 RAG 檢索策略，並可將結果推送至 Telegram</div>
+  <div class="hdr-sub">支援 PDF / DOCX 上傳，採用 ChromaDB 持久化向量資料庫與 8 種 RAG 檢索策略，並可將回答推播至 Telegram</div>
   <div>
     <span class="pill">▸ Groq API</span>
     <span class="pill">▸ llama-3.1-8b-instant</span>
     <span class="pill pill-amber">▸ ChromaDB</span>
     <span class="pill pill-amber">▸ PDF / DOCX</span>
     <span class="pill">▸ SentenceTransformers</span>
-    <span class="pill">▸ Telegram</span>
+    <span class="pill pill-blue">▸ Telegram Bot</span>
   </div>
 </div>
 """
 
+
+def build_strategy_menu(selected: str = "semantic") -> str:
+    cards = []
+    for key, name, desc, icon in STRATEGY_INFO:
+        active_cls = "active" if key == selected else ""
+        cards.append(
+            f"""<button class="strat-card {active_cls}" onclick="selectStrategy('{key}', this)" type="button">
+  <div class="strat-icon">{icon}</div>
+  <div class="strat-name">{name}</div>
+  <div class="strat-desc">{desc}</div>
+</button>"""
+        )
+    return f'<div class="strat-grid">{"".join(cards)}</div>'
+
+
+STRATEGY_MENU_JS = """
+<script>
+function selectStrategy(key, el) {
+    document.querySelectorAll('.strat-card').forEach(c => c.classList.remove('active'));
+    el.classList.add('active');
+    const inp = document.getElementById('strategy-hidden');
+    if (inp) { inp.value = key; inp.dispatchEvent(new Event('input')); }
+}
+</script>
+"""
+
 EXAMPLE_QS = [
     ["這份文件的主要內容是什麼？"],
     ["文件中提到哪些重要概念或定義？"],
@@ -519,14 +542,17 @@ EXAMPLE_QS = [
 
 
 def create_interface():
-    env_key = os.getenv("GROQ_API_KEY", "").strip()
+    # 啟動時嘗試從環境變數讀取（建議使用環境變數，避免在介面或程式碼中硬編碼）
+    env_groq_key = os.getenv("GROQ_API_KEY", "").strip()
+    env_tg_token = os.getenv("TELEGRAM_BOT_TOKEN", "").strip()
+    env_tg_chat_id = os.getenv("TELEGRAM_CHAT_ID", "").strip()
+
     rag = MultiStrategyRAG(chroma_path="./chroma_db")
-    if env_key:
-        rag.set_api_key(env_key)
+    if env_groq_key:
+        rag.set_api_key(env_groq_key)
 
     current_strategy = {"key": "semantic"}
-    # 暫存最近一次回答 / 來源片段，供 Telegram 推送使用
-    last_result = {"answer": "", "source": ""}
+    telegram_config = {"token": env_tg_token, "chat_id": env_tg_chat_id}
 
     def apply_api_key(api_key: str):
         key = (api_key or "").strip()
@@ -536,46 +562,52 @@ def create_interface():
             return f"✓ API Key 已套用（{masked}）"
         return "⚠ API Key 已清除，無法呼叫 LLM"
 
+    def apply_telegram_config(token: str, chat_id: str):
+        telegram_config["token"] = (token or "").strip()
+        telegram_config["chat_id"] = (chat_id or "").strip()
+        if telegram_config["token"] and telegram_config["chat_id"]:
+            masked = (
+                telegram_config["token"][:6] + "****" + telegram_config["token"][-4:]
+                if len(telegram_config["token"]) > 10
+                else "****"
+            )
+            return f"✓ Telegram 設定已套用（Token：{masked} · Chat ID：{telegram_config['chat_id']}）"
+        return "⚠ Telegram Token 或 Chat ID 尚未完整設定"
+
+    def do_test_telegram():
+        return test_telegram_connection(telegram_config["token"], telegram_config["chat_id"])
+
     def upload_document(file):
         if file is None:
             return "⚠ 請選擇 PDF 或 DOCX 檔案"
         return rag.load_document(file.name)
 
-    def set_strategy(label: str):
-        key = STRATEGY_LABEL_TO_KEY.get(label, "semantic")
+    def set_strategy(key: str):
         current_strategy["key"] = key
-        return f"✓ 已選擇策略：{label}"
-
-    def ask(query, top_k):
-        answer, source = rag.generate_answer(query, current_strategy["key"], int(top_k))
-        last_result["answer"] = answer or ""
-        last_result["source"] = source or ""
-        return answer, source
+        return f"✓ 已選擇策略：{dict((k, n) for k, n, *_ in STRATEGY_INFO).get(key, key)}"
 
-    def push_to_telegram(include_source: bool, chat_id_input: str, token_input: str):
-        if not last_result["answer"]:
-            return "⚠ 尚無可推送的回答，請先提問。"
+    def ask(query, top_k, send_to_telegram):
+        answer, source_preview = rag.generate_answer(query, current_strategy["key"], int(top_k))
 
-        token = (token_input or "").strip()
-        if not token:
-            return "⚠ 請先輸入 Telegram Bot Token"
-
-        chat_id = (chat_id_input or "").strip() or DEFAULT_TELEGRAM_CHAT_ID
-
-        text = f"📄 文件：{rag.source_name or '未知'}\n"
-        text += f"❓ 策略：{rag.STRATEGY_MAP.get(current_strategy['key'], current_strategy['key'])}\n\n"
-        text += f"💬 回答：\n{last_result['answer']}"
-
-        if include_source:
-            text += f"\n\n— 檢索片段 —\n{last_result['source']}"
+        telegram_status = ""
+        if send_to_telegram:
+            strategy_label = MultiStrategyRAG.STRATEGY_MAP.get(
+                current_strategy["key"], current_strategy["key"]
+            )
+            tg_text = (
+                f"📄 文件：{rag.source_name or '(未載入)'}\n"
+                f"🧩 策略：{strategy_label}\n"
+                f"❓ 問題：{query}\n\n"
+                f"💬 回答：\n{answer}"
+            )
+            ok, telegram_status = send_telegram_message(
+                telegram_config["token"], telegram_config["chat_id"], tg_text
+            )
 
-        result = send_telegram_message(text, chat_id=chat_id, token=token)
-        if result.get("ok"):
-            return "✓ 已成功推送至 Telegram"
-        return f"✗ 推送失敗：{result.get('description') or result.get('error') or result}"
+        return answer, source_preview, telegram_status
 
     with gr.Blocks(
-        title="多策略 RAG 文件問答 v2",
+        title="多策略 RAG 文件問答 v2 + Telegram",
         css=CSS,
         theme=gr.themes.Base(
             primary_hue=gr.themes.colors.green,
@@ -588,37 +620,13 @@ def create_interface():
             # ── 左欄 ──────────────────────────────────
             with gr.Column(scale=1, min_width=320, elem_classes="card-box"):
 
-                # Step 00 · Telegram 推送（移至最上方，避免找不到）
-                gr.HTML("<div class='sec-label'>Step 00 · Telegram 推送</div>")
-                with gr.Group(elem_id="telegram-box"):
-                    tg_token = gr.Textbox(
-                        label="Bot Token",
-                        placeholder="例如：123456789:AAAxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
-                        type="password",
-                        lines=1,
-                    )
-                    tg_chat_id = gr.Textbox(
-                        label="Chat ID",
-                        placeholder=f"預設：{DEFAULT_TELEGRAM_CHAT_ID}（留空使用此預設值）",
-                        lines=1,
-                    )
-                    tg_include_source = gr.Checkbox(
-                        label="同時推送檢索片段內容", value=False
-                    )
-                    send_tg_btn = gr.Button(
-                        "📨 推送回答到 Telegram", elem_id="send-tg-btn"
-                    )
-                    tg_status = gr.Textbox(
-                        label="推送狀態", interactive=False, lines=1
-                    )
-
-                # Step 01：Groq API Key
-                gr.HTML("<div class='sec-label'>Step 01 · Groq API Key</div>")
+                # Step 00：Groq API Key
+                gr.HTML("<div class='sec-label'>Step 00 · Groq API Key</div>")
                 with gr.Group(elem_id="apikey-box"):
                     api_key_input = gr.Textbox(
                         label="",
                         placeholder="gsk_xxxxxxxxxxxxxxxxxxxxxxxx",
-                        value=env_key,
+                        value=env_groq_key,
                         type="password",
                         lines=1,
                         show_label=False,
@@ -627,43 +635,78 @@ def create_interface():
                         "套用 API Key", size="sm", elem_id="apply-key-btn"
                     )
                     api_key_status = gr.Textbox(
-                        value="✓ API Key 已從環境變數載入" if env_key else "⚠ 尚未設定 API Key",
+                        value="✓ API Key 已從環境變數載入" if env_groq_key else "⚠ 尚未設定 API Key",
                         interactive=False,
                         lines=1,
                         label="",
                         show_label=False,
                     )
 
-                # Step 02：上傳文件
-                gr.HTML("<div class='sec-label'>Step 02 · 上傳文件</div>")
+                # Step 00b：Telegram 推播設定
+                gr.HTML("<div class='sec-label'>Step 00b · Telegram 推播設定</div>")
+                with gr.Group(elem_id="telegram-box"):
+                    tg_token_input = gr.Textbox(
+                        label="Bot Token",
+                        placeholder="123456789:AAxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
+                        value=env_tg_token,
+                        type="password",
+                        lines=1,
+                    )
+                    tg_chat_id_input = gr.Textbox(
+                        label="Chat ID",
+                        placeholder="例如：123456789",
+                        value=env_tg_chat_id,
+                        lines=1,
+                    )
+                    with gr.Row():
+                        apply_tg_btn = gr.Button("套用設定", size="sm", elem_id="apply-tg-btn")
+                        test_tg_btn = gr.Button("發送測試訊息", size="sm", elem_id="test-tg-btn")
+                    tg_status = gr.Textbox(
+                        value=(
+                            "✓ Telegram 設定已從環境變數載入"
+                            if env_tg_token and env_tg_chat_id
+                            else "⚠ 尚未設定 Telegram"
+                        ),
+                        interactive=False,
+                        lines=1,
+                        label="",
+                        show_label=False,
+                    )
+                    send_to_telegram_checkbox = gr.Checkbox(
+                        label="提問後自動將回答推播至 Telegram",
+                        value=False,
+                    )
+
+                # Step 01：上傳文件
+                gr.HTML("<div class='sec-label'>Step 01 · 上傳文件</div>")
                 file_input = gr.File(label="PDF / DOCX", file_types=[".pdf", ".docx"])
                 load_btn = gr.Button("↑ 載入文件")
                 status = gr.Textbox(label="狀態", interactive=False, lines=3)
 
-                # Step 03：RAG 策略（改用 Radio，可正常點選）
-                gr.HTML("<div class='sec-label'>Step 03 · 選擇 RAG 策略</div>")
-                with gr.Group(elem_id="strategy-box"):
-                    strategy_radio = gr.Radio(
-                        choices=STRATEGY_CHOICES,
-                        value=STRATEGY_CHOICES[0],
-                        label="",
-                        show_label=False,
-                    )
-                    gr.HTML(f"<div id='strategy-desc'>{STRATEGY_DESC_HTML}</div>")
+                # Step 02：RAG 策略
+                gr.HTML("<div class='sec-label'>Step 02 · 選擇 RAG 策略</div>")
+                gr.HTML(build_strategy_menu("semantic"))
+                strategy_input = gr.Textbox(
+                    value="semantic",
+                    elem_id="strategy-hidden",
+                    label="",
+                    visible=False,
+                )
                 strategy_status = gr.Textbox(
-                    value=f"✓ 已選擇策略：{STRATEGY_CHOICES[0]}",
+                    value="✓ 已選擇策略：語意搜尋",
                     interactive=False,
                     lines=1,
                     label="目前策略",
                 )
+                gr.HTML(STRATEGY_MENU_JS)
 
-                # Step 04：參數
-                gr.HTML("<div class='sec-label'>Step 04 · 搜尋參數</div>")
+                # Step 03：參數
+                gr.HTML("<div class='sec-label'>Step 03 · 搜尋參數</div>")
                 topk = gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Top-K 片段數量")
 
             # ── 右欄：問答 ────────────────────────────
             with gr.Column(scale=2, elem_classes="card-box"):
-                gr.HTML("<div class='sec-label'>Step 05 · 輸入問題</div>")
+                gr.HTML("<div class='sec-label'>Step 04 · 輸入問題</div>")
                 qin = gr.Textbox(
                     label="",
                     placeholder="例如：這份文件的核心論點是什麼？",
@@ -674,6 +717,9 @@ def create_interface():
                 gr.HTML("<div class='sec-label'>AI 回答</div>")
                 ans = gr.Textbox(label="", lines=12, interactive=False)
 
+                gr.HTML("<div class='sec-label'>Telegram 推播狀態</div>")
+                tg_send_status = gr.Textbox(label="", lines=1, interactive=False)
+
                 with gr.Accordion("▸ 查看檢索到的文本片段", open=False):
                     src = gr.Textbox(label="", lines=10, interactive=False)
 
@@ -682,19 +728,31 @@ def create_interface():
         # ── 事件綁定 ──────────────────────────────────
         apply_key_btn.click(fn=apply_api_key, inputs=[api_key_input], outputs=[api_key_status])
         api_key_input.submit(fn=apply_api_key, inputs=[api_key_input], outputs=[api_key_status])
-        load_btn.click(fn=upload_document, inputs=[file_input], outputs=[status])
-        strategy_radio.change(fn=set_strategy, inputs=[strategy_radio], outputs=[strategy_status])
-        ask_btn.click(fn=ask, inputs=[qin, topk], outputs=[ans, src])
-        qin.submit(fn=ask, inputs=[qin, topk], outputs=[ans, src])
-        send_tg_btn.click(
-            fn=push_to_telegram,
-            inputs=[tg_include_source, tg_chat_id, tg_token],
+
+        apply_tg_btn.click(
+            fn=apply_telegram_config,
+            inputs=[tg_token_input, tg_chat_id_input],
             outputs=[tg_status],
         )
+        test_tg_btn.click(fn=do_test_telegram, outputs=[tg_status])
+
+        load_btn.click(fn=upload_document, inputs=[file_input], outputs=[status])
+        strategy_input.change(fn=set_strategy, inputs=[strategy_input], outputs=[strategy_status])
+
+        ask_btn.click(
+            fn=ask,
+            inputs=[qin, topk, send_to_telegram_checkbox],
+            outputs=[ans, src, tg_send_status],
+        )
+        qin.submit(
+            fn=ask,
+            inputs=[qin, topk, send_to_telegram_checkbox],
+            outputs=[ans, src, tg_send_status],
+        )
 
     return demo
 
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(share=False, server_name="0.0.0.0")
+    demo.launch(share=False, server_name="0.0.0.0")