Update app.py

app.py

@@ -71,22 +71,29 @@ 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
 
+    # Handle follow-up answer
     if followup_mode:
         comment_prompt = [
             {"role": "system", "content": f"Write a short, empathetic comment in {language} responding to the user's last answer."},
@@ -102,8 +109,9 @@ async def respond(user_input, history, step, language, questions, followup_mode)
 
         rand = random.random()
         if rand < 0.1:
+            # Trigger second follow-up
             followup_prompt = [
-                {"role": "system", "content": f"Ask one open-ended follow-up question in {language} based on the user's previous answer."},
+                {"role": "system", "content": f"Ask one open-ended follow-up question in {language} based on the user's last answer."},
                 history[-2],
                 history[-1]
             ]
@@ -113,23 +121,32 @@ async def respond(user_input, history, step, language, questions, followup_mode)
                 temperature=0.7
             )
             followup_question = followup_response.choices[0].message.content.strip()
-            content = comment + "\n\n" + followup_question
+
+            # Comment bubble
             history.append({"role": "assistant", "content": ""})
-            for c in content:
+            for c in comment:
                 history[-1]["content"] += c
                 await asyncio.sleep(0.03)
+            # Follow-up question bubble
+            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:
-            next_q = f"{step+1}. {questions[step]}"
-            content = comment + "\n\n" + next_q
+            # No more follow-ups, move to next main question
+            next_question = f"{step+1}. {questions[step]}"
+            content = comment + "\n\n" + next_question
             history.append({"role": "assistant", "content": ""})
             for c in content:
                 history[-1]["content"] += c
                 await asyncio.sleep(0.03)
             return history, "", step + 1, language, questions, False
 
+    # Handle main question answer
     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 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]
     ]
@@ -141,7 +158,8 @@ async def respond(user_input, history, step, language, questions, followup_mode)
     comment = comment_response.choices[0].message.content.strip()
 
     rand = random.random()
-    if rand < 0.5:
+    if rand < 0.4:
+        # Trigger first follow-up
         followup_prompt = [
             {"role": "system", "content": f"Ask one open-ended follow-up question in {language} based on the user's answer."},
             history[-2],
@@ -153,15 +171,23 @@ async def respond(user_input, history, step, language, questions, followup_mode)
             temperature=0.7
         )
         followup_question = followup_response.choices[0].message.content.strip()
-        content = comment + "\n\n" + followup_question
+
+        # Comment bubble
         history.append({"role": "assistant", "content": ""})
-        for c in content:
+        for c in comment:
             history[-1]["content"] += c
             await asyncio.sleep(0.03)
+        # Follow-up question bubble
+        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:
-        next_q = f"{step+1}. {questions[step]}"
-        content = comment + "\n\n" + next_q
+        # No follow-up, move to next question
+        next_question = f"{step+1}. {questions[step]}"
+        content = comment + "\n\n" + next_question
         history.append({"role": "assistant", "content": ""})
         for c in content:
             history[-1]["content"] += c