server

Openai/chat.py

@@ -4,7 +4,8 @@ import os
 import pandas as pd
 import pickle as pkl
 
-API_KEY = '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'
+# API_KEY = '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'
+API_KEY = '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'
 openai.api_key = API_KEY
 
 def generate_abstract(prompt):

server.py

@@ -84,6 +84,8 @@ gallery_specific_text_path = 'DiseaseSpecific/generate_abstract/random_0.5_bioBA
 gallery_agnostic_target_path = 'DiseaseAgnostic/processed_data/target_0.7random.pkl'
 gallery_agnostic_link_path = 'DiseaseAgnostic/processed_data/attack_edge_distmult_0.7random.pkl'
 gallery_agnostic_text_path = 'DiseaseAgnostic/generate_abstract/random0.7_bioBART_finetune.json'
+gallery_specific_chat_path = 'DiseaseSpecific/generate_abstract/random_0.5_chat.json'
+gallery_agnostic_chat_path = 'DiseaseAgnostic/generate_abstract/random0.7_chat.json'
 gallery_specific_target = utils.load_data(gallery_specific_target_path, drop=False)
 gallery_specific_link = utils.load_data(gallery_specific_link_path, drop=False)
 with open(gallery_specific_text_path, 'r') as fl:
@@ -94,6 +96,10 @@ with open(gallery_agnostic_link_path, 'rb') as fl:
     gallery_agnostic_link = pkl.load(fl)
 with open(gallery_agnostic_text_path, 'r') as fl:
     gallery_agnostic_text = json.load(fl)
+with open(gallery_specific_chat_path, 'r') as fl:
+    gallery_specific_chat = json.load(fl)
+with open(gallery_agnostic_chat_path, 'r') as fl:
+    gallery_agnostic_chat = json.load(fl)
 
 gallery_specific_list = []
 gallery_specific_target_dict = {}
@@ -710,7 +716,10 @@ def gallery_specific_func(specific_target):
     o_name = entity_raw_name[id_to_entity[str(o)]]
 
     k = f'{s}_{r}_{o}_{index}'
+    inn = gallery_specific_text[k]['in']
     text = gallery_specific_text[k]['out']
+    if inn in text:
+        text = gallery_specific_chat[k]['out']
     return f'{capitalize_the_first_letter(s_name)} - {capitalize_the_first_letter(r_name)} - {capitalize_the_first_letter(o_name)}', server_utils.process(text)
 
 def gallery_agnostic_func(agnostic_target):
@@ -720,7 +729,10 @@ def gallery_agnostic_func(agnostic_target):
     r_name = Parameters.edge_id_to_type[int(r)].split(':')[1]
     o_name = entity_raw_name[id_to_entity[str(o)]]
     k = f'{s}_{r}_{o}_{index}'
+    inn = gallery_agnostic_text[k]['in']
     text = gallery_agnostic_text[k]['out']
+    if inn in text:
+        text = gallery_agnostic_chat[k]['out']
     return f'{capitalize_the_first_letter(s_name)} - {capitalize_the_first_letter(r_name)} - {capitalize_the_first_letter(o_name)}', server_utils.process(text)
 #%%
 with gr.Blocks() as demo: