update to simplest version

app.py

@@ -21,7 +21,7 @@ from utils import (
 # =========================
 CLIP_MODEL = "ViT-H-14"
 PRETRAINED_CLIP = "laion2b_s32b_b79k"
-default_t2i_model = "black-forest-labs/FLUX.1-schnell" # "black-forest-labs/FLUX.1-dev"
+default_t2i_model = "black-forest-labs/FLUX.1-dev"
 default_llm_model = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
@@ -46,15 +46,15 @@ responses_memory = {METHOD: {}}
 
 example_data = [
     [
-        "A futuristic city skyline at sunset",
+        PROMPTS["Tourist promotion"],
         IMAGES["Tourist promotion"]["ours"]
     ],
     [
-        "A fantasy castle in the clouds",
+        PROMPTS["Fictional character generation"],
         IMAGES["Fictional character generation"]["ours"]
     ],
     [
-        "A robot painting a portrait in a studio",
+        PROMPTS["Interior Design"],
         IMAGES["Interior Design"]["ours"]
     ],
 ]
@@ -251,12 +251,13 @@ css = """
 with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "Arial", "sans-serif"]), css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# 📌 **POET**")
-        instruction = gr.Markdown(" Supporting Prompting Creativity and Personalization with Automated Expansion of Text-to-Image Generation")
+        instruction = gr.Markdown("Supporting Prompting Creativity with Automated Expansion of Text-to-Image Generation")
+        instruction = gr.Markdown("Images generated by POET is more diverse in multiple aspects; e.g., background, race, ...")
 
     with gr.Tab(""):
         with gr.Row(elem_id="compact-row"):
             prompt = gr.Textbox(
-                label="🎨 Revise Prompt",
+                label="🎨 Prompt",
                 max_lines=5,
                 placeholder="Enter your prompt",
                 scale=3,
@@ -269,10 +270,10 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
                 images_method = gr.Gallery(label="Images", columns=[4], rows=[1], height=400, elem_id="gallery", format="png")
 
             with gr.Column(elem_id="col-container3"):
-                like_image = gr.Image(label="Satisfied Image", width=200, height=200, sources='upload', format="png", type="filepath")
-                dislike_image = gr.Image(label="Unsatisfied Image", width=200, height=200, sources='upload', format="png", type="filepath")
+                like_image = gr.Image(label="Satisfied Image", width=200, height=200, sources='upload', format="png", type="filepath", visible=False)
+                dislike_image = gr.Image(label="Unsatisfied Image", width=200, height=200, sources='upload', format="png", type="filepath", visible=False)
 
-        with gr.Column(elem_id="col-container2"):
+        with gr.Column(elem_id="col-container2", visible=False):
             gr.Markdown("### 📝 Evaluation")
             sim_radio = gr.Radio(
                 OPTIONS,
@@ -293,7 +294,7 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
                 elem_classes=["gradio-radio"]
             )
 
-        with gr.Column(elem_id="col-container2"):
+        with gr.Column(elem_id="col-container2", visible=False):
             example = gr.Examples([['']], prompt, label="Revised Prompt History", visible=False)
             history_images = gr.Gallery(label="History Images", columns=[4], rows=[1], elem_id="gallery", format="png")
 
@@ -324,7 +325,7 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
         fn=generate_image,
         inputs=[prompt, like_image, dislike_image],
         outputs=[images_method]
-    ).success(lambda: [gr.update(interactive=False), gr.update(interactive=False)], outputs=[next_btn, prompt])
+    ).success(lambda: [gr.update(interactive=True), gr.update(interactive=True)], outputs=[next_btn, prompt])
 
     redesign_btn.click(
         fn=redesign,