Update app.py

app.py

@@ -56,7 +56,7 @@ SET_III = [
 ]
 
 def generate_question_set():
-    return random.sample(SET_I, 3) + random.sample(SET_II, 3) + random.sample(SET_III, 3)
+    return random.sample(SET_I, 1) + random.sample(SET_II, 1) + random.sample(SET_III, 1)
 
 async def gpt_translate(text, target_lang):
     if target_lang == "English":
@@ -71,7 +71,6 @@ 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()
@@ -88,244 +87,54 @@ async def respond(user_input, history, step, language, questions, followup_mode,
         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],
             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": ""})
-        for c in comment:
-            history[-1]["content"] += c
-            await asyncio.sleep(0.03)
-
-        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:
-            # 回答完第二次追问，直接进入下一主问题
+        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:
             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)
+            history.append({"role": "assistant", "content": next_question})
             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."},
+        {"role": "system", "content": f"You are Robin, a warm and engaging conversational AI. Write 1–2 follow-up questions. Also comment briefly in {language}."},
         history[-2],
         history[-1]
     ]
-    comment_response = client.chat.completions.create(
-        model="gpt-3.5-turbo",
-        messages=comment_prompt,
-        temperature=0.8
-    )
+    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
-
+    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} 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
-        )
+        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": ""})
-        for c in bot_reply:
-            history[-1]["content"] += c
-            await asyncio.sleep(0.03)
+        history.append({"role": "assistant", "content": bot_reply})
         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)
-
+    history.append({"role": "assistant", "content": bot_reply})
     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)
+    history.append({"role": "assistant", "content": next_question})
     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
-
-    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
-
-
-
 def init():
-    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"
-    )
+    return [{"role": "assistant", "content": WELCOME_MESSAGE_EN}], 0, "", [], False, 0
 
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(label="", avatar_images=["user_avatar.jpg", "humanlike_avatar.png"], bubble_full_width=False, height=500)
     with gr.Row():
-        user_input = gr.Textbox(
-            show_label=False,
-            placeholder="Type your message here and press send...",
-            container=True,
-            scale=10
-        )
+        user_input = gr.Textbox(show_label=False, placeholder="Type your message here...", container=True, scale=10)
         send_btn = gr.Button(value="➤", elem_classes="send-btn")
 
     state = gr.State([])
@@ -335,19 +144,9 @@ label.svelte-1to105q {
     followup_mode = gr.State(False)
     followup_stage = gr.State(0)
 
-
-    demo.load(fn=init, outputs=[chatbot, step, language, questions, followup_mode, followup_stage])
     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.queue().launch()
+    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.queue().launch()