Update app.py

app.py

@@ -1,46 +1,63 @@
 import streamlit as st
 import os
-import requests
+from PIL import Image
+import argostranslate.package
+import argostranslate.translate
 
-# --- CONFIG ---
 SIGN_IMAGE_FOLDER = "sign_images"
-HF_TOKEN = os.getenv("HF_API_TOKEN")
-if not HF_TOKEN:
-    st.error("Hugging Face API token not found. Please set 'HF_API_TOKEN' in your environment.")
-    st.stop()
 
-HF_MODEL = "facebook/mbart-large-50-many-to-many-mmt"
-
-# --- STREAMLIT UI ---
 st.set_page_config(page_title="DeafTranslator", layout="centered")
 st.title("🤟 DeafTranslator")
-st.markdown("""
-Translate **Urdu ↔ English** and view it in **Sign Language** (ASL).
-
-You can also input ASL signs manually to reverse translate them into English and then Urdu.
-""")
+st.markdown("Translate **Urdu** text to **English** and view it in **Sign Language** (ASL). Or enter signs manually to reverse translate back to Urdu.")
 
-# --- HELPER FUNCTIONS ---
+@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
+    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)
+    if from_lang and to_lang:
+        return from_lang.get_translation(to_lang)
+    return None
 
-def hf_translate(text, src_lang, tgt_lang):
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    payload = {
-        "inputs": text,
-        "parameters": {"src_lang": src_lang, "tgt_lang": tgt_lang},
-        "options": {"wait_for_model": True}
-    }
-    url = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
+@st.cache_resource
+def load_argos_english_to_urdu():
     try:
-        response = requests.post(url, headers=headers, json=payload)
-        # Check if the response is empty or invalid
-        if response.status_code != 200:
-            return f"Error: Received status code {response.status_code}"
-        result = response.json()
-        if not result or not isinstance(result, list):
-            return f"Error: Invalid response format. {response.text}"
-        return result[0]["translation_text"]
-    except Exception as e:
-        return f"Error: {e}"
+        argostranslate.package.update_package_index()
+        available_packages = argostranslate.package.get_available_packages()
+        for pkg in available_packages:
+            if pkg.from_code == "en" and pkg.to_code == "ur":
+                argostranslate.package.install_from_path(pkg.download())
+                break
+    except Exception:
+        pass
+    installed_languages = argostranslate.translate.get_installed_languages()
+    from_lang = next((lang for lang in installed_languages if lang.code == "en"), None)
+    to_lang = next((lang for lang in installed_languages if lang.code == "ur"), None)
+    if from_lang and to_lang:
+        return from_lang.get_translation(to_lang)
+    return None
+
+translator_ur_en = load_argos_urdu_to_english()
+translator_en_ur = load_argos_english_to_urdu()
+
+def translate_urdu_to_english(text):
+    if translator_ur_en:
+        return translator_ur_en.translate(text)
+    return ""
+
+def translate_english_to_urdu(text):
+    if translator_en_ur:
+        return translator_en_ur.translate(text)
+    return ""
 
 def find_image_path(name):
     for ext in ["jpg", "JPG"]:
@@ -49,72 +66,66 @@ def find_image_path(name):
             return path
     return None
 
-def display_sign_language(text):
+def display_sign_language(english_text):
     st.markdown("### 👐 Sign Language Representation")
-    if not text:
-        st.warning("No translation available to display in ASL.")
-        return
-    words = text.lower().split()
-
+    words = english_text.lower().split()
     for word in words:
         st.markdown(f"**{word}**")
         cols = st.columns(len(word))
-        for i, char in enumerate(word):
-            if char.isalpha():
-                img_path = find_image_path(char)
-                if img_path:
-                    cols[i].image(img_path, caption=char.upper(), width=70)
-                else:
-                    cols[i].warning(f"Missing: {char.upper()}")
-        # Show space image between words
-        space_img = find_image_path("space")
-        if space_img:
-            st.image(space_img, width=40)
+        for i, c in enumerate(word):
+            img_path = find_image_path(c)
+            if img_path:
+                with cols[i]:
+                    st.image(img_path, caption=c.upper(), width=80)
+            else:
+                st.warning(f"Missing image for: {c.upper()}")
+        space_path = find_image_path("space")
+        if space_path:
+            st.image(space_path, caption="", width=40)
 
