Update app.py

app.py

@@ -80,7 +80,7 @@ def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width
 # Gradio interface
 def create_demo():
     with gr.Blocks() as demo:
-        # CSS for compact layout, 320px gap, and download button scaling
+        # CSS for layout, 320px gap, and download button scaling
         gr.HTML("""
         <style>
         body::before {
@@ -104,76 +104,70 @@ def create_demo():
             margin: 0 !important;
             padding: 6px !important;
         }
-        .constrained-container {
-            max-width: 600px;
-            margin: 0 auto;
-        }
         </style>
         """)
 
-        gr.Markdown("# Gjenero Imazhe")
-        gr.Markdown("Krijo një imazh të ri teper real bazuar në përshkrimin tënd")
+        gr.Markdown("# Krijo Imazhe")
+        gr.Markdown("Gjenero imazhe të reja nga përshkrimin yt me fuqinë e inteligjencës artificiale.")
+
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="Përshkrimi",
+                placeholder="Shkruani përshkrimin këtu",
+                lines=3
+            )
+            aspect_ratio = gr.Radio(
+                label="Raporti i fotos",
+                choices=["9:16", "1:1", "16:9"],
+                value="1:1"
+            )
+            generate_button = gr.Button(value="Gjenero")
+            # Hidden components for processing
+            negative_prompt = gr.Textbox(
+                value="(low quality, worst quality:1.2), very displeasing, 3d, watermark, signature, ugly, poorly drawn, (deformed | distorted | disfigured:1.3), bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers:1.4, disconnected limbs, blurry, amputation.",
+                visible=False
+            )
+            num_inference_steps = gr.Slider(
+                value=60,
+                minimum=1,
+                maximum=100,
+                step=1,
+                visible=False
+            )
+            guidance_scale = gr.Slider(
+                value=7,
+                minimum=1,
+                maximum=20,
+                step=0.1,
+                visible=False
+            )
+            width_slider = gr.Slider(
+                value=1024,
+                minimum=256,
+                maximum=1536,
+                step=8,
+                visible=False
+            )
+            height_slider = gr.Slider(
+                value=1024,
+                minimum=256,
+                maximum=1536,
+                step=8,
+                visible=False
+            )
+            num_samples = gr.Slider(
+                value=1,
+                minimum=1,
+                maximum=1,
+                step=1,
+                visible=False
+            )
 
         with gr.Row():
-            with gr.Column(elem_classes="constrained-container"):
-                prompt = gr.Textbox(
-                    label="Përshkrimi",
-                    placeholder="Shkruani përshkrimin këtu"
-                )
-                aspect_ratio = gr.Radio(
-                    choices=["9:16", "1:1", "16:9"],
-                    value="1:1",
-                    label="Raporti i Imazhit"
-                )
-                generate_button = gr.Button(value="Gjenero")
-                result_image = gr.Image(
-                    label="Rezultati",
-                    type="pil",
-                    height=480,
-                    width=480,
-                    interactive=False,
-                    elem_classes="constrained-container"
-                )
-                # Hidden components for processing
-                negative_prompt = gr.Textbox(
-                    value="(low quality, worst quality:1.2), very displeasing, 3d, watermark, signature, ugly, poorly drawn, (deformed | distorted | disfigured:1.3), bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers:1.4, disconnected limbs, blurry, amputation.",
-                    visible=False
-                )
-                num_inference_steps = gr.Slider(
-                    value=60,
-                    minimum=1,
-                    maximum=100,
-                    step=1,
-                    visible=False
-                )
-                guidance_scale = gr.Slider(
-                    value=7,
-                    minimum=1,
-                    maximum=20,
-                    step=0.1,
-                    visible=False
-                )
-                width_slider = gr.Slider(
-                    value=1024,
-                    minimum=256,
-                    maximum=1536,
-                    step=8,
-                    visible=False
-                )
-                height_slider = gr.Slider(
-                    value=1024,
-                    minimum=256,
-                    maximum=1536,
-                    step=8,
-                    visible=False
-                )
-                num_samples = gr.Slider(
-                    value=1,
-                    minimum=1,
-                    maximum=1,
-                    step=1,
-                    visible=False
-                )
+            result_image = gr.Image(
+                label="Imazhi i Gjeneruar",
+                interactive=False
+            )
 
         # Update hidden sliders based on aspect ratio
         aspect_ratio.change(
@@ -191,7 +185,8 @@ def create_demo():
         generate_button.click(
             fn=generate,
             inputs=inputs,
-            outputs=[result_image]
+            outputs=[result_image],
+            show_progress="full"
         )
 
     return demo