Update app.py

app.py

@@ -139,8 +139,98 @@ comment_images = [
     "comment_images/55.png"
 ]
 
+example_prompts = [
+    "Painting of a lady",
+    "Cityscape during a thunderstorm",
+    "Inside an abondoned train, window view",
+    "A person reaching for stars",
+    "Abandoned robot at the depth of the sea",
+    "Lonely astronaut in abyss",
+    "Human in a frame"
+]
+
+examples = {
+    "A person reaching for stars":[
+        "examples/A person reaching fo_1.png",
+        "examples/A person reaching fo_2.png",
+        "examples/A person reaching fo_3.png",
+        "examples/A person reaching fo_4.png",
+        "examples/A person reaching fo_5.png",
+        "examples/A person reaching fo_6.png",
+        "examples/A person reaching fo_8.png",
+        "examples/A person reaching fo_9.png",
+        "examples/A person reaching fo_10.png",
+    ],
+    "Abandoned robot at the depth of the sea":[
+        "examples/Abandoned robot at t_1.png",
+        "examples/Abandoned robot at t_2.png",
+        "examples/Abandoned robot at t_3.png",
+        "examples/Abandoned robot at t_4.png",
+        "examples/Abandoned robot at t_5.png",
+        "examples/Abandoned robot at t_6.png",
+        "examples/Abandoned robot at t_8.png",
+        "examples/Abandoned robot at t_9.png",
+        "examples/Abandoned robot at t_10.png",
+    ],
+    "Cityscape during a thunderstorm":[
+        "examples/Cityscape during a t_1.png",
+        "examples/Cityscape during a t_2.png",
+        "examples/Cityscape during a t_3.png",
+        "examples/Cityscape during a t_4.png",
+        "examples/Cityscape during a t_5.png",
+        "examples/Cityscape during a t_6.png",
+        "examples/Cityscape during a t_8.png",
+        "examples/Cityscape during a t_9.png",
+        "examples/Cityscape during a t_10.png",
+    ],
+    "Human in a frame":[
+        "examples/Human in a frame_1.png",
+        "examples/Human in a frame_2.png",
+        "examples/Human in a frame_3.png",
+        "examples/Human in a frame_4.png",
+        "examples/Human in a frame_5.png",
+        "examples/Human in a frame_6.png",
+        "examples/Human in a frame_8.png",
+        "examples/Human in a frame_9.png",
+        "examples/Human in a frame_10.png",
+    ],
+    "Inside an abondoned train, window view":[
+        "examples/Inside an abondoned _1.png",
+        "examples/Inside an abondoned _2.png",
+        "examples/Inside an abondoned _3.png",
+        "examples/Inside an abondoned _4.png",
+        "examples/Inside an abondoned _5.png",
+        "examples/Inside an abondoned _6.png",
+        "examples/Inside an abondoned _8.png",
+        "examples/Inside an abondoned _9.png",
+        "examples/Inside an abondoned _10.png",
+    ],
+    "Lonely astronaut in abyss":[
+        "examples/Lonely astronaut in _1.png",
+        "examples/Lonely astronaut in _2.png",
+        "examples/Lonely astronaut in _3.png",
+        "examples/Lonely astronaut in _4.png",
+        "examples/Lonely astronaut in _5.png",
+        "examples/Lonely astronaut in _6.png",
+        "examples/Lonely astronaut in _8.png",
+        "examples/Lonely astronaut in _9.png",
+        "examples/Lonely astronaut in _10.png",
+    ],
+    "Painting of a lady":[
+        "examples/Painting of a lady_1.png",
+        "examples/Painting of a lady_2.png",
+        "examples/Painting of a lady_3.png",
+        "examples/Painting of a lady_4.png",
+        "examples/Painting of a lady_5.png",
+        "examples/Painting of a lady_6.png",
+        "examples/Painting of a lady_8.png",
+        "examples/Painting of a lady_9.png",
+        "examples/Painting of a lady_10.png",
+    ]
+}
+
 comments = {'test.png': "Not sure about the concept, it's too straightforward. Though the boy looks kinda creepy which makes it exciting. the art style is pretty to look at. I like that the colors are muted, but wish they were a bit darker to make it more eerie and add depth.", 'comment_images/0.png': "Hate this with a passion. The colors are too vibrant and don't match at all. I hate these colors in general. The patterns are too abstract and contemporary. a 5-year-old could draw this. pass.", 'comment_images/1.png': "Woah I love the art style. The texture feels like old paper which is oh so beautiful. There are so many details to focus on. I love the expressive lines and how busy the composition is. Even though orange isn't my favorite, the greenish blue color of the water is so gorgeous.", 'comment_images/2.png': "I don't like how monochromatic and muted this one is. but the paperish texture is nice and the details are so intricate.", 'comment_images/3.png': "Oh super pretty! Looks so smooth and wet. Love the details and loose lines too. Feels mystical and magical and eerie. Also dark purples and blues? deep indigo? My fav ever. I'm here for it.", 'comment_images/4.png': "Love the art style. The uncanny vibe and nightmarish horror is so cool. Like its horror but if you squint you can't tell? Love the strange. wish it had more colors though. not a fan of greyscale.", 'comment_images/5.png': 'omg I hate this haha. what the hell. everything about it disgusts me so boring and childish ew.', 'comment_images/6.png': 'yessss. give it to the texture give it to the brushstrokes give it to the style. perfect. just wish the colors were less beige and more bold. I want an active nightmare. but kisses to the surrealism.'}
