Update app.py

app.py

@@ -4,6 +4,7 @@ import random
 import spaces
 import torch
 import time
+import requests
 from diffusers import DiffusionPipeline, AutoencoderTiny
 from custom_pipeline import FluxWithCFGPipeline
 
@@ -32,14 +33,31 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
 pipe.to(device)
 print("✅ Flux pipeline loaded.")
 
+@spaces.GPU
+def translate_albanian_to_english(text):
+    """Translate Albanian text to English using LibreTranslate public instance."""
+    if not text.strip():
+        return ""
+    try:
+        response = requests.post(
+            "https://translate.fediverse.gay/api/v1/translate",
+            json={"q": text, "source": "sq", "target": "en"}
+        )
+        response.raise_for_status()
+        return response.json().get("translatedText", "")
+    except Exception as e:
+        print(f"Translation error: {e}")
+        return ""
+
 @spaces.GPU
 def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", randomize_seed: bool = False):
+    """Generate an image based on the provided English prompt."""
     if pipe is None:
         raise gr.Error("Pipeline nuk u ngarkua.")
 
     prompt_clean = prompt.strip()
     if not prompt_clean:
-        return None, seed, "Gabim: Plotësoni përshkrimin."
+        return None, seed, "Gabim: Përshkrimi është bosh."
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -51,7 +69,7 @@ def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", rand
     print(f"🎯 Prompt for generation: {prompt_final}")
 
     try:
-        generator = torch.Generator(device=device).manual_seed(int(seed))
+        generator = torch.Generator(device=device).manual_seed(seed)
         start_time = time.time()
         with torch.inference_mode():
             images = pipe(
@@ -73,38 +91,63 @@ def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", rand
         raise gr.Error(f"Gabim gjatë gjenerimit: {e}")
 
 # --- UI Layout ---
-with gr.Blocks(css="""
-body::before, body::after {
-    content: "";
-    display: block;
-    height: 320px;  /* This creates the vertical space below */
-    background-color: inherit;
-}
-button[aria-label="Download"] {
-    transform: scale(3);
-    transform-origin: top right;
-    margin: 0 !important;
-    padding: 6px !important;
-}
-""") as app:
-    gr.Markdown("# Gjenerues Imazhesh")
-    gr.Markdown("Gjenero imazhe fantastike sipas përshkrimit në anglisht")
-
-    with gr.Row():
-        with gr.Column(scale=2):
-            output_image = gr.Image(label="Imazhi i Gjeneruar", interactive=False, show_download_button=True)
-        with gr.Column(scale=1):
-            prompt_input = gr.Text(label="Përshkrimi (vetëm anglisht)", placeholder="Write your prompt here...", lines=3)
-            generate_btn = gr.Button("🎨 Gjenero")
-            aspect_ratio = gr.Radio(label="Raporti i Imazhit", choices=list(ASPECT_RATIOS.keys()), value="16:9")
-            randomize_seed = gr.Checkbox(label="Përdor numër të rastësishëm", value=True)
-            latency = gr.Text(label="Koha", interactive=False)
-
-    generate_btn.click(
-        fn=generate_image,
-        inputs=[prompt_input, gr.Number(value=42, visible=False), aspect_ratio, randomize_seed],
-        outputs=[output_image, gr.Number(visible=False), latency],
-        show_progress="full"
-    )
-
-app.launch(share=True)
+def create_demo():
+    with gr.Blocks() as app:
+        gr.HTML("""
+        <style>
+        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;
+        }
+        </style>
+        """)
+
+        gr.Markdown("")
+        gr.Markdown("")
+
+        with gr.Row():
+            with gr.Column(scale=2):
+                output_image = gr.Image(label="Imazhi i Gjeneruar", interactive=False, show_download_button=True)
+            with gr.Column(scale=1):
+                prompt_albanian = gr.Textbox(label="Përshkrimi në Shqip", placeholder="Shkruani përshkrimin në shqip këtu...", lines=3)
+                prompt_english = gr.Textbox(label="Përshkrimi në Anglisht", placeholder="Përkthimi do të shfaqet këtu...", interactive=False, lines=3)
+                aspect_ratio = gr.Radio(label="Raporti i Imazhit", choices=list(ASPECT_RATIOS.keys()), value="16:9")
+                randomize_seed = gr.Checkbox(label="Përdor numër të rastësishëm", value=True)
+                generate_btn = gr.Button("🎨 Gjenero")
+                latency = gr.Textbox(label="Koha", interactive=False)
+
+        prompt_albanian.change(
+            fn=translate_albanian_to_english,
+            inputs=[prompt_albanian],
+            outputs=[prompt_english]
+        )
+
+        generate_btn.click(
+            fn=generate_image,
+            inputs=[prompt_english, gr.Number(value=42, visible=False), aspect_ratio, randomize_seed],
+            outputs=[output_image, gr.Number(visible=False), latency],
+            show_progress="full"
+        )
+
+    return app
+
+if __name__ == "__main__":
+    app = create_demo()
+    app.launch(share=True)