change layout to sidebar mode

README.md

@@ -1,8 +1,8 @@
 ---
 title: ModelCofferGallery
-emoji: 🦀
-colorFrom: green
-colorTo: pink
+emoji: 🖼️
+colorFrom: blue
+colorTo: purple
 sdk: streamlit
 sdk_version: 1.21.0
 app_file: app.py

app.py

@@ -15,7 +15,6 @@ class GalleryApp:
         self.promptBook = promptBook
         st.set_page_config(layout="wide")
 
-
     def gallery(self, items, col_num, info):
         cols = st.columns(col_num)
         # # sort items by brisque score
@@ -33,7 +32,7 @@ class GalleryApp:
         st.title('Model Coffer Gallery')
         st.write('This is a gallery of images generated by the models in the Model Coffer')
 
-        metadata, images = st.columns([1, 3])
+        # metadata, images = st.columns([1, 3])
         # with images:
             # prompt_tags = self.promptBook['tag'].unique()
             # # sort tags by alphabetical order
@@ -43,20 +42,25 @@ class GalleryApp:
             # with selecters[0]:
             #     tag = st.selectbox('Select a tag', prompt_tags)
 
-        with metadata:
+        images = st.container()
+
+        with st.sidebar:
             prompt_tags = self.promptBook['tag'].unique()
             # sort tags by alphabetical order
-            prompt_tags = np.sort(prompt_tags)
+            prompt_tags = np.sort(prompt_tags)[::-1]
 
             tag = st.selectbox('Select a tag', prompt_tags)
 
             items = self.promptBook[self.promptBook['tag'] == tag].reset_index(drop=True)
 
-            prompts = [', '.join(x.split(', ')[4:]) for x in items['prompt'].unique()]
+            original_prompts = np.sort(items['prompt'].unique())[::-1]
+
+            # remove the first four items in the prompt, which are mostly the same
+            prompts = [', '.join(x.split(', ')[4:]) for x in original_prompts]
             prompt = st.selectbox('Select prompt', prompts)
 
             idx = prompts.index(prompt)
-            prompt_full = ', '.join(items['prompt'].unique()[idx].split(', ')[:4]) + ', ' + prompt
+            prompt_full = ', '.join(original_prompts[idx].split(', ')[:4]) + ', ' + prompt
 
             prompt_id = items[items['prompt'] == prompt_full]['prompt_id'].unique()[0]
             items = items[items['prompt_id'] == prompt_id].reset_index(drop=True)

data/download_script.py

@@ -21,5 +21,3 @@ def test():
 
 if __name__ == '__main__':
     main()
-    # load()
-    # test()

data/promptbook/data-00009-of-00010.arrow

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e78fd3a4bf99ef29d2800ceff0962795dccb8e6ff3bcbbfa9433008c5519a6f2
-size 477989312

data/promptbook/dataset_info.json

