Update app_image_edit.py

app_image_edit.py

@@ -22,7 +22,7 @@ from diffusers import QwenImageEditPlusPipeline, FlowMatchEulerDiscreteScheduler
 from huggingface_hub import InferenceClient
 
 
-from config_image_edit import TITLE, LOGO_HTML, CUSTOM_HEAD, CUSTOM_CSS, HEADLINE_MD, SUBHEADLINE_MD, THEME, DEFAULT_PROMPT, PROMPT_PLACEHOLDER, DEFAULT_PROMPT_NEGATIVE, DATA_IMAGE_WINK, DATA_IMAGE_HEART
+from config_image_edit import TITLE, LOGO_HTML, CUSTOM_HEAD, HEADLINE_MD, SUBHEADLINE_MD, THEME, DEFAULT_PROMPT, PROMPT_PLACEHOLDER, DEFAULT_PROMPT_NEGATIVE, DATA_IMAGE_WINK, DATA_IMAGE_HEART
 
 log = logging.getLogger(__name__)
 
@@ -243,13 +243,22 @@ def polish_prompt_hf(original_prompt, img_list):
         return original_prompt 
 
 
-
 def encode_image(pil_image):
     import io
     buffered = io.BytesIO()
     pil_image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
+
+def use_history_as_input(evt: gr.SelectData, history):
+    if history and evt.index < len(history):
+        im = history[evt.index][0]
+        image_path = f"image_cache/{history[evt.index][1]}"
+        im.save(image_path, format="PNG")
+        return image_path  # gr.update(value=history[evt.index][0])
+    return  # gr.update()
+
+
 @spaces.GPU(duration=300)
 def infer(
     images,
@@ -323,10 +332,11 @@ def infer(
     except Exception as exc:
         raise gr.Error(f"Fehler beim Speichern des Bildes: {exc}")
     
-    gallery_items = [item for item in gallery] if gallery else []
-    gallery_items.append((str(image_path), str(image_path.name)))
+    if gallery is None:
+        gallery = []
+    gallery.insert(0, (image, str(f"{timestamp}.png")))
 
-    return str(image_path), seed, gr.update(value=gallery_items, interactive=True)
+    return str(image_path), seed, gallery
 
 
 # ------------------
@@ -402,18 +412,19 @@ with gr.Blocks(title=TITLE) as demo:
 
                                 rewrite_prompt = gr.Checkbox(label="✨ Prompt verbessern", value=True, elem_classes="einstellungen-rewrite-prompt", interactive=True)
 
-                generate_button = gr.Button("Bearbeiten ✨", variant="primary", elem_classes="gradio-button run-button", size="lg")
+                game_button = gr.Button("Bearbeiten ✨", variant="primary", elem_classes="gradio-button run-button", elem_id="gameBtn", size="lg")
+                generate_button = gr.Button("Bearbeiten ✨", variant="primary", elem_classes="gradio-button run-button", elem_id="runBtn", size="lg")
 
                 with gr.Column(elem_classes="output-column", visible=False) as output_column_1:
                     # output_video = gr.Video(value=None, label="Ergebnis", show_label=False, elem_classes="result-video", visible=True, interactive=False)
                     with gr.Row(elem_classes="gradio-row", equal_height=True):
                         with gr.Column():
                             gr.Markdown("# **** Ergebnis", elem_classes="markdown-label")
-                            result_image = gr.Image(label="Ergebnis", type="filepath", show_label=False, buttons=["download", "fullscreen"], elem_classes="result-image", visible=True, interactive=False)
+                            result_image = gr.Image(label="Ergebnis", type="pil", show_label=False, buttons=["download", "fullscreen"], elem_classes="result-image", visible=True, interactive=False)
                         
                         with gr.Column():
                             gr.Markdown("# **** Verlauf", elem_classes="markdown-label")
-                            result_history = gr.Gallery([], label="Ergebnis", show_label=False, buttons=["download", "fullscreen"], object_fit="cover", elem_classes="result-gallery", interactive=False, allow_preview=True)
+                            result_history = gr.Gallery([], label="Ergebnis", show_label=False, buttons=["download", "fullscreen"], object_fit="cover", elem_classes="result-gallery", interactive=False, allow_preview=False, type="pil")
 
                 # greet_btn.click(fn=greet, inputs=[], outputs=output, api_name="greet")
     with gr.Tab(" Info", elem_classes="info-tab"):
@@ -438,11 +449,14 @@ with gr.Blocks(title=TITLE) as demo:
 
     randomize_seed_checkbox.change(fn=lambda x: gr.update(interactive=(not x)), inputs=[randomize_seed_checkbox], outputs=[seed_input], show_progress=False)
     guidance_scale_input.change(fn=lambda x: gr.update(interactive=(x > 2)), inputs=[guidance_scale_input], outputs=[negative_prompt_input], show_progress=False)
+
+    result_history.select(fn=use_history_as_input, inputs=[result_history], outputs=[result_image], show_progress=False)
     
     ui_inputs = [input_image_component, prompt, negative_prompt_input, steps_slider, guidance_scale_input, seed_input, randomize_seed_checkbox, rewrite_prompt, result_history]
-    generate_button.click(fn=lambda: (gr.Column(visible=True), gr.Button("Arbeitet ...", variant="secondary", elem_classes="gradio-button run-button", size="lg", interactive=False)), outputs=[output_column_1, generate_button], show_progress=False, scroll_to_output=True).then(fn=infer, inputs=ui_inputs, outputs=[result_image, seed_input, result_history], show_progress_on=[result_image, result_history], scroll_to_output=True).then(
-        fn=lambda: gr.Button("Bearbeiten ✨", variant="primary", elem_classes="gradio-button run-button", size="lg", interactive=True), outputs=[generate_button], show_progress=False
+    generate_button.click(fn=lambda: gr.Column(visible=True), outputs=[output_column_1], show_progress=False, scroll_to_output=True).then(fn=infer, inputs=ui_inputs, outputs=[result_image, seed_input, result_history], show_progress_on=[result_image, result_history], scroll_to_output=True).then(
+        fn=None, outputs=generate_button, js="() => { document.getElementById('gameBtn').classList.remove('process-running'); }"
     )
 
 if __name__ == "__main__":
-    demo.launch(theme=THEME, head=CUSTOM_HEAD, css=CUSTOM_CSS, footer_links=["gradio", "settings"])
+    # demo.launch(theme=THEME, head=CUSTOM_HEAD, css=CUSTOM_CSS, footer_links=["gradio", "settings"])
+    demo.launch(theme=THEME, head=CUSTOM_HEAD, css_paths=["_res/_custom_image_edit.css", "_res/miniGameButton.css"], footer_links=["gradio", "settings"])