Upload app.py

app.py

@@ -10,15 +10,15 @@ DEFAULT_LANGUAGE = "English"
 WELCOME_MESSAGE_EN = "Hi! I'm your friendly assistant 🤖 Let's begin!\n\nPlease tell me which language you'd like to use."
 
 QUESTION_TEMPLATES = [
-    "Given the choice of anyone in the world, whom would you want as a dinner guest.",
-    "Would you like to be famous? In what way?",
-    "Before making a telephone call, do you ever rehearse what you are going to say? Why?",
-    "If a crystal ball could tell you the truth about yourself, your life, the future or anything else, what would you want to know?",
-    "What do you value most in a friendship?",
-    "What is your most terrible memory?",
+    "Please share a memorable shopping experience you had recently.",
+    "What factors do you value most when shopping? (e.g., price, quality, service)",
+    "How do promotions influence your buying behavior?",
+    "Have you ever purchased something because of a recommendation on social media?",
+    "Do you prefer online or in-store shopping? Why?",
+    "Have you used an AI shopping assistant before? Was it helpful?",
     "What improvements would you like to see in future shopping experiences?",
-    "If you were going to become a close friend with your partner, please share what would be important for him or her to know.",
-    "When did you last cry in front of another person? By yourself?"
+    "Have you ever encountered shopping traps or unpleasant experiences?",
+    "Describe your ideal shopping experience in three words."
 ]
 
 BOT_COMMENTS_EN = [
@@ -28,6 +28,12 @@ BOT_COMMENTS_EN = [
     "That’s so real—I get what you mean!"
 ]
 
+BOT_NAME_PREFIX = {
+    "English": "🤖 Assistant:",
+    "中文": "🤖 小秘书：",
+    "Deutsch": "🤖 Assistent:"
+}
+
 async def gpt_translate(text, target_lang):
     if target_lang == "English":
         return text
@@ -39,29 +45,30 @@ async def gpt_translate(text, target_lang):
     )
     return response.choices[0].message.content.strip()
 
-async def respond(user_input, chat_history, step, language, questions):
+async def respond(user_input, history, step, language, questions):
     if language == "":
         user_language = user_input.strip().capitalize()
         translated_questions = [await gpt_translate(q, user_language) for q in QUESTION_TEMPLATES]
         welcome = await gpt_translate("Great! We will now continue in your selected language.", user_language)
-        chat_history.append((user_input, None))
-        chat_history.append((None, f"{welcome}\n\n1. {translated_questions[0]}"))
-        return chat_history, "", 1, user_language, translated_questions
+        history.append({"role": "user", "content": user_input})
+        history.append({"role": "assistant", "content": f"{welcome}\n\n1. {translated_questions[0]}"} )
+        return history, "", 1, user_language, translated_questions
 
     if step >= len(questions):
+        history.append({"role": "user", "content": user_input})
         thank_you = await gpt_translate("Thank you for your response!", language)
         end_note = await gpt_translate("This concludes our questions. Please proceed with the rest of the survey. 📝", language)
-        chat_history.append((user_input, None))
-        chat_history.append((None, thank_you))
-        chat_history.append((None, end_note))
-        return chat_history, "", step, language, questions
+        history.append({"role": "assistant", "content": thank_you})
+        history.append({"role": "assistant", "content": end_note})
+        return history, "", step, language, questions
 
-    chat_history.append((user_input, None))
+    history.append({"role": "user", "content": user_input})
 
-    # GPT 正式回应（+ 强制分气泡）
+    # GPT 主回答
     messages = [
         {"role": "system", "content": f"You are a friendly assistant. Provide a short, positive comment on the user's answer in {language}, and include a light emoji. Do not ask a new question."},
-        {"role": "user", "content": user_input}
+        history[-2],
+        history[-1]
     ]
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",
@@ -69,25 +76,23 @@ async def respond(user_input, chat_history, step, language, questions):
         temperature=0.7
     )
     bot_reply = response.choices[0].message.content.strip()
-    await asyncio.sleep(0.3)
-    chat_history.append((None, bot_reply))
-    chat_history.append((" ", None))  # 用空白用户消息打断气泡
+    history.append({"role": "assistant", "content": bot_reply})
+
+    # ✅ 插入空白用户消息用于“断开”气泡
+    history.append({"role": "user", "content": " "})
 
     if random.random() < 0.4:
         extra_comment = await gpt_translate(random.choice(BOT_COMMENTS_EN), language)
-        await asyncio.sleep(0.3)
-        chat_history.append((None, f"🤖 {extra_comment}"))
-        chat_history.append((" ", None))  # 再插入打断
-
-    await asyncio.sleep(0.3)
-    next_question = f"{step+1}. {questions[step]}"
-    chat_history.append((None, next_question))
+        history.append({"role": "assistant", "content": f"{BOT_NAME_PREFIX.get(language, '🤖')}{extra_comment}"})
+        history.append({"role": "user", "content": " "})
 
+    # 下一题
+    history.append({"role": "assistant", "content": f"{step+1}. {questions[step]}"})
     step += 1
-    return chat_history, "", step, language, questions
+    return history, "", step, language, questions
 
 def init():
-    return [], 0, "", []
+    return [{"role": "assistant", "content": WELCOME_MESSAGE_EN}], 0, "", []
 
 with gr.Blocks(css="""
 #chatbox .message.user {
@@ -147,7 +152,7 @@ with gr.Blocks(css="""
         bubble_full_width=False,
         height=500,
         show_copy_button=False,
-        type="tuples"
+        type="messages"
     )
 
     with gr.Row():