Update app.py

app.py

@@ -1,68 +1,63 @@
 import streamlit as st
-from PIL import Image
 import os
+from PIL import Image
 import argostranslate.package
 import argostranslate.translate
 
-# Load Argos Translate Urdu to English model (first time only)
-def load_argos_translate_model():
-    argostranslate.package.update_package_index()  # 🔧 Add this line
-    available_packages = argostranslate.package.get_available_packages()
-    for pkg in available_packages:
-        if pkg.from_code == "ur" and pkg.to_code == "en":
-            argostranslate.package.install_from_path(pkg.download())
-            break
+# Streamlit UI setup
+st.set_page_config(page_title="DeafTranslator", layout="centered")
+st.title("🤟 DeafTranslator")
+st.markdown("Translate **Urdu** text to **English** and view it in **Sign Language** (ASL).")
+
+SIGN_IMAGE_FOLDER = "sign_images"  # Place images like a.jpg, b.jpg, hello.jpg, etc. here
 
-# Check if translation model is installed
-installed_languages = argostranslate.translate.get_installed_languages()
-urdu_to_english = None
-for lang in installed_languages:
-    if lang.code == "ur":
-        for to_lang in lang.translations:
-            if to_lang.to_lang.code == "en":
-                urdu_to_english = to_lang
+@st.cache_resource
+def load_argos_urdu_to_english():
+    try:
+        argostranslate.package.update_package_index()
+        available_packages = argostranslate.package.get_available_packages()
+        for pkg in available_packages:
+            if pkg.from_code == "ur" and pkg.to_code == "en":
+                argostranslate.package.install_from_path(pkg.download())
                 break
+    except Exception:
+        pass  # ignore if already installed or network issue
 
-# If not installed, download and install
-if urdu_to_english is None:
-    with st.spinner("Downloading translation model..."):
-        load_argos_translate_model()
-        installed_languages = argostranslate.translate.get_installed_languages()
-        for lang in installed_languages:
-            if lang.code == "ur":
-                for to_lang in lang.translations:
-                    if to_lang.to_lang.code == "en":
-                        urdu_to_english = to_lang
-                        break
+    installed_languages = argostranslate.translate.get_installed_languages()
+    from_lang = next((lang for lang in installed_languages if lang.code == "ur"), None)
+    to_lang = next((lang for lang in installed_languages if lang.code == "en"), None)
 
-# Streamlit UI
-st.set_page_config(page_title="DeafTranslator", layout="centered")
-st.title("🤟 DeafTranslator")
-st.markdown("Translate **Urdu** text to **English** and view it in **Sign Language** (ASL).")
+    if from_lang and to_lang:
+        return argostranslate.translate.get_translation(from_lang, to_lang)
+    return None
 
-SIGN_IMAGE_FOLDER = "sign_images"  # Ensure images like a.jpg, b.jpg etc. are in this folder
+translator = load_argos_urdu_to_english()
 
 def translate_urdu_to_english(urdu_text):
-    try:
-        return urdu_to_english.translate(urdu_text)
-    except Exception as e:
-        st.error(f"Translation failed: {str(e)}")
-        return ""
+    if translator:
+        try:
+            return translator.translate(urdu_text)
+        except Exception as e:
+            st.error(f"Translation failed: {e}")
+    else:
+        st.error("Translation model not loaded.")
+    return ""
 
 def display_sign_language(english_text):
     words = english_text.lower().split()
     st.markdown("### 👐 Sign Language Representation")
     for word in words:
-        word_image_path = os.path.join(SIGN_IMAGE_FOLDER, f"{word}.jpg")
-        if os.path.exists(word_image_path):
-            st.image(word_image_path, caption=word, width=120)
+        word_img_path = os.path.join(SIGN_IMAGE_FOLDER, f"{word}.jpg")
+        if os.path.exists(word_img_path):
+            st.image(word_img_path, caption=word, width=120)
         else:
             for char in word:
                 if char.isalpha():
-                    char_image_path = os.path.join(SIGN_IMAGE_FOLDER, f"{char}.jpg")
-                    if os.path.exists(char_image_path):
-                        st.image(char_image_path, caption=char.upper(), width=80)
+                    char_img_path = os.path.join(SIGN_IMAGE_FOLDER, f"{char}.jpg")
+                    if os.path.exists(char_img_path):
+                        st.image(char_img_path, caption=char.upper(), width=80)
 
+# --- Streamlit UI ---
 urdu_text = st.text_area("📝 Enter Urdu Text", height=150)
 
 if st.button("🔄 Translate & Show Signs"):
@@ -71,6 +66,7 @@ if st.button("🔄 Translate & Show Signs"):
     else:
         with st.spinner("Translating..."):
             english_translation = translate_urdu_to_english(urdu_text)
+
         if english_translation:
             st.markdown("### ✅ Translated English")
             st.success(english_translation)