Update app.py

app.py

@@ -75,6 +75,13 @@ def search_unit(unit_query, mode):
 """
     return text
 
+def search_unit_all(unit_query):
+    text_bge = search_unit(unit_query, "BGE")
+    text_snu = search_unit(unit_query, "SBERT-snunlp")
+    text_jh = search_unit(unit_query, "SBERT-jhgan")
+
+    return text_bge, text_snu, text_jh
+
 downsample_model_ori = SentenceTransformer('BAAI/bge-m3')
 downsample_model = SentenceTransformer('snunlp/KR-SBERT-V40K-klueNLI-augSTS')
 downsample_model_2 = SentenceTransformer('jhgan/ko-sbert-sts')
@@ -326,7 +333,7 @@ with gr.Blocks(css=css) as demo:
         - SBERT-snunlp : 의미 중심 검색.
         - SBERT-jhgan : 유사도 중심 검색.
         """)
-        mode = gr.Dropdown(choices={"BGE","SBERT-snunlp","SBERT-jhgan"}, label="모델을 선택하세요")
+        #mode = gr.Dropdown(choices={"BGE","SBERT-snunlp","SBERT-jhgan"}, label="모델을 선택하세요")
         gr.Markdown("                                                              ")
     with gr.Tab("KPI Pool 검색"):
         with gr.Column(elem_classes="left-padding"):
@@ -342,7 +349,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column(visible=False) as output_area:
             with gr.Group():
                 with gr.Row():
-                    kpi_checkbox = gr.CheckboxGroup(choices=[], interactive=True, elem_classes="v_check", container=False, min_width=5, scale=1)
+                    #kpi_checkbox = gr.CheckboxGroup(choices=[], interactive=True, elem_classes="v_check", container=False, min_width=5, scale=1)
 
                     with gr.Group():
                         gr.Markdown("BGE")
@@ -377,17 +384,17 @@ with gr.Blocks(css=css) as demo:
         
         #kpi_query.submit(search_kpi, inputs = [kpi_query, kpi_count, mode], outputs = [output_area, kpi_checkbox, kpi_table, df_state, check_state, copyright])
         kpi_query.submit(search_kpi_all_models, inputs = [kpi_query, kpi_count], outputs = [output_area, kpi_table1, kpi_table2, kpi_table3, copyright])
-        select_button.click(fn=toggle_selection, inputs=[kpi_checkbox, check_state], outputs=kpi_checkbox, show_progress='hidden')
+        #select_button.click(fn=toggle_selection, inputs=[kpi_checkbox, check_state], outputs=kpi_checkbox, show_progress='hidden')
         #download_button.click(
         #    generate_excel,
         #    inputs=[df_state, kpi_checkbox, kpi_query],
         #    outputs=[file_download]
         #)
         clear_button.click(
-            fn=lambda: (None, None, None, gr.update(visible=False),
+            fn=lambda: (None, None, gr.update(visible=False),
                         gr.update(choices=[], value=[]), gr.update(value=[]),gr.update(value=[]),gr.update(value=[]), 
                         gr.update(value=None, visible=False), gr.update(visible=True)),
-            outputs=[df_state, check_state, kpi_query, output_area,
+            outputs=[df_state, kpi_query, output_area,
                      kpi_checkbox, kpi_table1, kpi_table2, kpi_table3, 
                      file_download, copyright],
             show_progress='hidden'
@@ -397,8 +404,19 @@ with gr.Blocks(css=css) as demo:
     with gr.Tab("[참고] NCS 능력단위"):
         unit_query = gr.Textbox(label="업종 or 직종 + 직무명을 입력하세요😊", scale=1, submit_btn=True,
                             placeholder="예: 의약품 법률자문, 공공행정 경영기획, 재무회계 자금")
-        unit_result = gr.Markdown()
-        unit_query.submit(search_unit, inputs=[unit_query, mode], outputs=[unit_result])
+        with gr.Row():
+            with gr.Group():
+                gr.Markdown("BGE")
+                unit_result1 = gr.Markdown()
+            with gr.Group():
+                gr.Markdown("SBERT-snunlp")
+            unit_result2 = gr.Markdown()
+            with gr.Group():
+                gr.Markdown("SBERT-jhgan")
+            unit_result3 = gr.Markdown()
+            
+        #unit_query.submit(search_unit, inputs=[unit_query, mode], outputs=[unit_result])
+        unit_query.submit(search_unit_all, inputs=unit_query, outputs=[unit_result1, unit_result2, unit_result3])
 
 
 demo.launch(debug=True)