Update app.py

app.py

@@ -9,20 +9,20 @@ from diffusers import DiffusionPipeline, AutoencoderTiny
 from custom_pipeline import FluxWithCFGPipeline
 from libretranslatepy import LibreTranslateAPI
 
-# --- Start LibreTranslate Server ---
+# Start LibreTranslate Server
 def start_libretranslate():
     from libretranslate.main import start
     start(host="0.0.0.0", port=5000, no_cache=True)
 
 threading.Thread(target=start_libretranslate, daemon=True).start()
-time.sleep(5)  # Wait for server startup
+time.sleep(5)
 lt = LibreTranslateAPI("http://localhost:5000")
 
-# --- Torch Optimizations ---
+# Torch Optimizations
 torch.backends.cuda.matmul.allow_tf32 = True
 torch.backends.cudnn.benchmark = True
 
-# --- Constants ---
+# Constants
 MAX_SEED = np.iinfo(np.int32).max
 DEFAULT_WIDTH = 1024
 DEFAULT_HEIGHT = 576
@@ -33,7 +33,7 @@ ASPECT_RATIOS = {
 }
 INFERENCE_STEPS = 8
 
-# --- Device and Model Setup ---
+# Device and Model Setup
 dtype = torch.float16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -45,24 +45,24 @@ print("✅ Flux pipeline loaded.")
 
 @spaces.GPU
 def translate_albanian_to_english(text):
-    """Translate Albanian text to English using local LibreTranslate."""
+    """Translate Albanian to English using local LibreTranslate."""
     if not text.strip():
         return ""
     for attempt in range(2):
         try:
             translated = lt.translate(text, source_lang="sq", target_lang="en")
             if translated:
-                print(f"LibreTranslate response: {translated}")  # Debug
+                print(f"LibreTranslate response: {translated}")
                 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."
+                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."
 
 @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."""
+    """Generate image from English prompt."""
     if pipe is None:
         raise gr.Error("Pipeline nuk u ngarkua.")
 
@@ -101,7 +101,7 @@ 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}")
 
-# --- UI Layout ---
+# UI Layout
 def create_demo():
     with gr.Blocks() as app:
         gr.HTML("""