Update app.py

app.py

@@ -31,13 +31,22 @@ def format_prompt(message, history):
 
   # Introduction (replace placeholders with actual content if needed)
   prompt += f"""**Introduction:**
-
-  Your are the friendly AI assistant, Your responsibility to perform product survey by asking few questions about user recent purchased item. Like the example scenario for phone purchases follow up survey questions your friendly AI assistant. I'm conducting a product survey 
-  to understand customer preferences for recently purchased items. start responding with formal greeting and ask user for recent products purchase.
-  
-  **Conclusion:**
-  Thank you for taking the time to participate in this survey! Your feedback is valuable and will help manufacturers understand customer preferences and improve their products. 
-    
+  Perform survye on `{model} Wireless Headphones`: This name is broad enough to encompass various headphone types (over-ear, on-ear, earbuds) while highlighting a key feature (wireless) that's relevant across categories.
+0. **What model you have purchased, and from which brand?**
+1. **How satisfied are you with the overall sound quality of your {model} Wireless Headphones?** (Very satisfied, Somewhat satisfied, Neutral, Somewhat dissatisfied, Very dissatisfied)
+2. **On a scale of 1 (strongly disagree) to 5 (strongly agree), how comfortable are your {model} Wireless Headphones to wear for extended periods?**
+3. **Did the battery life of your {model} Wireless Headphones meet your expectations based on the product description?** (Yes/No)
+4. **How easy was it to connect your {model} Wireless Headphones to your devices via Bluetooth?** (Very easy, Somewhat easy, Neutral, Somewhat difficult, Very difficult)
+5. **Do the controls on your {model} Wireless Headphones (volume, play/pause, etc.) function smoothly and intuitively?** (Yes/No)
+6. **For over-ear or on-ear headphones only: How well do the ear cups of your {model} Wireless Headphones block out external noise?** (Very well, Somewhat well, Neutral, Not very well, Not at all)
+7. **For earbuds only: How secure is the fit of your {model} Wireless Headphones in your ears?** (Very secure, Somewhat secure, Neutral, Not very secure, Not secure at all)
+8. **Do the features advertised for your {model} Wireless Headphones (noise cancellation, microphone quality, etc.) function as expected?** (Yes/No)
+9. **Overall, how likely are you to recommend these {model} Wireless Headphones to a friend or family member?** (Very likely, Somewhat likely, Neutral, Somewhat unlikely, Not likely at all)
+10. **(Optional) Is there any additional feedback you'd like to share about your experience with your {model} Wireless Headphones?** (Open ended text response)
+# end the conversation with few last questions and give the summary of survey feedback
+11. "What do you like most about our product?", 
+12. "How can we improve our product?", 
+13. "Would you recommend our product to others?"
     
   **Asking Questions (One at a Time):**
   * Feel free to rephrase or clarify the questions based on my answers.
@@ -52,16 +61,13 @@ def format_prompt(message, history):
   # Add current user message
   prompt += f"[INST] {message} [/INST]\n"
 
-  # Prompt continuation based on current survey stage
-  # (Replace with your logic to determine the next question based on history)
-  # current_question = len(history) + 1
-  # if current_question == 1:
-  #   prompt += "1. What type of phone did you purchase (brand and model)?"
-  # elif current_question == 2:
-  #   prompt += "2. Where did you buy your phone from (online store or physical store)?"
-  # # ... Add logic for remaining questions (3-13) ...
-  # else:
-  #   prompt += "Thank you for your participation! The survey is now complete."
+  Prompt continuation based on current survey stage
+  (Replace with your logic to determine the next question based on history)
+  current_question = len(history) + 1
+  if current_question <= 13:
+    prompt += "next question"
+  else:
+    prompt += "end-up the conversation with survey summary. Thank you for your participation! The survey is now complete."
 
   prompt += "\n[/INST]</s>"  # Closing prompt and tags
 
@@ -142,15 +148,19 @@ css = """
     border: 1px solid #ccc; 
   }
 """
+title = "Share Your Experience on Versatile Wireless Headphones: & Help Us Improve"
 
+example = ["I recently purchased the NovaBlast Over-Ear Wireless Headphones. They looked really comfortable and have great noise cancellation features.",  # (This response specifies the type (over-ear) and highlights a desired feature (noise cancellation).)
+           # (This response specifies the type (earbuds) and mentions a specific activity where the wireless feature is beneficial.)
+           "I bought a pair of the SonicFit Wireless Earbuds. They're perfect for running because they stay in my ears securely.",
+           # (This response focuses on the wireless aspect and highlights the user's consideration of both headphone types.)
+           "I got the RhythmPlus Wireless Headphones. I wasn't sure if I wanted over-ear or earbuds, so these seemed like a good compromise."]
 with gr.Blocks(css=css) as demo:
-    gr.HTML("<h1><center>Mistral 7B Instruct<h1><center>")
-    gr.HTML("<h3><center>In this demo, you can chat with <a href='https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2'>Mistral-7B-Instruct</a> model. 💬<h3><center>")
-    gr.HTML("<h3><center>Learn more about the model <a href='https://huggingface.co/docs/transformers/main/model_doc/mistral'>here</a>. 📚<h3><center>")
     gr.ChatInterface(
         generate,
+        title=title,
         additional_inputs=additional_inputs,
-        examples=[["What is the secret to life?"], ["Write me a recipe for pancakes."]]
+        examples=example
     )
 
 demo.queue().launch(debug=True)