Update app.py

app.py

@@ -53,25 +53,29 @@ def translate_albanian_to_english(text):
         except Exception as e:
             print(f"Translation error (attempt {attempt + 1}): {e}")
             if attempt == 1:
-                return f"Përkthimi dështoi: {str(e)}. Provoni përsëri ose përdorni anglisht."
-    return f"Përkthimi dështoi. Provoni përsëri ose përdorni anglisht."
+                return f"Përkthimi dështoi: {str(e)}"
+    return f"Përkthimi dështoi"
 
 @spaces.GPU
-def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", randomize_seed: bool = False):
-    """Generate image from English prompt."""
-    if pipe is None:
-        raise gr.Error("Pipeline nuk u ngarkua.")
-
-    prompt_clean = prompt.strip()
-    if not prompt_clean:
+def generate_image_from_albanian(prompt_albanian: str, seed: int = 42, aspect_ratio: str = "1:1", randomize_seed: bool = False):
+    """Translate Albanian prompt to English and generate image."""
+    if not prompt_albanian.strip():
         return None, seed, "Gabim: Përshkrimi është bosh."
 
+    # Translate stealthily
+    prompt_english = translate_albanian_to_english(prompt_albanian)
+    if prompt_english.startswith("Përkthimi dështoi"):
+        return None, seed, prompt_english
+
+    if pipe is None:
+        return None, seed, "Pipeline nuk u ngarkua."
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
     width, height = ASPECT_RATIOS.get(aspect_ratio, (DEFAULT_WIDTH, DEFAULT_HEIGHT))
 
-    prompt_final = prompt_clean + ", ultra realistic, sharp, 8k resolution"
+    prompt_final = prompt_english + ", ultra realistic, sharp, 8k resolution"
 
     print(f"🎯 Prompt for generation: {prompt_final}")
 
@@ -95,7 +99,7 @@ def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", rand
     except Exception as e:
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
-        raise gr.Error(f"Gabim gjatë gjenerimit: {e}")
+        return None, seed, f"Gabim gjatë gjenerimit: {str(e)}"
 
 # UI Layout
 def create_demo():
@@ -134,26 +138,14 @@ def create_demo():
                 output_image = gr.Image(label="Imazhi i Gjeneruar", interactive=False)
             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="1:1")
                 randomize_seed = gr.Checkbox(label="Përdor numër të rastësishëm", value=False)
                 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],
+            fn=generate_image_from_albanian,
+            inputs=[prompt_albanian, gr.Number(value=42, visible=False), aspect_ratio, randomize_seed],
             outputs=[output_image, gr.Number(visible=False), latency],
             show_progress="full"
         )