Update app.py

app.py

@@ -40,10 +40,58 @@ if (!window.hasBeenRun) {
 }
 </script>
 """
+
 with open(os.path.join(gr.networking.STATIC_TEMPLATE_LIB, "frontend", "index.html"), "a") as f:
     f.write(SCRIPT)
 
+# -------------------------------------------- For Memory - you will need to set up a dataset and HF_TOKEN ---------
+UseMemory=False
+
+
+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")
 
+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 get_df(name: str):
+    dataset = load_dataset(str, split="train")
+    return dataset
+    
+#def store_message(name: str, message: str) -> str:
+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", ])
+            writer.writerow(
+                {"time": str(datetime.now()), "message": message.strip(), "name": name.strip()  }
+            )
+        commit_url = repo.push_to_hub()
+
+        # test api retrieval of any dataset that is saved, then return it...
+        # app = FastAPI()
+        # see: https://gradio.app/sharing_your_app/#api-page
+
+        # f=get_df(DATASET_REPO_ID)
+        # print(f)
+    #return commit_url
+    return ""
+# ----------------------------------------------- For Memory
+    
 
 repo = Repository(
     local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
@@ -69,21 +117,61 @@ def generate_html() -> str:
             html += "</div>"
             return html
 
+    return generate_html()
 
-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, "message": message, "time": str(datetime.now())}
-            )
-        commit_url = repo.push_to_hub()
-        print(commit_url)
 
-    return generate_html()
 
 
-iface = gr.Interface(
+def gan():
+    
+    images = [
+        (random.choice(
+            [
+                "Comectress.png",
+                "Comic Plate 1 - Amnesia.png",
+                "Comic Plate 1 - Depths.png",
+                "Comic Plate 1 - Helion.png",
+                "Comic Plate 1 - Kitsune.png",
+                "Comic Plate 1 - Sinbad.png",
+                "Comic Plate 1 - Vampiress.png",
+                "Comic Plate 2 - Amnesia.png",
+                "Comic Plate 2 - Depths.png",
+                "Comic Plate 2 - Helion.png",
+                "Comic Plate 2 - Kitsune.png",
+                "Comic Plate 2 - Sinbad.png",
+                "Comic Plate 2 - Vampiress.png",
+                "Comic Plate 3 - Amnesia.png",
+                "Comic Plate 3 - Depths.png",
+                "Comic Plate 3 - Helion.png",
+                "Comic Plate 3 - Kitsune.png",
+                "Comic Plate 3 - Vampiress.png",
+                "Comic Plate 4 - Amnesia.png",
+                "Comic Plate 4 - Depths.png",
+                "Comic Plate 4 - Helion.png",
+                "Comic Plate 4 - Kitsune.png",
+                "Comic Plate 4 - Sinbad.png",
+                "Comic Plate 4 - Vampiress.png",
+                "Comic Plate 5 - Vampiress.png",
+                "Gold Suit.png",
+                "Heavens.png",
+                "Red Dot Sight.png",
+                "Starfire.png",
+                "Vamp 1.png",
+                "Vamp 2.png",
+                "Vamp 3.png",
+                "Vamp 4.png",
+                "Whirlwind.png",
+                "Zyphoria.png",
+            ]
+        ), f"label {i}" if i != 0 else "label" * 50)
+        for i in range(10)
+    ]
+    return images
+
+
+def gan2():
+    r=store_message()
+    iface = gr.Interface(
     store_message,
     [
         inputs.Textbox(placeholder="Your name"),
@@ -95,15 +183,8 @@ iface = gr.Interface(
     """,
     title="Reading/writing to a HuggingFace dataset repo from Spaces",
     description=f"This is a demo of how to do simple *shared data persistence* in a Gradio Space, backed by a dataset repo.",
-    article=f"The dataset repo is [{DATASET_REPO_URL}]({DATASET_REPO_URL}) (open in new tab)",
-)
-
-iface.launch()
-
-
-
-def gan():
-    r=store_message()
+    article=f"The dataset repo is [{DATASET_REPO_URL}]({DATASET_REPO_URL}) (open in new tab)",)
+    
     
     images = [
         (random.choice(
@@ -147,9 +228,10 @@ def gan():
         ), f"label {i}" if i != 0 else "label" * 50)
         for i in range(10)
     ]
+    iface.launch()
+    
     return images
 
-
 with gr.Blocks() as demo:
     with gr.Column(variant="panel"):
         [
@@ -166,12 +248,17 @@ with gr.Blocks() as demo:
                 container=False,
             )
             btn = gr.Button("Generate image").style(full_width=False)
+            btn2 = gr.Button("Generate story").style(full_width=False)
 
         gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
         ).style(grid=[2], height="auto")
+        gallery2 = gr.Gallery(
+            label="Generated images 2", show_label=False, elem_id="gallery"
+        ).style(grid=[2], height="auto")
 
     btn.click(gan, None, gallery)
+    btn2.click(gan2, None, gallery2)
 
 if __name__ == "__main__":
     demo.launch()