Update app.py

app.py

@@ -4,11 +4,8 @@ import random
 import spaces
 import torch
 import time
-import os
 from diffusers import DiffusionPipeline, AutoencoderTiny
 from custom_pipeline import FluxWithCFGPipeline
-from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration
-from langdetect import detect
 
 # --- Torch Optimizations ---
 torch.backends.cuda.matmul.allow_tf32 = True
@@ -35,41 +32,6 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
 pipe.to(device)
 print("✅ Flux pipeline loaded.")
 
-# --- Load M2M100 Translator ---
-try:
-    print("⏳ Loading M2M100 tokenizer and model...")
-    tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
-    model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M").to(device)
-    print("✅ M2M100 loaded.")
-except Exception as e:
-    print(f"❌ Failed to load M2M100: {e}")
-    tokenizer = None
-    model = None
-
-def translate_sq_to_en(text):
-    if not tokenizer or not model:
-        print("⚠️ Translator not loaded, returning original text")
-        return text
-    try:
-        tokenizer.src_lang = "sq"
-        encoded = tokenizer(text, return_tensors="pt").to(device)
-        generated = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id("en"))
-        translated = tokenizer.batch_decode(generated, skip_special_tokens=True)[0]
-        print(f"🌐 Translation successful: {translated}")
-        return translated
-    except Exception as e:
-        print(f"❌ Translation failed: {e}")
-        return text
-
-def is_albanian(text):
-    try:
-        lang = detect(text)
-        print(f"🕵️ Language detected: {lang}")
-        return lang == "sq"
-    except Exception as e:
-        print(f"⚠️ Language detection failed: {e}")
-        return False
-
 @spaces.GPU
 def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", randomize_seed: bool = False):
     if pipe is None:
@@ -79,21 +41,14 @@ def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", rand
     if not prompt_clean:
         return None, seed, "Gabim: Plotësoni përshkrimin."
 
-    if not is_albanian(prompt_clean):
-        return None, seed, "Ju lutemi shkruani vetëm në gjuhën shqipe."
-
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
     width, height = ASPECT_RATIOS.get(aspect_ratio, (DEFAULT_WIDTH, DEFAULT_HEIGHT))
 
-    # Translate Albanian prompt to English
-    prompt_final = translate_sq_to_en(prompt_clean)
-
-    # Add quality tags for generation
-    prompt_final += ", ultra realistic, sharp, 8k resolution"
+    prompt_final = prompt_clean + ", ultra realistic, sharp, 8k resolution"
 
-    print(f"🎯 Final prompt for generation: {prompt_final}")
+    print(f"🎯 Prompt for generation: {prompt_final}")
 
     try:
         generator = torch.Generator(device=device).manual_seed(int(seed))
@@ -117,20 +72,12 @@ def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", rand
             torch.cuda.empty_cache()
         raise gr.Error(f"Gabim gjatë gjenerimit: {e}")
 
-# --- Examples ---
-examples = [
-    "Qytet futuristik natën me drita neon",
-    "Një mace e bardhë që mban një tabelë përshëndetëse",
-    "Një astronaut që del nga një vezë në Hënë",
-    "Pamje nga një shtëpi moderne në stilin Minecraft"
-]
-
 # --- UI Layout ---
 with gr.Blocks(css="""
 body::before, body::after {
     content: "";
     display: block;
-    height: 640px;
+    height: 640px;  /* This creates the vertical space below */
     background-color: inherit;
 }
 button[aria-label="Download"] {
@@ -141,28 +88,18 @@ button[aria-label="Download"] {
 }
 """) as app:
     gr.Markdown("# Gjenerues Imazhesh")
-    gr.Markdown("Gjenero imazhe fantastike sipas përshkrimit me inteligjencë artificiale")
+    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 në shqip)", placeholder="Shkruani vetëm në gjuhën shqipe...", lines=3)
+            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)
 
-    gr.Markdown("### 📌 Shembuj Frymëzues")
-    gr.Examples(
-        examples=examples,
-        fn=generate_image,
-        inputs=[prompt_input],
-        outputs=[output_image, gr.Number(visible=False), latency],
-        cache_examples=True,
-        cache_mode="eager"
-    )
-
     generate_btn.click(
         fn=generate_image,
         inputs=[prompt_input, gr.Number(value=42, visible=False), aspect_ratio, randomize_seed],