Update app.py

app.py

@@ -57,21 +57,18 @@ def translate_albanian_to_english(text):
     return f"Përkthimi dështoi"
 
 @spaces.GPU
-def generate_image_from_albanian(prompt_albanian: str, seed: int = 42, aspect_ratio: str = "1:1", randomize_seed: bool = False):
+def generate_image_from_albanian(prompt_albanian: str, aspect_ratio: str = "1:1"):
     """Translate Albanian prompt to English and generate image."""
     if not prompt_albanian.strip():
-        return None, seed, "Gabim: Përshkrimi është bosh."
+        return None
 
     # Translate stealthily
     prompt_english = translate_albanian_to_english(prompt_albanian)
     if prompt_english.startswith("Përkthimi dështoi"):
-        return None, seed, prompt_english
+        return None
 
     if pipe is None:
-        return None, seed, "Pipeline nuk u ngarkua."
-
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
+        return None
 
     width, height = ASPECT_RATIOS.get(aspect_ratio, (DEFAULT_WIDTH, DEFAULT_HEIGHT))
 
@@ -80,8 +77,8 @@ def generate_image_from_albanian(prompt_albanian: str, seed: int = 42, aspect_ra
     print(f"🎯 Prompt for generation: {prompt_final}")
 
     try:
+        seed = random.randint(0, MAX_SEED)
         generator = torch.Generator(device=device).manual_seed(seed)
-        start_time = time.time()
         with torch.inference_mode():
             images = pipe(
                 prompt=prompt_final,
@@ -93,13 +90,11 @@ def generate_image_from_albanian(prompt_albanian: str, seed: int = 42, aspect_ra
                 return_dict=False
             )
         image = images[0][0]
-        latency = time.time() - start_time
-        status = f"Koha e përpunimit: {latency:.2f} sekonda"
-        return image, seed, status
+        return image
     except Exception as e:
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
-        return None, seed, f"Gabim gjatë gjenerimit: {str(e)}"
+        return None
 
 # UI Layout
 def create_demo():