Update app.py

app.py

@@ -21,47 +21,14 @@ import huggingface_hub
 from huggingface_hub import Repository
 from datetime import datetime
 
-DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/BookComposer"
-DATA_FILENAME = "BookComposer.csv"
-DATA_FILE = os.path.join("data", DATA_FILENAME)
-
 HF_TOKEN = os.environ.get("HF_TOKEN")
 print("is none?", HF_TOKEN is None)
-
-print("hfh", huggingface_hub.__version__)
-
-# overriding/appending to the gradio template
-SCRIPT = """
-<script>
-if (!window.hasBeenRun) {
-    window.hasBeenRun = true;
-    console.log("should only happen once");
-    document.querySelector("button.submit").click();
-}
-</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=True
-
-
 DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/BookComposer"
 DATA_FILENAME = "BookComposer.csv"
 DATA_FILE=os.path.join("data", DATA_FILENAME)
 HF_TOKEN=os.environ.get("HF_TOKEN")
-repo = Repository(
-    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-)
-
-
-
-
 
 def gan():
-    
     images = [
         (random.choice(
             [
@@ -106,13 +73,55 @@ def gan():
     ]
     return images
 
+def gan2():
+    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
+
+
 
 with gr.Blocks() as demo:
     with gr.Column(variant="panel"):
-        [
-        inputs.Textbox(placeholder="Your name"),
-        inputs.Textbox(placeholder="Your message", lines=2),
-        ]
         with gr.Row(variant="compact"):
             text = gr.Textbox(
                 label="Health and Medical Icon Sets",
@@ -130,6 +139,7 @@ with gr.Blocks() as demo:
         ).style(grid=[2], height="auto")
 
     btn.click(gan, None, gallery)
+    btn2.click(gan2, None, gallery)
 
 if __name__ == "__main__":
     demo.launch()