Update app.py

app.py

@@ -114,7 +114,7 @@ i2i_pipe = StableDiffusionImg2ImgPipeline(
     image_encoder=None
 )
 
-# Translation function (adapted from example)
+# Translation function (unchanged)
 @spaces.GPU
 def translate_albanian_to_english(text):
     if not text.strip():
@@ -327,7 +327,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
         output_type='latent',
         guidance_scale=cfg,
         cross_attention_kwargs={'concat_conds': concat_conds},
-    ).images.to(vae.dtype) / vae.config.scaling_factor
+        ).images.to(vae.dtype) / vae.config.scaling_factor
 
     pixels = vae.decode(latents).sample
 
@@ -388,11 +388,26 @@ def create_demo():
             prompt = gr.Textbox(label="Përshkrimi", placeholder="Shkruani përshkrimin këtu")
             bg_source = gr.Radio(choices=[e.value for e in BGSource], value=BGSource.NONE.value, label="Preferenca e Ndriçimit", type='value')
             relight_button = gr.Button(value="Rindriço")
-            result_gallery = gr.Gallery(label='Rezultatet', visible=False)  # Hidden output
-            output_bg = gr.Image(type="numpy", label="Parapërpunimi i Planit të Parë", visible=False)  # Hidden output
+            # Hidden components for other parameters
+            image_width = gr.Slider(label="Gjerësia e Imazhit", minimum=256, maximum=1024, value=640, step=64, visible=False)
+            image_height = gr.Slider(label="Lartësia e Imazhit", minimum=256, maximum=1024, value=640, step=64, visible=False)
+            num_samples = gr.Slider(label="Numri i Imazheve", minimum=1, maximum=12, value=1, step=1, visible=False)
+            seed = gr.Number(label="Fara", value=12345, precision=0, visible=False)
+            steps = gr.Slider(label="Hapat", minimum=1, maximum=100, value=50, step=1, visible=False)
+            a_prompt = gr.Textbox(label="Përshkrim i Shtuar", value='best quality', visible=False)
+            n_prompt = gr.Textbox(label="Përshkrim Negativ", value='lowres, bad anatomy, bad hands, cropped, worst quality', visible=False)
+            cfg = gr.Slider(label="Shkalla CFG", minimum=1.0, maximum=32.0, value=2, step=0.01, visible=False)
+            highres_scale = gr.Slider(label="Shkalla e Rezolutës së Lartë", minimum=1.0, maximum=3.0, value=2, step=0.01, visible=False)
+            highres_denoise = gr.Slider(label="Denoise i Rezolutës së Lartë", minimum=0.1, maximum=1.0, value=0.5, step=0.01, visible=False)
+            lowres_denoise = gr.Slider(label="Denoise i Rezolutës së Ulët", minimum=0.1, maximum=1.0, value=0.9, step=0.01, visible=False)
+            result_gallery = gr.Gallery(label='Rezultatet', visible=False)
+            output_bg = gr.Image(type="numpy", label="Parapërpunimi i Planit të Parë", visible=False)
 
         # Bind the relight button
-        ips = [input_fg, prompt, 640, 640, 1, 12345, 50, 'best quality', 'lowres, bad anatomy, bad hands, cropped, worst quality', 2, 2, 0.5, 0.9, bg_source]
+        ips = [
+            input_fg, prompt, image_width, image_height, num_samples, seed, steps,
+            a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source
+        ]
         relight_button.click(fn=process_relight, inputs=ips, outputs=[output_bg, result_gallery])
 
     return block