Update app.py

app.py

@@ -38,16 +38,22 @@ def translate_albanian_to_english(text):
     """Translate Albanian text to English using LibreTranslate public instance."""
     if not text.strip():
         return ""
-    try:
-        response = requests.post(
-            "https://translate.fediverse.gay/api/v1/translate",
-            json={"q": text, "source": "sq", "target": "en"}
-        )
-        response.raise_for_status()
-        return response.json().get("translatedText", "")
-    except Exception as e:
-        print(f"Translation error: {e}")
-        return ""
+    for attempt in range(2):  # Retry once
+        try:
+            response = requests.post(
+                "https://libretranslate.de/api/translate",
+                json={"q": text, "source": "sq", "target": "en"},
+                timeout=5  # 5-second timeout
+            )
+            response.raise_for_status()
+            translated = response.json().get("translatedText", "")
+            if translated:
+                return translated
+        except Exception as e:
+            print(f"Translation error (attempt {attempt + 1}): {e}")
+            if attempt == 1:  # Last attempt
+                return f"Gabim në përkthim: {str(e)}. Duke përdorur tekstin origjinal: {text}"
+    return text  # Fallback to original text
 
 @spaces.GPU
 def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", randomize_seed: bool = False):
@@ -124,13 +130,18 @@ def create_demo():
 
         with gr.Row():
             with gr.Column(scale=2):
-                output_image = gr.Image(label="Imazhi i Gjeneruar", interactive=False, show_download_button=True)
+                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="16:9")
-                randomize_seed = gr.Checkbox(label="Përdor numër të rastësishëm", value=True)
-                generate_btn = gr.Button("🎨 Gjenero")
+                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(
@@ -149,5 +160,6 @@ def create_demo():
     return app
 
 if __name__ == "__main__":
+    print(f"Gradio version: {gr.__version__}")
     app = create_demo()
     app.launch(share=True)