@@ -1,123 +0,0 @@
-{
-  "citation": "",
-  "dataset_size": 4790396199,
-  "description": "",
-  "download_checksums": {
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00000-of-00010-cd1633256b85c720.parquet": {
-      "num_bytes": 483210906,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00001-of-00010-5b5a32ac3990b07d.parquet": {
-      "num_bytes": 476302101,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00002-of-00010-53e82c1a34dc8a24.parquet": {
-      "num_bytes": 477247646,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00003-of-00010-1584b725a142a964.parquet": {
-      "num_bytes": 480767762,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00004-of-00010-71df51519e286b37.parquet": {
-      "num_bytes": 480021191,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00005-of-00010-496f4d5e272abbcb.parquet": {
-      "num_bytes": 479774809,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00006-of-00010-fdeaab3f5af10d27.parquet": {
-      "num_bytes": 476046395,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00007-of-00010-d2388810ca704795.parquet": {
-      "num_bytes": 477758243,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00008-of-00010-2bc31600a6722b7f.parquet": {
-      "num_bytes": 479373326,
-      "checksum": null
-    },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00009-of-00010-e3f7e9b1da9d35f7.parquet": {
-      "num_bytes": 477778395,
-      "checksum": null
-    }
-  },
-  "download_size": 4788280774,
-  "features": {
-    "image": {
-      "_type": "Image"
-    },
-    "image_id": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "tag": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "model_id": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "modelVersion_id": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "prompt_id": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "size": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "seed": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "prompt": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "negativePrompt": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "cfgScale": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "sampler": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "brisque_score": {
-      "dtype": "float64",
-      "_type": "Value"
-    },
-    "clip_score": {
-      "dtype": "float64",
-      "_type": "Value"
-    }
-  },
-  "homepage": "",
-  "license": "",
-  "size_in_bytes": 9578676973,
-  "splits": {
-    "train": {
-      "name": "train",
-      "num_bytes": 4790396199,
-      "num_examples": 7935,
-      "shard_lengths": [
-        1588,
-        1588,
-        1587,
-        1586,
-        1586
-      ],
-      "dataset_name": "parquet"
-    }
-  }
-}

data/promptbook/state.json

@@ -1,40 +0,0 @@
-{
-  "_data_files": [
-    {
-      "filename": "data-00000-of-00010.arrow"
-    },
-    {
-      "filename": "data-00001-of-00010.arrow"
-    },
-    {
-      "filename": "data-00002-of-00010.arrow"
-    },
-    {
-      "filename": "data-00003-of-00010.arrow"
-    },
-    {
-      "filename": "data-00004-of-00010.arrow"
-    },
-    {
-      "filename": "data-00005-of-00010.arrow"
-    },
-    {
-      "filename": "data-00006-of-00010.arrow"
-    },
-    {
-      "filename": "data-00007-of-00010.arrow"
-    },
-    {
-      "filename": "data-00008-of-00010.arrow"
-    },
-    {
-      "filename": "data-00009-of-00010.arrow"
-    }
-  ],
-  "_fingerprint": "1c800ac7347ade50",
-  "_format_columns": null,
-  "_format_kwargs": {},
-  "_format_type": null,
-  "_output_all_columns": false,
-  "_split": "train"
-}

data/roster/data-00000-of-00001.arrow

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:fe153e2d30eb6f28bcbe03bdbf78d99fee777e03917848575f5dc0715f8c3a64
-size 109016

data/roster/dataset_info.json

@@ -1,57 +0,0 @@
-{
-  "citation": "",
-  "dataset_size": 107638,
-  "description": "",
-  "download_checksums": {
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferRoster/resolve/ad2d1c28c610862c42861ca20baa814312d06bea/data/train-00000-of-00001-e90719e7788f6218.parquet": {
-      "num_bytes": 16643,
-      "checksum": null
-    }
-  },
-  "download_size": 16643,
-  "features": {
-    "tag": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "model_name": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "model_id": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "modelVersion_name": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "modelVersion_id": {
-      "dtype": "int64",
-      "_type": "Value"
-    },
-    "modelVersion_url": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "modelVersion_trainedWords": {
-      "dtype": "string",
-      "_type": "Value"
-    },
-    "model_download_count": {
-      "dtype": "int64",
-      "_type": "Value"
-    }
-  },
-  "homepage": "",
-  "license": "",
-  "size_in_bytes": 124281,
-  "splits": {
-    "train": {
-      "name": "train",
-      "num_bytes": 107638,
-      "num_examples": 773,
-      "dataset_name": "parquet"
-    }
-  }
-}

data/roster/state.json

@@ -1,13 +0,0 @@
-{
-  "_data_files": [
-    {
-      "filename": "data-00000-of-00001.arrow"
-    }
-  ],
-  "_fingerprint": "e1e2eff17d4ef545",
-  "_format_columns": null,
-  "_format_kwargs": {},
-  "_format_type": null,
-  "_output_all_columns": false,
-  "_split": "train"
-}