Update app.py

app.py

@@ -54,7 +54,7 @@ NORMAL_SET_III = [
 ]
 
 def generate_question_set():
-    return random.sample(NORMAL_SET_I, 3) + random.sample(NORMAL_SET_II, 3) + random.sample(NORMAL_SET_III, 3)
+    return random.sample(SET_I, 3) + random.sample(SET_II, 3) + random.sample(SET_III, 3)
 
 async def gpt_translate(text, target_lang):
     if target_lang == "English":
@@ -69,19 +69,28 @@ async def gpt_translate(text, target_lang):
 
 async def respond(user_input, history, step, language, questions, followup_mode):
     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": ""})
+        for c in f"{welcome}\n\n1. {translated_questions[0]}":
+            history[-1]["content"] += c
+            await asyncio.sleep(0.03)
         return history, "", 1, user_language, translated_questions, False
+
     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})
+        for msg in [thank_you, end_note]:
+            history.append({"role": "assistant", "content": ""})
+            for c in msg:
+                history[-1]["content"] += c
+                await asyncio.sleep(0.03)
         return history, "", step, language, questions, False
+
     if followup_mode:
         comment_prompt = [
             {"role": "system", "content": f"Write a short, empathetic comment in {language} responding to the user's last answer."},
@@ -94,39 +103,62 @@ async def respond(user_input, history, step, language, questions, followup_mode)
             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)
-        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
+
+        rand = random.random()
+        if rand < 0.1:
+            followup_prompt = [
+                {"role": "system", "content": f"Ask one open-ended follow-up question in {language} based on the user's last answer."},
+                history[-2],
+                history[-1]
+            ]
+            followup_response = client.chat.completions.create(
+                model="gpt-3.5-turbo",
+                messages=followup_prompt,
+                temperature=0.7
+            )
+            followup_question = followup_response.choices[0].message.content.strip()
+
+            history.append({"role": "assistant", "content": ""})
+            for c in comment:
+                history[-1]["content"] += c
+                await asyncio.sleep(0.03)
+
+            history.append({"role": "assistant", "content": ""})
+            for c in followup_question:
+                history[-1]["content"] += c
+                await asyncio.sleep(0.03)
+
+            return history, "", step, language, questions, True
+        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
 
     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. Write a short, personalized comment on the user's answer in {language}. Add a light emoji at the end."},
+        {"role": "system", "content": f"Write a short, personalized comment in {language} responding to the user's answer. Add a light emoji."},
         history[-2],
         history[-1]
     ]
     comment_response = client.chat.completions.create(
         model="gpt-3.5-turbo",
         messages=comment_prompt,
-        temperature=0.8
+        temperature=0.7
     )
-    bot_reply = comment_response.choices[0].message.content.strip()
+    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
-
-    if followup_count > 0:
+    if rand < 0.4:
         followup_prompt = [
-            {"role": "system", "content": f"Ask {followup_count} open-ended follow-up question(s) based on the user's last answer in {language}."},
+            {"role": "system", "content": f"Ask one open-ended follow-up question in {language} based on the user's answer."},
             history[-2],
             history[-1]
         ]
@@ -135,27 +167,37 @@ async def respond(user_input, history, step, language, questions, followup_mode)
             messages=followup_prompt,
             temperature=0.7
         )
-        bot_reply += "\n\n" + followup_response.choices[0].message.content.strip()
+        followup_question = followup_response.choices[0].message.content.strip()
+
+        history.append({"role": "assistant", "content": ""})
+        for c in comment:
+            history[-1]["content"] += c
+            await asyncio.sleep(0.03)
+
         history.append({"role": "assistant", "content": ""})
-        for c in bot_reply:
+        for c in followup_question:
             history[-1]["content"] += c
             await asyncio.sleep(0.03)
+
         return history, "", step, language, questions, True
+    else:
+        history.append({"role": "assistant", "content": ""})
+        for c in comment:
+            history[-1]["content"] += c
+            await asyncio.sleep(0.03)
 
-    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
+        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;
@@ -187,15 +229,12 @@ with gr.Blocks(css="""
 #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>
@@ -245,4 +284,4 @@ label.svelte-1to105q {
     send_btn.click(respond, [user_input, state, step, language, questions, followup_mode],
                    [chatbot, user_input, step, language, questions, followup_mode])
 
-    demo.queue().launch()
+    demo.queue().launch()