Update app.py

app.py

@@ -14,11 +14,14 @@ from datetime import datetime
 
 
 # -------------------------------------------- For Memory - you will need to set up a dataset and HF_TOKEN ---------
-#DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/ChatbotMemory.csv"
-#DATASET_REPO_ID = "awacke1/ChatbotMemory.csv"
-#DATA_FILENAME = "ChatbotMemory.csv"
-#DATA_FILE = os.path.join("data", DATA_FILENAME)
-#HF_TOKEN = os.environ.get("HF_TOKEN")
+UseMemory=True
+
+if UseMemory:
+   DATASET_REPO_URL="https://huggingface.co/datasets/awacke1/ChatbotMemory.csv"
+   DATASET_REPO_ID="awacke1/ChatbotMemory.csv"
+   DATA_FILENAME="ChatbotMemory.csv"
+   DATA_FILE=os.path.join("data", DATA_FILENAME)
+   HF_TOKEN=os.environ.get("HF_TOKEN")
 
 #SCRIPT = """
 #<script>
@@ -30,29 +33,30 @@ from datetime import datetime
 #</script>
 #"""
 
-#try:
-#    hf_hub_download(
-#        repo_id=DATASET_REPO_ID,
-#        filename=DATA_FILENAME,
-#        cache_dir=DATA_DIRNAME,
-#        force_filename=DATA_FILENAME
-#    )
-#except:
-#    print("file not found")
-#repo = Repository(
-#    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-#)
+if UseMemory: 
+   try:
+      hf_hub_download(
+      repo_id=DATASET_REPO_ID,
+      filename=DATA_FILENAME,
+      cache_dir=DATA_DIRNAME,
+      force_filename=DATA_FILENAME
+      )
+   except:
+      print("file not found")
+      repo = Repository(
+      local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+      )
             
-#def store_message(name: str, message: str):
-#    if name and message:
-#        with open(DATA_FILE, "a") as csvfile:
-#            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
-#            writer.writerow(
-#                {"name": name.strip(), "message": message.strip(), "time": str(datetime.now())}
-#            )
+def store_message(name: str, message: str):
+    if name and message:
+        with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
+            writer.writerow(
+                {"name": name.strip(), "message": message.strip(), "time": str(datetime.now())}
+            )
 # uncomment line below to begin saving.  If creating your own copy you will need to add a access token called "HF_TOKEN" to your profile, then create a secret for your repo with the access code naming it "HF_TOKEN"  For the CSV as well you can copy the header and first few lines to your own then update the paths above which should work to save to your own repository for datasets.    
-#        commit_url = repo.push_to_hub()
-#    return ""
+        commit_url = repo.push_to_hub()
+    return ""
     
 #iface = gr.Interface(
 #    store_message,
@@ -89,6 +93,8 @@ def add_note_to_history(note, note_history):
     note_history = '</s> <s>'.join(note_history)
     return [note_history]
 
+
+
 title = "State of the Art Chatbot with Memory Dataset"
 description = """Chatbot With Memory"""
 
@@ -106,9 +112,17 @@ def chat(message, history):
     history_useful = add_note_to_history(response, history_useful)
     list_history = history_useful[0].split('</s> <s>')
     history.append((list_history[-2], list_history[-1]))
-#    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.
+
+    # AGI semantic and episodic memory here we come....
+    
+    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
 
+
+
 gr.Interface(
     fn=chat,
     theme="huggingface",
@@ -117,8 +131,11 @@ gr.Interface(
     outputs=["chatbot", "state"],
     title=title,
     allow_flagging="never",
+
+    
     description=f"Gradio chatbot backed by memory in a dataset repository.",
-#    article=f"The dataset repo is [{DATASET_REPO_URL}]({DATASET_REPO_URL})"
+    article=f"The memory dataset for saves is [{DATASET_REPO_URL}]({DATASET_REPO_URL})  Use the dataset preview tool here to check it out: https://huggingface.co/spaces/awacke1/FreddysDatasetViewer  "
+
     ).launch(debug=True)
         
 #demo = gr.Blocks()