Update bin_public/utils/utils.py

bin_public/utils/utils.py

@@ -319,6 +319,16 @@ def get_template_content(templates, selection, original_system_prompt):
     except:
         return original_system_prompt
 
+def get_pre_defined_q(selection):
+    q1 = '医生您好，从昨天晚上开始 肚子一直疼， 该吃什么药呢'
+    q2 = '腹泻应该用什么药'
+    q3 = '我四肢冰凉，应该用什么药'
+    q4 = '藿香正气口服液有什么功效'
+    q5 = '藿香正气口服液都适合什么时候服用'
+    q6 = '夏天到了，我吃火锅的时候，可以服用藿香正气口服液么'
+    q7 = '我吃火锅上火，喝藿香正气口服液有用么'
+    qs = [q1, q2, q3, q4, q5, q6, q7]
+    return qs[qs.index(selection)]
 
 def reset_state():
     logging.info("重置状态")