update

app.py

@@ -10,11 +10,9 @@ EXAMPLES_DIR = os.path.abspath("./examples")
 def generate_images(prompt, init_pop, total_itrs):
     init_pop = int(init_pop)
     total_itrs = int(total_itrs)
-
     # Use placeholder if prompt is empty
     if not prompt.strip():
         prompt = "a duck crossing the street"
-
     archive_plots = []
     for archive, plt_fig in run_qdhf(prompt, init_pop, total_itrs):
         buf = io.BytesIO()
@@ -36,14 +34,15 @@ def generate_images(prompt, init_pop, total_itrs):
     
     return temp_archive_file, temp_images_file
 
-def show_example(example):
-    index = EXAMPLE_PROMPTS.index(example)
+def show_example(prompt):
+    index = EXAMPLE_PROMPTS.index(prompt)
     archive_plot_path = os.path.join(EXAMPLES_DIR, f"archive_{index}.mp4")
     images_path = os.path.join(EXAMPLES_DIR, f"archive_pics_{index}.png")
-    return archive_plot_path, images_path
+    return prompt, archive_plot_path, images_path
 
 with gr.Blocks() as demo:
     gr.Markdown("# Quality Diversity through Human Feedback")
+    gr.Markdown("[Paper](https://arxiv.org/abs/2310.12103) | [Project Website](https://liding.info/qdhf/)")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -60,10 +59,14 @@ with gr.Blocks() as demo:
                           inputs=[prompt_input, init_pop, total_itrs],
                           outputs=[archive_output, images_output])
     
-    gr.Markdown("## Examples:")
-    for example in EXAMPLE_PROMPTS:
-        example_button = gr.Button(example)
-        example_button.click(show_example, inputs=example_button, outputs=[archive_output, images_output])
+    gr.Examples(
+        examples=EXAMPLE_PROMPTS,
+        inputs=prompt_input,
+        outputs=[prompt_input, archive_output, images_output],
+        fn=show_example,
+        cache_examples=True,
+        label="Example Prompts"
+    )
 
 if __name__ == "__main__":
     demo.launch()