Update app.py

app.py

@@ -260,21 +260,43 @@ def two_stage_qa(question, candidate_paragraphs_str, max_seq_len_mc=512, max_seq
     return final_answer, f"選中的段落 (索引 {selected_idx}):\n{selected_paragraph}", predictions_dict
 
 # --- 創建 Gradio 界面 ---
+# 定義預設的問題和段落內容
+DEFAULT_QUESTION = "世界最高峰是什麼？"
+DEFAULT_PARAGRAPHS = (
+    "珠穆朗瑪峰是喜馬拉雅山脈的主峰，位於中國與尼泊爾邊界上，是世界海拔最高的山峰。\n"
+    "喬戈里峰，又稱K2，是喀喇崑崙山脈的主峰，海拔8611米，是世界第二高峰，位於中國與巴基斯坦邊界。\n"
+    "干城章嘉峰位於喜馬拉雅山脈中段尼泊爾和印度邊界線上，海拔8586米，為世界第三高峰。\n"
+    "洛子峰，海拔8516米，為世界第四高峰，位於珠穆朗瑪峰以南約3公里處，同屬喜馬拉雅山脈。"
+)
+
 iface = gr.Interface(
-    fn=two_stage_qa,
+    fn=two_stage_qa, # 您的兩階段問答處理函數
     inputs=[
-        gr.Textbox(lines=2, placeholder="輸入您的問題...", label="問題 (Question)"),
-        gr.Textbox(lines=10, placeholder="在此處輸入候選段落，每段一行...", label="候選段落 (Candidate Paragraphs - One per line)")
+        gr.Textbox(
+            lines=2, 
+            placeholder="輸入您的問題...", 
+            label="問題 (Question)", 
+            value=DEFAULT_QUESTION  # <--- 為問題設置預設值
+        ),
+        gr.Textbox(
+            lines=10, 
+            placeholder="在此處輸入候選段落，每段一行...", 
+            label="候選段落 (Candidate Paragraphs - One per line)", 
+            value=DEFAULT_PARAGRAPHS # <--- 為段落設置預設值
+        )
     ],
     outputs=[
         gr.Textbox(label="預測答案 (Predicted Answer)"),
         gr.Textbox(label="選中的相關段落 (Selected Relevant Paragraph)"),
-        gr.JSON(label="原始預測字典 (Raw Predictions Dict - for debugging)") # 可選的調試輸出
+        gr.JSON(label="原始預測字典 (Raw Predictions Dict - for debugging)") 
     ],
     title="兩階段中文抽取式問答系統",
     description="輸入一個問題和多個候選段落（每行一個段落）。系統會先選擇最相關的段落，然後從中抽取答案。",
-    allow_flagging="never"
+    allow_flagging="never" # 或者您希望的標記設置
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    if models_loaded_successfully: # 確保模型已加載才啟動
+        iface.launch()
+    else:
+        print(f"Gradio 應用無法啟動，因為模型加載失敗: {error_message if 'error_message' in locals() else '未知錯誤'}")