response = openai.Completion.create(

engine="gpt-3.5-turbo",
prompt=prompt,
max_tokens=150 * rounds
)

app.py

@@ -2,13 +2,43 @@ import gradio as gr
 from gtts import gTTS
 import json
 import os
+import openai
 
 PASSWORD = os.environ['PASSWORD']
 OPEN_AI_KEY = os.environ['OPEN_AI_KEY']
 
+
+def generate_dialogue_with_openai(rounds, role1, role2):
+    openai.api_key = os.environ["OPEN_AI_KEY"]  # 從環境變數中取得API鑰匙
+
+    dialogue = []
+
+    # 這裡，我們使用一個基本的prompt，您可以根據需要修改它
+    prompt = "Write a conversation between two people, " + role1 + " and " + role2 + "."
+
+    response = openai.Completion.create(
+      engine="gpt-3.5-turbo",
+      prompt=prompt,
+      max_tokens=150 * rounds
+)
+
+    # 解析OpenAI的回應以獲取對話內容
+    # 這只是一個基本的方法，您可能需要根據實際的回應內容進行調整
+    lines = response.choices[0].text.strip().split("\n")
+
+    for i, line in enumerate(lines[:rounds]):
+        if i % 2 == 0:
+            dialogue.append({"role": role1, "content": line.split(":")[1].strip()})
+        else:
+            dialogue.append({"role": role2, "content": line.split(":")[1].strip()})
+
+    return dialogue
+
+
+
 def generate_dialogue(rounds, method, role1, role2):
     if method == "auto":
-        dialogue = [{"role": role1 if i % 2 == 0 else role2, "content": f"自動文本 {i+1}"} for i in range(rounds)]
+        dialogue = generate_dialogue_with_openai(rounds, role1, role2)
     else:
         dialogue = [{"role": role1, "content": "手動輸入文本 1"}, {"role": role2, "content": "手動輸入文本 2"}]
     return dialogue

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-gtts
+gtts
+openai