add project files

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from src.check_backpack import sort_yaocai
-from src.gr_func import init,get_medicines,get_first_material,get_second_material,get_possible_material
+from src.gr_func import init,get_medicines,get_first_material,get_second_material,get_possible_material,get_basename
 
 medicine_list_init = init()
 
@@ -23,13 +23,23 @@ def medicine_select_acc_change(medicine_select_acc):
 def run_btn_click(medicine_select,material_1_select,material_2_select):
     rtn = medicine_select+"\n"
     possible_material_list = get_possible_material(medicine_select,material_1_select,material_2_select)
+    if len(possible_material_list) == 1:
+        main_material, auxi_material, material_third_list = possible_material_list[0]
+        rtn += f"""###
+- **主药**:{main_material}
+- **辅药**:{auxi_material}
+"""
+        peifang = f"配方：主药{get_basename(main_material)}药引{get_basename(material_third_list[0])}辅药{get_basename(auxi_material)}丹炉陨铁炉"
+        return rtn,gr.Radio.update(choices=material_third_list,value=material_third_list[0],visible=True),gr.Markdown.update(visible=True,value=peifang),(main_material,auxi_material)
+    # else:
+
     for index,(main_material,auxi_material,material_third_list) in enumerate(possible_material_list):
         rtn += f"""### 选择{index+1}
 - **主药**:{main_material}
 - **药引**:{",".join(material_third_list)}
 - **辅药**:{auxi_material}
 """
-    return rtn
+    return rtn,gr.Radio.update(visible=False),gr.Markdown.update(visible=False,value=""),(main_material,auxi_material)
 
 def medicine_select_change(medicine_select):
     a = get_first_material(medicine_select)
@@ -51,6 +61,10 @@ def material_2_grade_select_change(medicine_select,material_1_select,material_2_
     a = get_second_material(medicine_select,material_1_select,material_2_grade_select,material_2_num)
     return gr.Dropdown.update(choices=["无"]+a, value="无", visible=True)
 
+def output_Radio_change(output_state,output_Radio):
+    main_material, auxi_material = output_state
+    return f"配方：主药{get_basename(main_material)}药引{get_basename(output_Radio)}辅药{get_basename(auxi_material)}丹炉陨铁炉"
+
 with gr.Blocks() as demo:
     with gr.Tab("丹药配方"):
         gr.Markdown("选择你要炼制的丹药")
@@ -71,8 +85,12 @@ with gr.Blocks() as demo:
                     material_2_select = gr.Dropdown(visible=False,label="第二个药材")
 
                 run_btn = gr.Button("Run")
-
-            output_mk = gr.Markdown("输出结果")
+            with gr.Column():
+                output_mk = gr.Markdown("输出结果")
+                with gr.Blocks():
+                    output_Radio = gr.Radio(visible=False,label="药引")
+                    output_state = gr.State((None,None))
+                    output_end = gr.Markdown(visible=False)
 
     with gr.Tab("背包查询"):
         gr.Markdown("复制全部药材到左边文本框")
@@ -104,7 +122,9 @@ with gr.Blocks() as demo:
     material_2_grade_select.change(fn=material_2_grade_select_change,inputs=[medicine_select, material_1_select, material_2_grade_select, material_2_num],outputs=[material_2_select])
     material_2_num.change(fn=material_2_grade_select_change,inputs=[medicine_select, material_1_select, material_2_grade_select, material_2_num],outputs=[material_2_select])
 
-    run_btn.click(fn=run_btn_click,inputs=[medicine_select,material_1_select,material_2_select],outputs=[output_mk])
+    run_btn.click(fn=run_btn_click,inputs=[medicine_select,material_1_select,material_2_select],outputs=[output_mk,output_Radio,output_end,output_state])
+
+    output_Radio.change(fn=output_Radio_change,inputs=[output_state,output_Radio],outputs=[output_end])
 
     # 背包
     medicine_select_acc_b.change(fn=medicine_select_acc_change_b, inputs=[medicine_select_acc_b],

src/gr_func.py

@@ -163,6 +163,9 @@ def get_possible_material(medicine_select, first_material:str="无", second_mate
         rtn.append((main_material,auxi_material,material_third_list))
     return rtn
 
+def get_basename(text):
+    name,num = text.split("*")
+    return name[:-4]+num
 
 
 def init():