Update app.py

app.py

@@ -93,7 +93,6 @@ async def respond(user_input, history, step, language, questions, followup_mode)
                 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."},
@@ -109,7 +108,6 @@ 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 last answer."},
                 history[-2],
@@ -122,12 +120,11 @@ async def respond(user_input, history, step, language, questions, followup_mode)
             )
             followup_question = followup_response.choices[0].message.content.strip()
 
-            # Comment bubble
             history.append({"role": "assistant", "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
@@ -135,16 +132,19 @@ async def respond(user_input, history, step, language, questions, followup_mode)
 
             return history, "", step, language, questions, True
         else:
-            # No more follow-ups, move to next main question
+            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]}"
-            content = comment + "\n\n" + next_question
             history.append({"role": "assistant", "content": ""})
-            for c in content:
+            for c in next_question:
                 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"Write a short, personalized comment in {language} responding to the user's answer. Add a light emoji."},
         history[-2],
@@ -159,7 +159,6 @@ async def respond(user_input, history, step, language, questions, followup_mode)
 
     rand = random.random()
     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],
@@ -172,12 +171,11 @@ async def respond(user_input, history, step, language, questions, followup_mode)
         )
         followup_question = followup_response.choices[0].message.content.strip()
 
-        # Comment bubble
         history.append({"role": "assistant", "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
@@ -185,13 +183,17 @@ async def respond(user_input, history, step, language, questions, followup_mode)
 
         return history, "", step, language, questions, True
     else:
-        # No follow-up, move to next question
+        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]}"
-        content = comment + "\n\n" + next_question
         history.append({"role": "assistant", "content": ""})
-        for c in content:
+        for c in next_question:
             history[-1]["content"] += c
             await asyncio.sleep(0.03)
+
         return history, "", step + 1, language, questions, False
 
 def init():