Update app.py

app.py

@@ -40,11 +40,11 @@ if UseMemory:
       local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
       )
 
-def get_df(name: str):
+async def get_df(name: str):
     dataset = load_dataset(str, split="train")
     return dataset
     
-def store_message(name: str, message: str):
+async def store_message(name: str, message: str):
     if name and message:
         with open(DATA_FILE, "a") as csvfile:
             writer = csv.DictWriter(csvfile, fieldnames=[ "time", "message", "name", ])
@@ -59,14 +59,14 @@ def store_message(name: str, message: str):
 
         # f=get_df(DATASET_REPO_ID)
         # print(f)
-    return ""
+    return commit_url
 # ----------------------------------------------- For Memory
     
 mname = "facebook/blenderbot-400M-distill"
 model = BlenderbotForConditionalGeneration.from_pretrained(mname)
 tokenizer = BlenderbotTokenizer.from_pretrained(mname)
 
-def take_last_tokens(inputs, note_history, history):
+async def take_last_tokens(inputs, note_history, history):
     """Filter the last 128 tokens"""
     if inputs['input_ids'].shape[1] > 128:
         inputs['input_ids'] = torch.tensor([inputs['input_ids'][0][-128:].tolist()])
@@ -75,7 +75,7 @@ def take_last_tokens(inputs, note_history, history):
         history = history[1:]
     return inputs, note_history, history
     
-def add_note_to_history(note, note_history):
+async def add_note_to_history(note, note_history):
     """Add a note to the historical information"""
     note_history.append(note)
     note_history = '</s> <s>'.join(note_history)
@@ -85,7 +85,7 @@ title = "💬ChatBack🧠💾"
 description = """Chatbot With persistent memory dataset allowing multiagent system AI to access a shared dataset as memory pool with stored interactions. 
  Current Best SOTA Chatbot:  https://huggingface.co/facebook/blenderbot-400M-distill?text=Hey+my+name+is+ChatBack%21+Are+you+ready+to+rock%3F  """
 
-def chat(message, history):
+async def chat(message, history):
     history = history or []
     if history: 
         history_useful = ['</s> <s>'.join([str(a[0])+'</s> <s>'+str(a[1]) for a in history])]
@@ -101,6 +101,11 @@ def chat(message, history):
     history.append((list_history[-2], list_history[-1]))  
     ret = store_message(message, response) # Save to dataset  -- uncomment with code above, create a dataset to store and add your HF_TOKEN from profile to this repo to use.
     return history, history
+    
+#with gr.Blocks() as demo:
+#    gr.Markdown("<h1><center>🥫Datasets🎨</center></h1>")
+#    gr.Markdown("""<div align="center">Curated Datasets: <a href = "https://www.kaggle.com/datasets">Kaggle</a>. <a href="https://www.nlm.nih.gov/research/umls/index.html">NLM UMLS</a>.  <a href="https://loinc.org/downloads/">LOINC</a>. <a href="https://www.cms.gov/medicare/icd-10/2022-icd-10-cm">ICD10 Diagnosis</a>. <a href="https://icd.who.int/dev11/downloads">ICD11</a>.  <a href="https://paperswithcode.com/datasets?q=medical&v=lst&o=newest">Papers,Code,Datasets for SOTA in Medicine</a>.   <a href="https://paperswithcode.com/datasets?q=mental&v=lst&o=newest">Mental</a>.  <a href="https://paperswithcode.com/datasets?q=behavior&v=lst&o=newest">Behavior</a>. <a href="https://www.cms.gov/medicare-coverage-database/downloads/downloads.aspx">CMS Downloads</a>.  <a href="https://www.cms.gov/medicare/fraud-and-abuse/physicianselfreferral/list_of_codes">CMS CPT and HCPCS Procedures and Services</a>  """)
+
 
 gr.Interface(
     fn=chat,