Update app.py

app.py

@@ -9,6 +9,7 @@ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 DEFAULT_LANGUAGE = "English"
 WELCOME_MESSAGE_EN = "Hi! I'm Robin, your dialogue partner.\nWe're about to begin a conversation that includes nine questions. Are you ready?\nPlease tell me which language you'd like to use."
 
+# === Question Bank ===
 SET_I = [
     "Given the choice of anyone in the world, whom would you want as a dinner guest?",
     "Would you like to be famous? In what way?",
@@ -20,7 +21,7 @@ SET_I = [
     "Name three things you and me appear to have in common.",
     "For what in your life do you feel most grateful?",
     "If you could change anything about the way you were raised, what would it be?",
-    "Take one minute and tell me your life story in as much detail as possible.",
+    "Take one minutes and tell me some of your life stories in as much detail as possible.",
     "If you could wake up tomorrow having gained any one quality or ability, what would it be?"
 ]
 
@@ -34,13 +35,13 @@ SET_II = [
     "If you knew that in one year you would die suddenly, would you change anything about the way you are now living? Why?",
     "What does friendship mean to you?",
     "What roles do love and affection play in your life?",
-    "Alternate sharing something you consider a positive characteristic of me, I will appreciate it.",
+    "Alternate sharing something you consider a positive characteristic of me, I will be appreciate.",
     "How close and warm is your family? Do you feel your childhood was happier than most other people’s?",
     "How do you feel about your relationship with your mother?"
 ]
 
 SET_III = [
-    "Let's try something fun: Can you write three sentences that start with 'We' and describe something true about us — you and me, chatting here together?",
+    "Let's try something fun: Can you write three sentences that start with 'We' and describe something true about us — you and me, chatting here together? (For example: 'We are both curious about each other.')",
     "Let’s complete this sentence together: 'I wish I had someone I could share ___ with.' What would you say?",
     "If you were going to become a close friend with me, please share what would be important for me to know.",
     "Tell me what you like about me; be very honest this time, saying things that you might not say to someone you’ve just met.",
@@ -70,12 +71,13 @@ async def gpt_translate(text, target_lang):
 
 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]}"})
+        history.append({"role": "assistant", "content": f"{welcome}\n\n1. {translated_questions[0]}"} )
         return history, "", 1, user_language, translated_questions, False, 0
 
     if step >= len(questions):
@@ -85,55 +87,196 @@ async def respond(user_input, history, step, language, questions, followup_mode,
         history.append({"role": "assistant", "content": end_note})
         return history, "", step, language, questions, False, 0
 
+    # ======== 追问阶段 ==========
     if followup_mode:
+        # 获取针对用户追问回答的评论
         comment_prompt = [
             {"role": "system", "content": f"Write a short, empathetic comment in {language} responding to the user's last answer."},
             history[-2],
             history[-1]
         ]
-        comment_response = client.chat.completions.create(model="gpt-3.5-turbo", messages=comment_prompt, temperature=0.7)
+        comment_response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=comment_prompt,
+            temperature=0.7
+        )
         comment = comment_response.choices[0].message.content.strip()