-def render_sign_input():
-    st.markdown("### ✋ Manual Sign Input (A–Z & Space)")
-    selected_chars = st.multiselect(
-        "Select ASL Signs in Order",
-        options=[chr(i) for i in range(65, 91)] + ["space"],
-        default=[]
-    )
-    return selected_chars
+# Urdu to Sign Language
+st.header("🔹 Urdu to Sign Language")
+urdu_text = st.text_area("📝 Enter Urdu Text", height=150)
+if st.button("🔄 Translate & Show Signs"):
+    if urdu_text.strip():
+        english_translation = translate_urdu_to_english(urdu_text)
+        if english_translation:
+            st.markdown("### ✅ Translated English")
+            st.success(english_translation)
+            display_sign_language(english_translation)
+    else:
+        st.warning("Please enter some Urdu text.")
 
-# --- TRANSLATION SECTION ---
-tab1, tab2 = st.tabs(["Urdu → English → ASL", "ASL → English → Urdu"])
+# Manual Input for Reverse
+st.header("🔹 Manual ASL Input (A–Z & Space)")
+if "manual_signs" not in st.session_state:
+    st.session_state.manual_signs = ""
 
-with tab1:
-    urdu_text = st.text_area("📝 Enter Urdu Text", height=150)
-    if st.button("🔄 Translate & Show Signs"):
-        if not urdu_text.strip():
-            st.warning("Please enter Urdu text.")
-        else:
-            with st.spinner("Translating to English..."):
-                english = hf_translate(urdu_text, "ur_PK", "en_XX")
-                if "Error" in english or "Translation failed" in english:
-                    st.error(f"Translation failed: {english}")
-                else:
-                    st.markdown("### ✅ English Translation")
-                    st.success(english)
-                    display_sign_language(english)
+cols1 = st.columns(13)
+for i, c in enumerate("ABCDEFGHIJKLM"):
+    with cols1[i]:
+        if st.button(c):
+            st.session_state.manual_signs += c.lower()
+cols2 = st.columns(14)
+for i, c in enumerate("NOPQRSTUVWXYZ "):
+    label = "Space" if c == " " else c
+    val = " " if c == " " else c.lower()
+    with cols2[i]:
+        if st.button(label):
+            st.session_state.manual_signs += val
 
-with tab2:
-    sign_sequence = render_sign_input()
-    if st.button("🔁 Translate Signs"):
-        if not sign_sequence:
-            st.warning("Please select at least one sign.")
-        else:
-            english_text = " ".join(
-                ''.join(c.lower() for c in group) if "space" not in group else ''.join(c.lower() for c in group if c != "space")
-                for group in "".join(sign_sequence).split("space")
-            )
-            st.markdown("### 🔡 Interpreted English")
-            st.success(english_text)
+st.markdown("### 🔠 Current ASL Input")
+st.code(st.session_state.manual_signs.strip())
 
-            with st.spinner("Translating to Urdu..."):
-                urdu = hf_translate(english_text, "en_XX", "ur_PK")
-                if "Error" in urdu or "Translation failed" in urdu:
-                    st.error(f"Translation failed: {urdu}")
-                else:
-                    st.markdown("### 🌐 Urdu Translation")
-                    st.success(urdu)
+if st.button("🔁 Translate Signs → English → Urdu"):
+    asl_input = st.session_state.manual_signs.strip()
+    if asl_input:
+        english_text = asl_input.replace("  ", " ")
+        st.markdown("### ✅ Detected English")
+        st.success(english_text)
+        urdu_text = translate_english_to_urdu(english_text)
+        if urdu_text:
+            st.markdown("### 🇵🇰 Urdu Translation")
+            st.success(urdu_text)
+    else:
+        st.warning("Please input some ASL letters.")