Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import numpy as np
 from prodiapy import Prodia
 import gradio as gr
 from style_template import styles
@@ -7,9 +8,10 @@ import base64
 prodia = Prodia()
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "Photographic (Default)"
+MAX_SEED = np.iinfo(np.int32).max
 
 
-def generate_image(upload_images, prompt, negative_prompt, style_name, steps, progress=gr.Progress(track_tqdm=True)):
+def generate_image(upload_images, prompt, negative_prompt, style_name, steps, strength, seed, progress=gr.Progress(track_tqdm=True)):
     error_if_no_img(prompt)
     p, n = apply_style(style_name, prompt, negative_prompt)
 
@@ -17,7 +19,9 @@ def generate_image(upload_images, prompt, negative_prompt, style_name, steps, pr
                         imageData=[file_to_base64(img) for img in upload_images],
                         prompt=p,
                         negative_prompt=n,
-                        steps=steps
+                        steps=steps,
+                        strength=strength,
+                        seed=seed
                         )
     result = prodia.wait(job)
 
@@ -117,6 +121,20 @@ with gr.Blocks(css=css) as demo:
                     step=1,
                     value=40,
                 )
+                strength_ratio = gr.Slider(
+                    label="Strength (%)",
+                    minimum=15,
+                    maximum=50,
+                    step=1,
+                    value=20,
+                )
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0,
+                )
         with gr.Column():
             result_image = gr.Image(label="Generated Image")
 
@@ -125,7 +143,7 @@ with gr.Blocks(css=css) as demo:
 
         submit.click(
             fn=generate_image,
-            inputs=[files, prompt, negative_prompt, style, steps],
+            inputs=[files, prompt, negative_prompt, style, steps, strength_ratio, seed],
             outputs=[result_image]
         )