Update app.py

app.py

@@ -11,51 +11,50 @@ WELCOME_MESSAGE_EN = "Hi! I'm Robin, your dialogue partner.\nWe're about to begi
 
 # === Normal Question Bank ===
 NORMAL_SET_I = [
-    "When was the last time you walked for more than an hour? Describe where you went and what you saw.",
+    "When was the last time you walked for more than an hour? Please describe what you saw.",
     "What is the best gift you have ever received? Why?",
     "If you had to leave your current city, where would you move to? What do you miss most about your current city?",
-    "How did you celebrate Halloween last year?",
-    "Do you read newspapers often? Which newspapers do you like? Why?",
+    "How did you celebrate the New Year last year?",
+    "Do you read newspapers often? Why?",
     "What is the ideal number of students to share a house with? Why?",
     "If you could invent a new flavor of ice cream, what flavor would you create?",
     "What is the best restaurant you have been to in the past month? Please describe it.",
-    "Describe the pet you last got.",
+    "Please describe the pet you last got.",
     "What is your favorite holiday? Why?",
-    "Tell me the funniest thing that happened when you were with a child."
+    "Can you tell me something funny that happened when you were spending time with a child?"
 ]
 NORMAL_SET_II = [
     "What gift did you get on your last birthday?",
-    "Describe your last trip to the zoo.",
+    "Please tell me something about your last trip to the zoo.",
     "Is there someone in your family you feel especially close to? What makes your bond strong?",
-    "Let’s play a quick word game! I’ll say a word, and you respond with a word that starts with the last letter of mine. We’ll go back and forth for a few rounds — no need to form sentences, just have fun!",
+    "Imagine we were both stranded on a tropical island. What would be the first thing you would do?",
     "Do you like to wake up early or stay up late? Has anything interesting ever happened to you because of this?",
-    "Where are you from? Tell me some places you have lived.",
+    "If you could choose to live anywhere, what city or environment would you choose?",
     "What is your favorite subject in school and why?",
     "What is your plan for the next holiday?",
-    "What gifts did you receive last Christmas/Hanukkah?",
+    "What gifts did you receive for Christmas or New Year last year?",
     "Who is your favorite actor or actress? ",
     "What was your impression of your university when you first came to it?",
-    "What is the best TV show you have seen in the past month? Describe it a little bit.",
+    "What is the best TV show you have seen in the past month? Please describe it a little bit.",
     "What is your favorite holiday? Why?"
 ]
 NORMAL_SET_III = [
-    "Where did you go to high school? What was it like?",
-    "What is the best book you have read in the past three months? Describe it a little bit.",
+    "What do you remember most about your time in high school?",
+    "What is the best book you have read in the past three months? ",
     "What country would you most like to visit? What attracts you to it?",
     "Do you prefer digital or analog watches/clocks? Why?",
     "Can you think of someone your mother or father is really close to? What do you think makes their friendship special?"
     "What are the pros and cons of artificial Christmas trees?",
-    "How often do you get your hair cut? Where do you go? Have you ever had a bad haircut?",
+    "How often do you get your hair cut? Have you ever had a bad haircut?",
     "Did you have a class pet in elementary school? Do you remember its name?",
     "Do you think left-handed people are more creative than right-handed people?",
-    "What was the last concert you attended? How many albums of the band do you have? Have you seen them perform before? Where?",
+    "What was the last concert you attended? How was the concert?",
     "What magazines do you subscribe to? What have you subscribed to in the past?",
-    "Have you ever participated in a school play? What role did you play? What was the plot? Did anything funny happen during the play?"
+    "Have you ever participated in a school play? What role did you play? Did anything funny happen during the play?"
 ]
 
-
 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":
@@ -70,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."},
@@ -95,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]
         ]
@@ -136,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)
+
+        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": ""})
-    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
+        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;
@@ -186,19 +227,14 @@ with gr.Blocks(css="""
 }
 #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>
@@ -248,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()