Update app.py

app.py

@@ -69,28 +69,26 @@ async def gpt_translate(text, target_lang):
     )
     return response.choices[0].message.content.strip()
 
-async def respond(user_input, history, step, language, questions, followup_mode):
+async def respond(user_input, history, step, language, questions, followup_mode, followup_stage):
     history.append({"role": "user", "content": user_input})
 
-    # === 用户选择语言 ===
     if language == "":
         user_language = user_input.strip().capitalize()
         questions_en = generate_question_set()
         translated_questions = [await gpt_translate(q, user_language) for q in questions_en]
         welcome = await gpt_translate("Great! We will now continue in your selected language.", user_language)
         history.append({"role": "assistant", "content": f"{welcome}\n\n1. {translated_questions[0]}"})
-        return history, "", 1, user_language, translated_questions, False
+        return history, "", 1, user_language, translated_questions, False, 0
 
-    # === 所有问题问完 ===
     if step >= len(questions):
         thank_you = await gpt_translate("Thank you for your response! It was a pleasure talking with you. I hope you enjoyed the conversation.", language)
         end_note = await gpt_translate("This concludes our questions. Please proceed with the rest of the survey. 📝", language)
         history.append({"role": "assistant", "content": thank_you})
         history.append({"role": "assistant", "content": end_note})
-        return history, "", step, language, questions, False
+        return history, "", step, language, questions, False, 0
 
-    # === 如果是追问回应 ===
     if followup_mode:
+        # bot 回复用户的追问回答
         comment_prompt = [
             {"role": "system", "content": f"Write a short, empathetic comment in {language} responding to the user's last answer."},
             history[-2],
@@ -108,16 +106,94 @@ async def respond(user_input, history, step, language, questions, followup_mode)
             history[-1]["content"] += c
             await asyncio.sleep(0.03)
 
-        # 确保只推进一个正确的问题（防止跳题）
-        if step < len(questions):
+        if followup_stage == 1:
+            # 可能还有第2次追问
+            if random.random() < 0.1:
+                followup_prompt = [
+                    {"role": "system", "content": f"Ask 1 more follow-up question based on the user's last answer in {language}. Do NOT number it."},
+                    history[-2],
+                    history[-1]
+                ]
+                followup_response = client.chat.completions.create(
+                    model="gpt-3.5-turbo",
+                    messages=followup_prompt,
+                    temperature=0.7
+                )
+                followup_q = followup_response.choices[0].message.content.strip()
+                history.append({"role": "assistant", "content": ""})
+                for c in followup_q:
+                    history[-1]["content"] += c
+                    await asyncio.sleep(0.03)
+                return history, "", step, language, questions, True, 2
+            else:
+                # 无第二次追问，进入下一题
+                next_question = f"{step+1}. {questions[step]}"
+                history.append({"role": "assistant", "content": ""})
+                for c in next_question:
+                    history[-1]["content"] += c
+                    await asyncio.sleep(0.03)
+                return history, "", step + 1, language, questions, False, 0
+
+        elif followup_stage == 2:
+            # 回答完第二次追问，直接进入下一主问题
             next_question = f"{step+1}. {questions[step]}"
             history.append({"role": "assistant", "content": ""})
             for c in next_question:
                 history[-1]["content"] += c
                 await asyncio.sleep(0.03)
-            return history, "", step + 1, language, questions, False
-        else:
-            return history, "", step, language, questions, False
+            return history, "", step + 1, language, questions, False, 0
+
+    # 正常主问题回答逻辑
+    comment_prompt = [
+        {"role": "system", "content": f"You are Robin, a warm and engaging conversational AI companion who chats naturally with the user, asks questions, and responds thoughtfully to their answers. Write 1–2 natural follow-up questions based on the user's answer. Do NOT number the questions. Just write them as natural, standalone questions. Write a short, personalized comment on the user's answer in {language}. Add a light emoji at the end."},
+        history[-2],
+        history[-1]
+    ]
+    comment_response = client.chat.completions.create(
+        model="gpt-3.5-turbo",
+        messages=comment_prompt,
+        temperature=0.8
+    )
+    bot_reply = comment_response.choices[0].message.content.strip()
+
+    followup_count = 0
+    rand = random.random()
+    if rand < 0.1:
+        followup_count = 2
+    elif rand < 0.5:
+        followup_count = 1
+
+    if followup_count > 0:
+        followup_prompt = [
+            {"role": "system", "content": f"Ask {followup_count} open-ended follow-up question(s) based on the user's last answer in {language}. Do NOT number them."},
+            history[-2],
+            history[-1]
+        ]
+        followup_response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=followup_prompt,
+            temperature=0.7
+        )
+        bot_reply += "\n\n" + followup_response.choices[0].message.content.strip()
+
+        history.append({"role": "assistant", "content": ""})
+        for c in bot_reply:
+            history[-1]["content"] += c
+            await asyncio.sleep(0.03)
+        return history, "", step, language, questions, True, 1
+
+    # 无追问，直接进入下一题
+    history.append({"role": "assistant", "content": ""})
+    for c in bot_reply:
+        history[-1]["content"] += c
+        await asyncio.sleep(0.03)
+
+    next_question = f"{step+1}. {questions[step]}"
+    history.append({"role": "assistant", "content": ""})
+    for c in next_question:
+        history[-1]["content"] += c
+        await asyncio.sleep(0.03)
+    return history, "", step + 1, language, questions, False, 0
 
     # === 正常主问题回答流程 ===
     comment_prompt = [