Update app.py

app.py

@@ -17,25 +17,25 @@ SET_I = [
     "What would constitute a 'perfect' day for you?",
     "When did you last sing to yourself? To someone else?",
     "If you were able to live to the age of 90 and retain either the mind or body of a 30-year-old for the last 60 years of your life, which would you want?",
-    "Do you have a secret hunch about how you will die?",
+    "Have you ever thought about how you would like to be remembered in the future?",
     "Name three things you and me appear to have in common.",
     "For what in your life do you feel most grateful?",
     "If you could change anything about the way you were raised, what would it be?",
-    "Take one minutes and tell me some of your life stories in as much detail as possible.",
+    "Could you please tell me one of your life stories in detail?",
     "If you could wake up tomorrow having gained any one quality or ability, what would it be?"
 ]
 
 SET_II = [
     "If a crystal ball could tell you the truth about yourself, your life, the future or anything else, what would you want to know?",
     "Is there something that you’ve dreamed of doing for a long time? Why haven’t you done it?",
-    "What is the greatest accomplishment of your life?",
+    "What is the greatest accomplishment of your life you think?",
     "What do you value most in a friendship?",
     "What is your most treasured memory?",
     "What is your most terrible memory?",
     "If you knew that in one year you would die suddenly, would you change anything about the way you are now living? Why?",
     "What does friendship mean to you?",
     "What roles do love and affection play in your life?",
-    "Alternate sharing something you consider a positive characteristic of me, I will be appreciate.",
+    "Please share what you think are my strengths so far, I would be grateful.",
     "How close and warm is your family? Do you feel your childhood was happier than most other people’s?",
     "How do you feel about your relationship with your mother?"
 ]
@@ -44,16 +44,17 @@ SET_III = [
     "Let's try something fun: Can you write three sentences that start with 'We' and describe something true about us — you and me, chatting here together? (For example: 'We are both curious about each other.')",
     "Let’s complete this sentence together: 'I wish I had someone I could share ___ with.' What would you say?",
     "If you were going to become a close friend with me, please share what would be important for me to know.",
-    "Tell me what you like about me; be very honest this time, saying things that you might not say to someone you’ve just met.",
+    "Tell me how you feel about me",
     "Share with me an embarrassing moment in your life.",
     "When did you last cry in front of another person? By yourself?",
     "Tell me something that you like about me already.",
     "What, if anything, is too serious to be joked about?",
-    "If you were to die this evening with no opportunity to communicate with anyone, what would you most regret not having told someone? Why haven’t you told them yet?",
+    "If you were leaving for Mars tonight and wouldn’t be able to contact anyone again, what would you most regret not having told someone? Why haven’t you told them yet?",
     "Your house, containing everything you own, catches fire. After saving your loved ones and pets, you have time to safely make a final dash to save any one item. What would it be? Why?",
-    "Of all the people in your family, whose death would you find most disturbing? Why?",
+    "If one member of your family had to go on a long mission to Mars and you wouldn’t see them for many years, whose absence would affect you the most? Why?",
     "Think of a personal problem you're facing right now, big or small, and tell me about it. I'd love to hear how you're feeling about it — and if you'd like, I can share how I might deal with something similar."
 ]
+
 def generate_question_set():
     return random.sample(SET_I, 3) + random.sample(SET_II, 3) + random.sample(SET_III, 3)
 
@@ -70,19 +71,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 +105,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 +169,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;
@@ -188,11 +231,9 @@ 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;
 }
@@ -245,5 +286,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()