Update app.py

app.py

@@ -5,6 +5,40 @@ import gradio as gr
 model_name = "deepset/electra-base-squad2"
 nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
 
+# dataset save ------------------------------------
+import huggingface_hub
+from huggingface_hub import Repository, hf_hub_download, upload_file
+from datetime import datetime
+# created new dataset as awacke1/MindfulStory.csv
+DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/MindfulStory.csv"
+DATASET_REPO_ID = "awacke1/MindfulStory.csv"
+DATA_FILENAME = "MindfulStory.csv"
+DATA_FILE = os.path.join("data", DATA_FILENAME)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+# Download dataset repo using hub download
+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")
+def AIMemory(title: str, story: str):
+    if title and story:
+        with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["title", "story", "time"])
+            writer.writerow({"title": title, "story": story, "time": str(datetime.now())})
+        commit_url = repo.push_to_hub()
+    return ""
+# Set up cloned dataset from repo for operations
+repo = Repository(
+    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
+# dataset save ------------------------------------
+
+
 def get_wiki_article(topic):
     topic=topic
     try:
@@ -18,6 +52,7 @@ def get_wiki_article(topic):
         choices = [x for x in e.options if ('disambiguation' not in x) and ('All pages' not in x) and (x!=topic)]
         s = random.choice(choices)
         p = wikipedia.page(s)
+        saved = AIMemory(p, s)
     return p.content, p.url
 
 def get_answer(topic, question):