Update app.py

app.py

@@ -7,6 +7,7 @@ import langchain
 import chromadb
 import glob
 import pickle
+from datetime import datetime
 
 from PyPDF2 import PdfReader
 from PyPDF2 import PdfWriter
@@ -28,6 +29,18 @@ os.environ["OPENAI_API_KEY"] = os.environ['openai_key']
 prompt_templates = {"All Needs Experts": "Respond as if you are combination of all needs assessment experts."}
 actor_description = {"All Needs Experts": "<div style='float: left;margin: 0px 5px 0px 5px;'><img src='https://na.weshareresearch.com/wp-content/uploads/2023/04/experts2.jpg' alt='needs expert image' style='width:70px;align:top;'></div>A combiation of all needs assessment experts."}
 
+
+prompts_archive_url = "https://huggingface.co/datasets/ryanrwatkins/na_prompts_archive"
+prompts_archive_file_name = "prompts_archive.txt"
+prompts_archive_file = os.path.join("prompts_archive", prompts_archive_file_name)
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+repo = Repository(
+    local_dir="data", clone_from=prompts_archive_url, use_auth_token=HF_TOKEN
+)
+repo.push_to_hub()
+
+
 def get_empty_state():
     return { "messages": []}
 
@@ -79,8 +92,12 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     prompt_template = prompt_templates[prompt_template]
 
     
-    with open('prompt_history.txt', 'w') as f:
-        f.write(str(prompt + "\n"))
+    
+    with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["prompt", "time"])
+            writer.writerow(
+                {"prompt": str(prompt), "time": str(datetime.now())}
+            )
     
 
     system_prompt = []