-        history.append({"role": "assistant", "content": comment})
-
-        if followup_stage == 1 and random.random() < 0.1:
-            followup_prompt = [{"role": "system", "content": f"Ask 1 more follow-up question in {language}."}, 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": followup_q})
-            return history, "", step, language, questions, True, 2
-        else:
+
+        if followup_stage == 1:
+            if random.random() < 0.1:
+                # 有第二次追问：评论 + 第二次追问
+                followup_prompt = [
+                    {"role": "system", "content": f"Ask 1 more open-ended 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()
+                combined = comment + "\n\n" + followup_q
+                history.append({"role": "assistant", "content": ""})
+                for c in combined:
+                    history[-1]["content"] += c
+                    await asyncio.sleep(0.03)
+                return history, "", step, language, questions, True, 2
+            else:
+                # 无第二次追问：评论 + 下一题
+                history.append({"role": "assistant", "content": ""})
+                for c in comment:
+                    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
+
+        elif followup_stage == 2:
+            # 第二次追问后，评论 + 下一题
+            history.append({"role": "assistant", "content": ""})
+            for c in comment:
+                history[-1]["content"] += c
+                await asyncio.sleep(0.03)
+
             next_question = f"{step+1}. {questions[step]}"
-            history.append({"role": "assistant", "content": next_question})
+            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 = [
-        {"role": "system", "content": f"You are Robin, a warm and engaging conversational AI. Write 1–2 follow-up questions. Also comment briefly in {language}."},
+        {"role": "system", "content": f"Write a short, warm comment in {language} responding to the user's last answer."},
         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()
+    comment_response = client.chat.completions.create(
+        model="gpt-3.5-turbo",
+        messages=comment_prompt,
+        temperature=0.7
+    )
+    comment = 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
 
-    followup_count = 2 if random.random() < 0.1 else 1 if random.random() < 0.5 else 0
     if followup_count > 0:
-        followup_prompt = [{"role": "system", "content": f"Ask {followup_count} follow-up question(s) in {language}."}, 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": bot_reply})
+        followup_prompt = [
+            {"role": "system", "content": f"Ask 1 open-ended 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()
+        combined = comment + "\n\n" + followup_q
+        history.append({"role": "assistant", "content": ""})
+        for c in combined:
+            history[-1]["content"] += c
+            await asyncio.sleep(0.03)
         return history, "", step, language, questions, True, 1
 
-    history.append({"role": "assistant", "content": bot_reply})
+    # 无追问：评论一泡，下一题一泡
+    history.append({"role": "assistant", "content": ""})
+    for c in comment:
+        history[-1]["content"] += c
+        await asyncio.sleep(0.03)
+
     next_question = f"{step+1}. {questions[step]}"
-    history.append({"role": "assistant", "content": next_question})
+    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
 
 def init():
-    return [{"role": "assistant", "content": WELCOME_MESSAGE_EN}], 0, "", [], False, 0
+    return [{"role": "assistant", "content": WELCOME_MESSAGE_EN}], 0, "", [], False
+
+with gr.Blocks(css="""
+.send-btn {
+  background-color: #25D366 !important;
+  color: white !important;
+  border: none;
+  border-radius: 24px;
+  font-size: 20px;
+  padding: 8px 20px;
+  height: 48px;
+  width: 60px;
+  margin-left: 8px;
+  cursor: pointer;
+}
+#chatbox .avatar-container {
+    width: 64px !important;
+    height: 64px !important;
+    min-width: 64px !important;
+    min-height: 64px !important;
+    background-size: 100% 100% !important;
+    background-position: center center !important;
+    border-radius: 50% !important;
+    padding: 0 !important;
+    margin: 0 !important;
+    border: none !important;
+    box-shadow: none !important;
+    background-color: transparent !important;
+}
+#chatbox .message.user { background-color: #DCF8C6 !important; }
+#chatbox .message.assistant { background-color: #ffffff !important; }
+footer { display: none !important; }
+.svelte-1ipelgc { display: none !important; }
+.icon-button-wrapper {
+  display: none !important;
+}
+label.svelte-1to105q {
+  display: none !important;
+}
+""") as demo:
+    gr.HTML("""
+    <script>
+      const waitForChatbox = () => {
+        const chatbox = document.getElementById("chatbox");
+        if (chatbox) {
+          const observer = new MutationObserver(() => {
+            chatbox.scrollTop = chatbox.scrollHeight;
+          });
+          observer.observe(chatbox, { childList: true, subtree: true });
+        } else {
+          setTimeout(waitForChatbox, 500);
+        }
+      };
+      waitForChatbox();
+    </script>
+    """)
+
+    chatbot = gr.Chatbot(
+        elem_id="chatbox",
+        label="",
+        avatar_images=["user_avatar.jpg", "humanlike_avatar.png"],
+        bubble_full_width=False,
+        height=500,
+        show_copy_button=False,
+        type="messages"
+    )
 
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(label="", avatar_images=["user_avatar.jpg", "humanlike_avatar.png"], height=500, type="messages")
     with gr.Row():
-        user_input = gr.Textbox(show_label=False, placeholder="Type your message here...", container=True, scale=10)
+        user_input = gr.Textbox(
+            show_label=False,
+            placeholder="Type your message here and press send...",
+            container=True,
+            scale=10
+        )
         send_btn = gr.Button(value="➤", elem_classes="send-btn")
 
     state = gr.State([])
@@ -141,11 +284,11 @@ with gr.Blocks() as demo:
     language = gr.State("")
     questions = gr.State([])
     followup_mode = gr.State(False)
-    followup_stage = gr.State(0)
 
-    demo.load(fn=init, outputs=[chatbot, step, language, questions, followup_mode, followup_stage])
-    user_input.submit(respond, [user_input, state, step, language, questions, followup_mode, followup_stage],
-                      [chatbot, user_input, step, language, questions, followup_mode, followup_stage])
-    send_btn.click(respond, [user_input, state, step, language, questions, followup_mode, followup_stage],
-                   [chatbot, user_input, step, language, questions, followup_mode, followup_stage])
+    demo.load(fn=init, outputs=[chatbot, step, language, questions, followup_mode])
+    user_input.submit(respond, [user_input, state, step, language, questions, followup_mode],
+                      [chatbot, user_input, step, language, questions, followup_mode])
+    send_btn.click(respond, [user_input, state, step, language, questions, followup_mode],
+                   [chatbot, user_input, step, language, questions, followup_mode])
+
     demo.queue().launch()