Update app.py

app.py

@@ -4,9 +4,19 @@ import random
 import spaces
 import torch
 import time
+import threading
 from diffusers import DiffusionPipeline, AutoencoderTiny
 from custom_pipeline import FluxWithCFGPipeline
-from googletrans import Translator, LANGUAGES
+from libretranslate import LibreTranslate
+
+# --- Start LibreTranslate Server ---
+def start_libretranslate():
+    from libretranslate import start_server
+    start_server(host="localhost", port=5000)
+
+threading.Thread(target=start_libretranslate, daemon=True).start()
+time.sleep(2)  # Wait for server to start
+lt = LibreTranslate(host="localhost", port=5000)
 
 # --- Torch Optimizations ---
 torch.backends.cuda.matmul.allow_tf32 = True
@@ -35,17 +45,19 @@ print("✅ Flux pipeline loaded.")
 
 @spaces.GPU
 def translate_albanian_to_english(text):
-    """Translate Albanian text to English using Google Translate."""
+    """Translate Albanian text to English using local LibreTranslate."""
     if not text.strip():
         return ""
-    translator = Translator()
-    try:
-        translated = translator.translate(text, src="sq", dest="en")
-        print(f"Google Translate response: {translated.text}")  # Debug
-        return translated.text
-    except Exception as e:
-        print(f"Translation error: {e}")
-        return f"Përkthimi dështoi: {str(e)}. Ju lutem provoni përsëri ose përdorni anglisht."
+    for attempt in range(2):
+        try:
+            translated = lt.translate(text, source="sq", target="en")
+            if translated:
+                print(f"LibreTranslate response: {translated}")  # Debug
+                return translated
+        except Exception as e:
+            print(f"Translation error (attempt {attempt + 1}): {e}")
+            if attempt == 1:
+                return f"Përkthimi dështoi: {str(e)}. Ju lutem provoni përsëri ose përdorni anglisht."
     return f"Përkthimi dështoi. Ju lutem provoni përsëri ose përdorni anglisht."
 
 @spaces.GPU