Update app.py

app.py

@@ -3,61 +3,57 @@ import requests
 import os
 from PIL import Image
 
-# Hugging Face model for Urdu to English translation (ensure it supports ur -> en)
-HF_API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-ur-en"
-HF_API_TOKEN = os.getenv("HF_API_TOKEN")  # Set this in your environment
+# Streamlit config
+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).")
 
-headers = {
-    "Authorization": f"Bearer {HF_API_TOKEN}"
-}
+# Hugging Face translation model (no token required for public models, but safer with token)
+TRANSLATE_API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-ur-en"
+HF_API_TOKEN = st.secrets.get("HF_API_TOKEN", None)
+HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"} if HF_API_TOKEN else {}
 
-# Sign images folder (e.g., a.jpg, b.jpg, hello.jpg, etc.)
-SIGN_IMAGE_FOLDER = "sign_images"
+# Sign language images folder (alphabet or word-based)
+SIGN_IMAGE_FOLDER = "sign_images"  # Place a.jpg, b.jpg, etc. here
 
 def translate_urdu_to_english(urdu_text):
     payload = {"inputs": urdu_text}
-    response = requests.post(HF_API_URL, headers=headers, json=payload)
-
     try:
+        response = requests.post(TRANSLATE_API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
-        return response.json()[0]['translation_text']
+        result = response.json()
+        return result[0]['translation_text']
     except requests.exceptions.HTTPError as err:
         st.error(f"Translation failed: {response.status_code} - {response.text}")
-        return ""
     except Exception as e:
         st.error(f"Unexpected error: {str(e)}")
-        return ""
-
+    return ""
 
-def display_sign_language(text):
-    words = text.lower().split()
+def display_sign_language(english_text):
+    words = english_text.lower().split()
+    st.markdown("### 👐 Sign Language Representation")
     for word in words:
-        img_path_word = os.path.join(SIGN_IMAGE_FOLDER, f"{word}.jpg")
-        if os.path.exists(img_path_word):
-            st.image(img_path_word, caption=word)
+        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)
         else:
-            # Fallback to letter-by-letter spelling
             for char in word:
                 if char.isalpha():
-                    img_path_char = os.path.join(SIGN_IMAGE_FOLDER, f"{char}.jpg")
-                    if os.path.exists(img_path_char):
-                        st.image(img_path_char, caption=char)
+                    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)
 
-# Streamlit UI
-st.set_page_config(page_title="DeafTranslator", layout="centered")
-st.title("🤟 DeafTranslator")
-st.markdown("Translate **Urdu text** to **English** and see it in **sign language**.")
+# --- Streamlit App UI ---
+urdu_text = st.text_area("📝 Enter Urdu Text", height=150)
 
-urdu_text = st.text_area("Enter Urdu Text", height=150)
-
-if st.button("Translate to Sign Language"):
+if st.button("🔄 Translate & Show Signs"):
     if not urdu_text.strip():
         st.warning("Please enter some Urdu text.")
     else:
         with st.spinner("Translating..."):
-            english_text = translate_urdu_to_english(urdu_text)
-        if english_text:
-            st.subheader("Translated English Text")
-            st.success(english_text)
-            st.subheader("Sign Language Representation")
-            display_sign_language(english_text)
+            english_translation = translate_urdu_to_english(urdu_text)
+
+        if english_translation:
+            st.markdown("### ✅ Translated English")
+            st.success(english_translation)
+            display_sign_language(english_translation)