2

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 from gtts import gTTS
 import json
+import os
+
+PASSWORD = os.environ['PASSWORD']
+OPEN_AI_KEY = os.environ['OPEN_AI_KEY']
 
 def generate_dialogue(rounds, method, role1, role2):
     if method == "auto":
@@ -10,8 +14,12 @@ def generate_dialogue(rounds, method, role1, role2):
     return dialogue
 
 def main_function(rounds: int, method: str, role1: str, role2: str):
+    if password != os.environ.get("PASSWORD", ""):
+        return "错误的密码，请重新输入。", "", ""
+
     structured_dialogue = generate_dialogue(rounds, method, role1, role2)
     
+    
     # Convert structured dialogue for Chatbot component to show "role1: content1" and "role2: content2" side by side
     chatbot_dialogue = []
     for i in range(0, len(structured_dialogue), 2):  # We iterate with a step of 2 to take pairs