Update app.py

app.py

@@ -9,7 +9,7 @@ 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"  # Make sure this folder contains a.jpg, b.jpg, etc.
+SIGN_IMAGE_FOLDER = "sign_images"  # Place images like a.jpg, A.JPG, hello.jpg, etc. here
 
 @st.cache_resource
 def load_argos_urdu_to_english():
@@ -43,18 +43,26 @@ def translate_urdu_to_english(urdu_text):
         st.error("Translation model not loaded.")
     return ""
 
+def find_image_path(name):
+    """Try to find image with common extensions."""
+    for ext in ["jpg", "JPG"]:
+        path = os.path.join(SIGN_IMAGE_FOLDER, f"{name}.{ext}")
+        if os.path.exists(path):
+            return path
+    return None
+
 def display_sign_language(english_text):
     words = english_text.lower().split()
     st.markdown("### 👐 Sign Language Representation")
     for word in words:
-        word_img_path = os.path.join(SIGN_IMAGE_FOLDER, f"{word}.jpg")
-        if os.path.exists(word_img_path):
+        word_img_path = find_image_path(word)
+        if word_img_path:
             st.image(word_img_path, caption=word, width=120)
         else:
             for char in word:
                 if char.isalpha():
-                    char_img_path = os.path.join(SIGN_IMAGE_FOLDER, f"{char.lower()}.jpg")
-                    if os.path.exists(char_img_path):
+                    char_img_path = find_image_path(char.lower())
+                    if char_img_path:
                         st.image(char_img_path, caption=char.upper(), width=80)
                     else:
                         st.warning(f"Missing image for: {char.upper()}")