def extract_json_from_response(response_text):

app.py

@@ -3,40 +3,45 @@ from gtts import gTTS
 import json
 import os
 import openai
+import re
+
 
 PASSWORD = os.environ['PASSWORD']
 OPEN_AI_KEY = os.environ['OPEN_AI_KEY']
 
 
+def extract_json_from_response(response_text):
+    # 使用正則表達式匹配 JSON 格式的對話
+    match = re.search(r'\[\s*\{.*?\}\s*\]', response_text, re.DOTALL)
+    if match:
+        json_str = match.group(0)
+        return json.loads(json_str)
+    else:
+        raise ValueError("JSON dialogue not found in the response.")
+
 
 def create_chat_dialogue(rounds, role1, role2, theme="購物"):
     openai.api_key = os.environ["OPEN_AI_KEY"]
 
     # 初始化對話
-    messages = [
-        {"role": "system", "content": f"您將進行一場以{theme}為主題的對話。{role1}和{role2}將是參與者。"}
-    ]
+    prompt = f"您將進行一場以{theme}為主題的對話。{role1}和{role2}將是參與者。請依次交談{rounds}輪。以json格式儲存對話。並回傳對話JSON文件。格式為：[{{role:\"A\", content: \".....\"}}, {{role:\"B\", content: \".....\"}}]"
 
-    # 呼叫API直到達到所需的對話輪數
-    while len(messages) - 1 < 2 * rounds:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=messages
-        )
+    response = openai.Completion.create(
+        model="gpt-3.5-turbo",
+        prompt=prompt,
+        max_tokens=500  # 設定一個較大的值，可根據需要調整
+    )
 
-        # 加入新消息
-        messages.extend(response['choices'][0]['messages'])
-
-    # 只保留role1和role2的消息
-    dialogue = [{"role": message["role"], "content": message["content"]} for message in messages if message["role"] in [role1, role2]]
+    print(response)
+    
+    response_text = response.choices[0].text.strip()
+    dialogue = extract_json_from_response(response_text)
 
+    print(dialogue)
+    # 這裡直接返回JSON格式的對話，但考慮到這可能只是一個字符串，您可能還需要將它解析為一個Python對象
     return dialogue
 
 
-
-
-
-
 def generate_dialogue(rounds, method, role1, role2):
     if method == "auto":
         dialogue = create_chat_dialogue(rounds, role1, role2)