update interface

app.py

@@ -452,13 +452,11 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
                         visible=True,
                 )
                 next_btn1 = gr.Button("Generate", variant="primary", scale=1, interactive=False, visible=False)
-            with gr.Row(elem_id="compact-row"):
-                example1 = gr.Examples([['']], prompt1, label="Revised Prompt History", visible=False)
-        
+            
             with gr.Row(elem_id="compact-row"):
                 with gr.Column(elem_id="col-container"):
-                    images_method1 = gr.Gallery(label="Images", columns=[4], rows=[1], height=200, elem_id="gallery", format="png")
-                    history_images1 = gr.Gallery(label="History Images", columns=[4], rows=[1], elem_id="gallery", format="png")
+                    images_method1 = gr.Gallery(label="Images", columns=[4], rows=[1], height=400, elem_id="gallery", format="png")
+                    
                 with gr.Column(elem_id="col-container3"):
                     like_image1 = gr.Image(label="Satisfied Image", width=200, height=200, sources='upload', format="png", type="filepath")
                     dislike_image1 = gr.Image(label="Unsatisfied Image", width=200, height=200, sources='upload', format="png", type="filepath")
@@ -490,7 +488,11 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
                     container=False,
                     value=VERBAL_MSG
                 )
-                
+            
+            with gr.Column(elem_id="col-container2"):
+                example1 = gr.Examples([['']], prompt1, label="Revised Prompt History", visible=False)
+                history_images1 = gr.Gallery(label="History Images", columns=[4], rows=[1], elem_id="gallery", format="png")
+            
                 with gr.Row(elem_id="button-container"):
                     redesign_btn1 = gr.Button("🎨 Redesign", variant="primary", scale=0)
                     submit_btn1 = gr.Button("✅ Submit", variant="primary", interactive=False, scale=0)
@@ -507,13 +509,11 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
                         visible=True,
                 )
                 next_btn2 = gr.Button("Generate", variant="primary", scale=1, interactive=False, visible=False)
-            with gr.Row(elem_id="compact-row"):
-                example2 = gr.Examples([['']], prompt2, label="Revised Prompt History", visible=False)
-
+            
             with gr.Row(elem_id="compact-row"):
                 with gr.Column(elem_id="col-container"):
                     images_method2 = gr.Gallery(label="Images", columns=[4], rows=[1], height=200, elem_id="gallery", format="png")
-                    history_images2 = gr.Gallery(label="History Images", columns=[4], rows=[1], elem_id="gallery", format="png")
+                    
                 with gr.Column(elem_id="col-container3"):
                     like_image2 = gr.Image(label="Satisfied Image", width=200, height=200, sources='upload', format="png", type="filepath")
                     dislike_image2 = gr.Image(label="Unsatisfied Image", width=200, height=200, sources='upload', format="png", type="filepath")
@@ -546,6 +546,11 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
                     container=False,
                     value=VERBAL_MSG
                 )
+            
+            with gr.Column(elem_id="col-container2"):
+                example2 = gr.Examples([['']], prompt2, label="Revised Prompt History", visible=False)
+                history_images2 = gr.Gallery(label="History Images", columns=[4], rows=[1], elem_id="gallery", format="png")
+            
                 with gr.Row(elem_id="button-container"):
                     redesign_btn2 = gr.Button("🎨 Redesign", variant="primary", scale=0)
                     submit_btn2 = gr.Button("✅ Submit", variant="primary", interactive=False, scale=0)
@@ -563,8 +568,8 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
     
     # prompt1.change(fn=reset_gallery, inputs=[], outputs=[gallery_state1])
     # prompt2.change(fn=reset_gallery, inputs=[], outputs=[gallery_state2])
-    next_btn1.click(fn=generate_image, inputs=[participant, scenario, prompt1, active_tab, like_image1, dislike_image1], outputs=[images_method1])
-    next_btn2.click(fn=generate_image, inputs=[participant, scenario, prompt2, active_tab, like_image2, dislike_image2], outputs=[images_method2])
+    next_btn1.click(fn=generate_image, inputs=[participant, scenario, prompt1, active_tab, like_image1, dislike_image1], outputs=[images_method1]).success(lambda: gr.update(interactive=False), outputs=[next_btn1])
+    next_btn2.click(fn=generate_image, inputs=[participant, scenario, prompt2, active_tab, like_image2, dislike_image2], outputs=[images_method2]).success(lambda: gr.update(interactive=False), outputs=[next_btn2])
     sim_radio1.change(fn=check_satisfaction, inputs=[sim_radio1, active_tab], outputs=[submit_btn1])
     sim_radio2.change(fn=check_satisfaction, inputs=[sim_radio2, active_tab], outputs=[submit_btn2])
     dislike_radio1.select(fn=select_image, inputs=[dislike_radio1, images_method1], outputs=[dislike_image1])

utils.py

@@ -112,10 +112,9 @@ Can you give me {num_prompts} modified prompts for the prompt '{prompt}' please.
     messages.append({"role": "user", "content": f"{message}"})
     return messages
 
-def encode_image(image):
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")  
-    return base64.b64encode(buffered.getvalue()).decode("utf-8")
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
 
 def get_personalize_message(prompt, history_prompts, history_feedback, like_image, dislike_image):
     messages = [