def main_function(password: str, rounds: int, method: str, role1: str, role2: str):

app.py

@@ -13,7 +13,7 @@ def generate_dialogue(rounds, method, role1, role2):
         dialogue = [{"role": role1, "content": "手動輸入文本 1"}, {"role": role2, "content": "手動輸入文本 2"}]
     return dialogue
 
-def main_function(rounds: int, method: str, role1: str, role2: str):
+def main_function(password: str, rounds: int, method: str, role1: str, role2: str):
     if password != os.environ.get("PASSWORD", ""):
         return "错误的密码，请重新输入。", "", ""
 
@@ -51,6 +51,7 @@ if __name__ == "__main__":
     gr.Interface(
         main_function, 
         [
+            gr.components.Textbox(label="输入密码", type="password"),  # 加入密碼輸入框
             gr.components.Slider(minimum=2, maximum=6, step=2, label="對話輪數"),
             gr.components.Dropdown(choices=["auto", "manual"], label="生成方式"),
             gr.components.Textbox(label="角色 1 名稱"),