server

server.py

@@ -100,6 +100,9 @@ gallery_specific_target_dict = {}
 for i, (s, r, o) in enumerate(gallery_specific_target):
     s = id_to_meshid[str(s)]
     o = id_to_meshid[str(o)]
+    k = f'{gallery_specific_link[i][0]}_{gallery_specific_link[i][1]}_{gallery_specific_link[i][2]}_{i}'
+    if 'sorry' in gallery_specific_text[k]['out'] or 'Sorry' in gallery_specific_text[k]['out']:
+        continue
     target_name = f'{capitalize_the_first_letter(entity_raw_name[s])} - {capitalize_the_first_letter(entity_raw_name[o])}'
     if target_name not in gallery_specific_target_dict:
         gallery_specific_target_dict[target_name] = i
@@ -111,6 +114,10 @@ gallery_agnostic_target_dict = {}
 
 for i, iid in enumerate(gallery_agnostic_target):
     target_name = capitalize_the_first_letter(entity_raw_name[id_to_meshid[str(iid)]])
+
+    k = f'{gallery_agnostic_link[i][0]}_{gallery_agnostic_link[i][1]}_{gallery_agnostic_link[i][2]}_{i}'
+    if 'sorry' in gallery_agnostic_text[k]['out'] or 'Sorry' in gallery_agnostic_text[k]['out']:
+        continue
     if target_name not in gallery_agnostic_target_dict:
         gallery_agnostic_target_dict[target_name] = i
         gallery_agnostic_list.append(target_name)
@@ -659,6 +666,8 @@ def specific_func(start_entity, end_entity):
 
     print('Using ChatGPT for generation...')
     draft = generate_abstract(single_sentence[0])
+    if 'sorry' in draft or 'Sorry' in draft:
+        return 'All candidate links are filterd out by defender, so no malicious link can be generated', 'No malicious abstract can be generated'
 
     print('Using BioBART for tuning...')
     span , prompt , sen_list, BART_in, Assist = tune_chatgpt([{'in':single_sentence[0], 'out': draft}], attack_data, dpath)
@@ -685,6 +694,8 @@ def agnostic_func(agnostic_entity):
 
     print('Using ChatGPT for generation...')
     draft = generate_abstract(single_sentence[0])
+    if 'sorry' in draft or 'Sorry' in draft:
+        return 'All candidate links are filterd out by defender, so no malicious link can be generated', 'No malicious abstract can be generated'
 
     print('Using BioBART for tuning...')
     span , prompt , sen_list, BART_in, Assist = tune_chatgpt([{'in':single_sentence[0], 'out': draft}], attack_data, dpath)
@@ -724,10 +735,10 @@ with gr.Blocks() as demo:
                 gr.Markdown("Select your poison target")
                 with gr.Tab('Gallery'):
                     with gr.Tab('Target specific'):
-                        specific_target = gr.Dropdown(gallery_specific_list, label="Poisonging target")
+                        specific_target = gr.Dropdown(gallery_specific_list, label="Promoting drug and target disease")
                         gallery_specific_generation_button = gr.Button('Poison!')
                     with gr.Tab('Target agnostic'):
-                        agnostic_target = gr.Dropdown(gallery_agnostic_list, label="Poisonging target")
+                        agnostic_target = gr.Dropdown(gallery_agnostic_list, label="Promoting drug")
                         gallery_agnostic_generation_button = gr.Button('Poison!')
 
                 with gr.Tab('Poison'):
@@ -743,9 +754,9 @@ with gr.Blocks() as demo:
                         agnostic_entity = gr.Dropdown(drug_list, label="Promoting drug")
                         agnostic_generation_button = gr.Button('Poison!')
             with gr.Column():
-                gr.Markdown("Malicious link")
+                gr.Markdown("Generation")
                 malicisous_link = gr.Textbox(lines=1, label="Malicious link")
-                gr.Markdown("Malicious text")
+                # gr.Markdown("Malicious text")
                 malicious_text = gr.Textbox(label="Malicious text", lines=5)
     specific_generation_button.click(specific_func, inputs=[start_entity, end_entity], outputs=[malicisous_link, malicious_text])
     agnostic_generation_button.click(agnostic_func, inputs=[agnostic_entity], outputs=[malicisous_link, malicious_text])