Update app.py

app.py

@@ -15,7 +15,7 @@ def predict(replicate_api, prompt, lora_id, lora_scale=0.95, aspect_ratio="1:1",
 
     # Validate API key and prompt
     if not replicate_api or not prompt:
-        return "Error: Missing necessary inputs.", -1
+        return "Error: Missing necessary inputs.", -1, None
     
     # Set the seed if randomize_seed is True
     if randomize_seed:
@@ -46,11 +46,11 @@ def predict(replicate_api, prompt, lora_id, lora_scale=0.95, aspect_ratio="1:1",
             "lucataco/flux-dev-lora:a22c463f11808638ad5e2ebd582e07a469031f48dd567366fb4c6fdab91d614d",
             input=input_params
         )
-        return output[0], seed  # Return the generated image and seed
+        return output[0], seed, seed  # Return the generated image and seed
 
     except Exception as e:
         # Catch any exceptions, such as invalid API token or lack of credits
-        return f"Error: {str(e)}", -1
+        return f"Error: {str(e)}", -1, None
 
     finally:
         # Always remove the API key from the environment
@@ -119,6 +119,8 @@ with gr.Blocks(css=css) as demo:
         submit = gr.Button("Generate Image", scale=1)
 
         output = gr.Image(label="Output Image", show_label=True)
+
+        seed_used = gr.Textbox(label="Seed Used", show_copy_button = True)
         
 
         gr.Examples(
@@ -130,7 +132,7 @@ with gr.Blocks(css=css) as demo:
             triggers=[submit.click, prompt.submit],
             fn=predict,
             inputs=[replicate_api, prompt, custom_lora, lora_scale, aspect_ratio, seed, randomize_seed, guidance_scale, num_inference_steps],
-            outputs = [output, seed]
+            outputs = [output, seed, seed_used]
         )
 
 demo.launch()