Update bin_public/app/local.py

bin_public/app/local.py

@@ -28,11 +28,12 @@ def get_file(choice):
     fs = HfFileSystem()
     with fs.open(f"spaces/AllenYkl/baikejiankang/{choice}") as f:
         data = f.read().decode()
-    return data
+    return data, choice[:-4]
 
 def info_context(file_list_m):
     print('Start sampling')
-    context = f'已知信息{get_file(randomly(file_list_m))}根据已知信息，生成服从以下格式的患者信息，且尽量精简。\n皮肤病类型：\n患者性别：\n患者年龄：\n病史：\n大致症状：'
+    doc, disease = get_file(randomly(file_list_m))
+    context = f"以下是 {disease} 的常见症状：{doc}\n\n请根据上文内容，生成一位 {disease} 患者的信息，服从以下格式：\n患者性别：\n患者年龄：\n病史：\n症状："
     print('sampling finished')
     return context