v3p3

app.py

@@ -183,12 +183,7 @@ def generate(
                 filepath = utils.save_image(image, metadata, OUTPUT_DIR)
                 logger.info(f"Image saved as {filepath} with metadata")
 
-        # Update history after generation
-        history = gr.get_state("history") or []
-        history.insert(0, {"prompt": prompt, "image": images[0], "metadata": metadata})
-        gr.set_state("history", history[:10])  # Keep only the last 10 entries
-
-        return images, metadata, gr.update(choices=[h["prompt"] for h in history])
+        return images, json.dumps(metadata)  # Return metadata as a JSON string
     except Exception as e:
         logger.exception(f"An error occurred: {e}")
         raise
@@ -214,7 +209,6 @@ def get_random_prompt():
     
     return f"{score}, {character}, {style}, show accurate"
 
-
 if torch.cuda.is_available():
     pipe = load_pipeline(MODEL)
     logger.info("Loaded on Device!")
@@ -390,11 +384,18 @@ with gr.Blocks(css="style.css") as demo:
         clear_button = gr.Button("Clear All")
         random_prompt_button = gr.Button("Random Prompt")
 
+    history = gr.State([])  # Add a state component to store history
     history_dropdown = gr.Dropdown(label="Generation History", choices=[], interactive=True, elem_id="history-dropdown")
     
     with gr.Accordion(label="Generation Parameters", open=False):
         gr_metadata = gr.JSON(label="Metadata", show_label=False)
     
+    def update_history(images, metadata, history):
+        if images:
+            new_entry = {"prompt": json.loads(metadata)["prompt"], "image": images[0]}
+            history = [new_entry] + history[:9]  # Keep only the last 10 entries
+        return gr.update(choices=[h["prompt"] for h in history]), history
+
     gr.Examples(
         examples=config.examples,
         inputs=prompt,
@@ -442,9 +443,14 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=[result, gr_metadata, history_dropdown],
+        outputs=[result, gr_metadata],
         api_name="run",
+    ).then(
+        fn=update_history,
+        inputs=[result, gr_metadata, history],
+        outputs=[history_dropdown, history],
     )
+
     negative_prompt.submit(
         fn=utils.randomize_seed_fn,
         inputs=[seed, randomize_seed],
@@ -454,9 +460,14 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=[result, gr_metadata, history_dropdown],
+        outputs=[result, gr_metadata],
         api_name=False,
+    ).then(
+        fn=update_history,
+        inputs=[result, gr_metadata, history],
+        outputs=[history_dropdown, history],
     )
+
     run_button.click(
         fn=utils.randomize_seed_fn,
         inputs=[seed, randomize_seed],
@@ -466,8 +477,12 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=[result, gr_metadata, history_dropdown],
+        outputs=[result, gr_metadata],
         api_name=False,
+    ).then(
+        fn=update_history,
+        inputs=[result, gr_metadata, history],
+        outputs=[history_dropdown, history],
     )
 
     apply_json_button.click(
@@ -498,8 +513,8 @@ with gr.Blocks(css="style.css") as demo:
     )
 
     history_dropdown.change(
-        fn=lambda x: gr.update(value=x),
-        inputs=history_dropdown,
+        fn=lambda x, history: next((h["prompt"] for h in history if h["prompt"] == x), ""),
+        inputs=[history_dropdown, history],
         outputs=prompt
     )