Update app.py

app.py

@@ -25,7 +25,6 @@ logger = logging.getLogger(__name__)
 
 @spaces.GPU
 def translate_albanian_to_english(text: str, language: str = "en"):
-    """Translate Albanian text to English using an external HF Space."""
     if not text.strip():
         raise gr.Error("Please enter a description.")
     for attempt in range(2):
@@ -93,7 +92,6 @@ QUALITY_PROMPT = ", high quality, detailed, vibrant, professional lighting"
 
 @spaces.GPU(duration=40)
 def infer(image, prompt):
-    """Generate an edited image from the input image and prompt."""
     negative_prompt = ""
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
@@ -164,7 +162,14 @@ def create_demo():
                 background:#000000 !important;
                 color:#FFFFFF !important;
             }
-            .gr-row{
+            .gr-row,.gr-column{
+                width:100% !important;
+                max-width:100vw !important;
+                margin:0 !important;
+                padding:0 !important;
+                box-sizing:border-box !important;
+            }
+            .gradio-container,.gradio-app,.gradio-interface{
                 width:100% !important;
                 max-width:100vw !important;
                 margin:0 !important;
@@ -336,9 +341,24 @@ def create_demo():
                 animation:slide 4s ease-in-out infinite,glow-hover 3s ease-in-out infinite;
                 transform:scale(1.05);
             }
-            button[aria-label="Fullscreen"],button[aria-label="Share"],button[aria-label="Download"]{
+            button[aria-label="Fullscreen"],button[aria-label="Share"]{
                 display:none !important;
             }
+            button[aria-label="Download"]{
+                transform:scale(3);
+                transform-origin:top right;
+                background:#000000 !important;
+                color:#FFFFFF !important;
+                border:1px solid #FFFFFF !important;
+                border-radius:4px;
+                padding:0.4rem !important;
+                margin:0.5rem !important;
+                box-shadow:0 0 8px rgba(255,255,255,0.3) !important;
+                transition:box-shadow 0.3s;
+            }
+            button[aria-label="Download"]:hover{
+                box-shadow:0 0 12px rgba(255,255,255,0.5) !important;
+            }
             .progress-text,.gr-progress,.gr-prose,.gr-log{
                 display:none !important;
             }
@@ -415,17 +435,17 @@ def create_demo():
                     show_download_button=False,
                     show_share_button=False,
                     interactive=True,
-                    elem_classes=["gradio-component", "image-container"],
+                    elem_classes=["gradio-component", "image-container"]
                 )
                 prompt = gr.Textbox(
                     label="Prompt",
                     lines=3,
-                    elem_classes=["gradio-component"],
+                    elem_classes=["gradio-component"]
                 )
                 run_button = gr.Button(
                     "Edit!",
                     variant="primary",
-                    elem_classes=["gradio-component", "gr-button-primary"],
+                    elem_classes=["gradio-component", "gr-button-primary"]
                 )
                 result_image = gr.Image(
                     label="Result Image",
@@ -433,7 +453,7 @@ def create_demo():
                     interactive=False,
                     show_download_button=True,
                     show_share_button=False,
-                    elem_classes=["gradio-component", "image-container"],
+                    elem_classes=["gradio-component", "image-container"]
                 )
         run_button.click(fn=infer, inputs=[input_image, prompt], outputs=[result_image])
         prompt.submit(fn=infer, inputs=[input_image, prompt], outputs=[result_image])