-#comments = dict()
+comments = dict()
 
 image_index = 0
 
@@ -371,7 +461,7 @@ def api_fn(api):
         
 
 @spaces.GPU(duration=45)
-def generate(prompt, vp_pos, vp_neg, slider):
+def generate(prompt, vp_pos, vp_neg, slider, example_prompt, gallery):
     if vp_pos == "" and vp_neg == "":
         slider = 0
     
@@ -383,7 +473,30 @@ def generate(prompt, vp_pos, vp_neg, slider):
                 vp_degree_pos=slider,
                 vp_degree_neg=slider
     ).images[0]
-    return image
+
+    global example_prompts, examples
+    if prompt in example_prompts:
+        while example_prompts[0] != prompt:
+            example_prompts.append(example_prompts[0])
+            example_prompts = example_prompts[1:]
+
+        example_prompt = gr.Markdown(f"prompt: {example_prompts[0]}")
+        if len(examples[example_prompts[0]]) == 10:
+            examples[example_prompts[0]] = examples[example_prompts[0]][:-1]
+
+        examples[example_prompts[0]].append(image)
+        gallery = gr.Gallery(
+            value=examples[example_prompts[0]],
+            label="", 
+            show_label=False,
+            columns=[5],
+            rows=[2], 
+            object_fit="contain", 
+            height=500)
+
+
+
+    return image, example_prompt, gallery
 
 def change_vp(extract_vp):
     return
@@ -396,6 +509,42 @@ def upload_file(files):
     next_comment = ""
     return gr.Image(value=comment_images[0], label=f"image {image_index+1}/{len(comment_images)}", show_label=True), gr.Text(label="Comment", show_label=False, lines=2, max_lines=3, placeholder="Enter your comment", value=next_comment, container=False)
 
+def next_prompt():
+    global example_prompts, examples
+    example_prompts.append(example_prompts[0])
+    example_prompts = example_prompts[1:]
+
+    example_prompt = gr.Markdown(f"prompt: {example_prompts[0]}")
+    gallery = gr.Gallery(
+        value=examples[example_prompts[0]],
+        label="", 
+        show_label=False,
+        columns=[5],
+        rows=[2], 
+        object_fit="contain", 
+        height=500)
+
+    return example_prompt, gallery
+
+def previous_prompt():
+    global example_prompts, examples
+    example_prompts = example_prompts[::-1]
+    example_prompts.append(example_prompts[0])
+    example_prompts = example_prompts[1:]
+    example_prompts = example_prompts[::-1]
+
+    example_prompt = gr.Markdown(f"prompt: {example_prompts[0]}")
+    gallery = gr.Gallery(
+        value=examples[example_prompts[0]],
+        label="", 
+        show_label=False,
+        columns=[5],
+        rows=[2], 
+        object_fit="contain", 
+        height=500)
+
+    return example_prompt, gallery
+
 with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
     with gr.Row(elem_id="title-container"):
         gr.Markdown(f"""
@@ -490,12 +639,10 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
     
             with gr.Row():
             
-                prompt = gr.Text(
-                    label="Prompt",
-                    show_label=False,
-                    max_lines=1,
-                    placeholder="Enter your prompt",
-                    container=False,
+                
+
+                prompt = gr.Dropdown(
+                    example_prompts, label="Prompt", info="Enter your prompt", allow_custom_value=True, multiselect=False, show_label=False
                 )
                 
                 run_button = gr.Button("Run", scale=0, interactive=True)
@@ -503,27 +650,22 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
             result = gr.Image(label="Result", show_label=False, interactive=False)
 
     with gr.Row(elem_id="main-container"):
-        with gr.Accordion("images generated from the same prompt but different extracted preferences (prompt: Picture of a lady)", open=False):
-            examples = [
-                "examples/(0).png",
-                "examples/(16).png",
-                "examples/(2).png",
-                "examples/(12).png",
-                "examples/(13).png",
-                "examples/(14).png",
-                "examples/(15).png",
-                "examples/(17).png",
-                "examples/(11).png",
-                "examples/(18).png",
-            ]
+        with gr.Accordion("images generated from the same prompt but different extracted preferences", open=False):
+            
+            
+            example_prompt = gr.Markdown(f"prompt: {example_prompts[0]}")
             gallery = gr.Gallery(
-                value=examples,
+                value=examples[example_prompts[0]],
                 label="", 
                 show_label=False,
                 columns=[5],
                 rows=[2], 
                 object_fit="contain", 
                 height=500)
+
+            with gr.Row():
+                pre_prompt_button = gr.Button("Previous prompt", scale=1, interactive=True)
+                next_prompt_button = gr.Button("Next prompt", scale=1, interactive=True)
             
     
     submit_comment_button.click(
@@ -558,8 +700,8 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
 
     run_button.click(
         fn = generate,
-        inputs = [prompt, positive_extracted_vp, negative_extracted_vp, slider],
-        outputs = [result],
+        inputs = [prompt, positive_extracted_vp, negative_extracted_vp, slider, example_prompt, gallery],
+        outputs = [result, example_prompt, gallery],
     )
 
     positive_extracted_vp.change(
@@ -586,4 +728,16 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
         [comment_image, comment]
     )
 
+    next_prompt_button.click(
+        fn = next_prompt,
+        inputs = [],
+        outputs = [example_prompt, gallery]
+    )
+
+    pre_prompt_button.click(
+        fn = previous_prompt,
+        inputs = [],
+        outputs = [example_prompt, gallery]
+    )
+
 demo.launch(share=True)