Update app.py

app.py

@@ -227,14 +227,14 @@ def translate_albanian_to_english(text):
 @torch.inference_mode()
 def process(input_fg, prompt_albanian, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source):
     if not input_fg.any():
-        return None, None, "Gabim: Nuk është dhënë asnjë imazh."
+        return None, None
 
     if not prompt_albanian.strip():
         prompt = ""
     else:
         prompt = translate_albanian_to_english(prompt_albanian)
         if prompt.startswith("Përkthimi dështoi"):
-            return None, None, prompt
+            return None, None
 
     bg_source = BGSource(bg_source)
     input_bg = None
@@ -336,17 +336,26 @@ def process(input_fg, prompt_albanian, image_width, image_height, num_samples, s
 
     pixels = vae.decode(latents).sample
 
-    return pytorch2numpy(pixels), "Imazhi u gjenerua me sukses."
+    return pytorch2numpy(pixels)
 
 @spaces.GPU
 @torch.inference_mode()
 def process_relight(input_fg, prompt_albanian, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source):
     if not input_fg.any():
-        return None, None, "Gabim: Nuk është dhënë asnjë imazh."
+        return None, None
 
     input_fg, matting = run_rmbg(input_fg)
-    results, status = process(input_fg, prompt_albanian, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source)
-    return input_fg, results, status
+    results = process(input_fg, prompt_albanian, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source)
+    return input_fg, results
+
+def update_resolution(aspect_ratio):
+    if aspect_ratio == "9:16":
+        return 512, 910
+    elif aspect_ratio == "1:1":
+        return 640, 640
+    elif aspect_ratio == "16:9":
+        return 910, 512
+    return 512, 910  # Default to 9:16
 
 class BGSource(Enum):
     NONE = "Asnjë"
@@ -355,7 +364,31 @@ class BGSource(Enum):
     TOP = "Dritë nga Sipër"
     BOTTOM = "Dritë nga Poshtë"
 
-block = gr.Blocks().queue()
+css = """
+body::before {
+    content: "";
+    display: block;
+    height: 320px;
+    background-color: var(--body-background-fill);
+}
+button[aria-label="Fullscreen"], button[aria-label="Fullscreen"]:hover {
+    display: none !important;
+    visibility: hidden !important;
+    opacity: 0 !important;
+    pointer-events: none !important;
+}
+button[aria-label="Share"], button[aria-label="Share"]:hover {
+    display: none !important;
+}
+button[aria-label="Download"] {
+    transform: scale(3);
+    transform-origin: top right;
+    margin: 0 !important;
+    padding: 6px !important;
+}
+"""
+
+block = gr.Blocks(css=css).queue()
 with block:
     with gr.Row():
         with gr.Column():
@@ -366,16 +399,15 @@ with block:
             bg_source = gr.Radio(choices=[e.value for e in BGSource],
                                  value=BGSource.NONE.value,
                                  label="Preferenca e Ndriçimit (Latenti Fillestar)", type='value')
+            aspect_ratio = gr.Radio(choices=["9:16", "1:1", "16:9"], value="9:16", label="Raporti i Imazhit")
             relight_button = gr.Button(value="Gjenero")
 
             with gr.Group():
                 with gr.Row():
                     num_samples = gr.Slider(label="Numri i Imazheve", minimum=1, maximum=12, value=1, step=1, visible=False)
                     seed = gr.Number(label="Farë", value=-1, precision=0, visible=False)
-
-                with gr.Row():
-                    image_width = gr.Slider(label="Gjerësia e Imazhit", minimum=256, maximum=1024, value=512, step=64)
-                    image_height = gr.Slider(label="Lartësia e Imazhit", minimum=256, maximum=1024, value=640, step=64)
+                    image_width = gr.State(value=512)
+                    image_height = gr.State(value=910)
 
             with gr.Accordion("Opsionet e Avancuara", open=False, visible=False):
                 steps = gr.Slider(label="Hapat", minimum=1, maximum=100, value=50, step=1)
@@ -387,10 +419,15 @@ with block:
                 n_prompt = gr.Textbox(label="Negative Prompt", value='lowres, bad anatomy, bad hands, cropped, worst quality')
         with gr.Column():
             result_gallery = gr.Gallery(height=832, object_fit='contain', label='Rezultatet')
-            status = gr.Textbox(label="Statusi", interactive=False)
     with gr.Row():
         dummy_image_for_outputs = gr.Image(visible=False, label='Rezultati')
     ips = [input_fg, prompt_albanian, 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, status])
+    aspect_ratio.change(
+        fn=update_resolution,
+        inputs=[aspect_ratio],
+        outputs=[image_width, image_height],
+        queue=False
+    )
+    relight_button.click(fn=process_relight, inputs=ips, outputs=[output_bg, result_gallery])
 
 block.launch(server_name='0.0.0.0')