Update app.py

app.py

@@ -1,8 +1,6 @@
 import streamlit as st
-from transformers import MarianMTModel, MarianTokenizer
-import torch
 from deep_translator import GoogleTranslator
-import re
+import time
 
 # ================================
 # PAGE CONFIGURATION
@@ -166,12 +164,12 @@ def load_css():
     }
     
     /* Clear Button */
-    .clear-button button {
+    div[data-testid="column"]:last-child .stButton button {
         background: linear-gradient(145deg, #f093fb, #f5576c);
         box-shadow: 0 10px 25px rgba(245, 87, 108, 0.4);
     }
     
-    .clear-button button:hover {
+    div[data-testid="column"]:last-child .stButton button:hover {
         box-shadow: 0 15px 35px rgba(245, 87, 108, 0.6);
     }
     
@@ -202,18 +200,13 @@ def load_css():
         border-top-color: #ffffff !important;
     }
     
-    /* Columns */
-    .element-container {
-        margin-bottom: 20px;
-    }
-    
-    /* Success Message */
-    .stSuccess {
-        background: rgba(72, 187, 120, 0.2);
+    /* Success/Warning Messages */
+    .stSuccess, .stWarning, .stError {
+        background: rgba(255, 255, 255, 0.2);
         backdrop-filter: blur(10px);
         border-radius: 10px;
-        border: 1px solid rgba(72, 187, 120, 0.4);
-        color: white;
+        border: 1px solid rgba(255, 255, 255, 0.4);
+        color: white !important;
     }
     
     /* Footer */
@@ -246,19 +239,6 @@ def load_css():
 # TRANSLATION FUNCTIONS
 # ================================
 
-@st.cache_resource
-def load_translation_model():
-    """Load the translation model (cached for performance)"""
-    try:
-        # Using a multilingual model that supports Urdu and Pashto
-        model_name = "Helsinki-NLP/opus-mt-mul-en"
-        tokenizer = MarianTokenizer.from_pretrained(model_name)
-        model = MarianMTModel.from_pretrained(model_name)
-        return tokenizer, model
-    except Exception as e:
-        st.error(f"Error loading model: {e}")
-        return None, None
-
 def urdu_to_pashto_roman(urdu_text):
     """
     Translate Urdu to Pashto and romanize it
@@ -271,35 +251,44 @@ def urdu_to_pashto_roman(urdu_text):
         translator_to_pashto = GoogleTranslator(source='ur', target='ps')
         pashto_text = translator_to_pashto.translate(urdu_text)
         
-        # Step 2: Transliterate Pashto to Roman script
-        # Using English as intermediate for romanization
-        translator_to_roman = GoogleTranslator(source='ps', target='en')
-        temp_english = translator_to_roman.translate(pashto_text)
-        
-        # Step 3: Create phonetic Roman Urdu representation
-        # This gives us a romanized version
-        romanized = create_roman_representation(pashto_text, temp_english)
+        # Step 2: Create Roman representation
+        romanized = create_roman_representation(pashto_text)
         
         return romanized
     
     except Exception as e:
-        st.error(f"Translation error: {e}")
-        return "Translation failed. Please try again."
+        return f"Translation error: {str(e)}"
 
-def create_roman_representation(pashto_text, english_reference):
+def create_roman_representation(pashto_text):
     """
     Create a Roman Urdu representation of Pashto text
     """
-    # Pashto to Roman Urdu character mapping (phonetic)
+    # Comprehensive Pashto to Roman Urdu character mapping (phonetic)
     pashto_to_roman = {
-        'ا': 'a', 'ب': 'b', 'پ': 'p', 'ت': 't', 'ټ': 't', 'ث': 's',
-        'ج': 'j', 'چ': 'ch', 'ح': 'h', 'خ': 'kh', 'د': 'd', 'ډ': 'd',
-        'ذ': 'z', 'ر': 'r', 'ړ': 'r', 'ز': 'z', 'ژ': 'zh', 'ږ': 'g',
-        'س': 's', 'ش': 'sh', 'ښ': 'kh', 'ص': 's', 'ض': 'z', 'ط': 't',
-        'ظ': 'z', 'ع': 'a', 'غ': 'gh', 'ف': 'f', 'ق': 'q', 'ک': 'k',
-        'ګ': 'g', 'ل': 'l', 'م': 'm', 'ن': 'n', 'ڼ': 'n', 'و': 'o',
-        'ه': 'h', 'ۀ': 'a', 'ي': 'i', 'ې': 'e', 'ی': 'i', 'ئ': 'y',
-        ' ': ' ', '،': ',', '؟': '?', '۔': '.'
+        # Vowels
+        'ا': 'a', 'آ': 'aa', 'ع': 'a', 'ء': '',
+        # Consonants
+        'ب': 'b', 'پ': 'p', 'ت': 't', 'ټ': 't', 'ث': 's',
+        'ج': 'j', 'چ': 'ch', 'ح': 'h', 'خ': 'kh',
+        'د': 'd', 'ډ': 'd', 'ذ': 'z', 'ر': 'r', 'ړ': 'r',
+        'ز': 'z', 'ژ': 'zh', 'ږ': 'g', 'س': 's', 'ش': 'sh',
+        'ښ': 'x', 'ص': 's', 'ض': 'z', 'ط': 't', 'ظ': 'z',
+        'غ': 'gh', 'ف': 'f', 'ق': 'q', 'ک': 'k', 'ګ': 'g',
+        'ل': 'l', 'م': 'm', 'ن': 'n', 'ڼ': 'n',
+        'و': 'w', 'ؤ': 'o', 'ه': 'h', 'ۀ': 'a', 'ة': 'a',
+        'ي': 'y', 'ې': 'e', 'ی': 'i', 'ئ': 'y', 'ى': 'a',
+        # Diacritics and special marks
+        'َ': 'a', 'ُ': 'u', 'ِ': 'i', 'ً': 'an', 'ٌ': 'un', 'ٍ': 'in',
+        'ّ': '', 'ْ': '', 'ٔ': '', 'ٰ': 'a',
+        # Punctuation
+        ' ': ' ', '،': ',', '؛': ';', '؟': '?', '۔': '.', '!': '!',
+        '(': '(', ')': ')', '[': '[', ']': ']', '{': '{', '}': '}',
+        '-': '-', '_': '_', '«': '"', '»': '"', '"': '"', '"': '"',
+        # Numbers (Pashto/Arabic numerals to Roman)
+        '۰': '0', '۱': '1', '۲': '2', '۳': '3', '۴': '4',
+        '۵': '5', '۶': '6', '۷': '7', '۸': '8', '۹': '9',
+        '٠': '0', '١': '1', '٢': '2', '٣': '3', '٤': '4',
+        '٥': '5', '٦': '6', '٧': '7', '٨': '8', '٩': '9',
     }
     
     # Transliterate Pashto to Roman
@@ -307,17 +296,22 @@ def create_roman_representation(pashto_text, english_reference):
     for char in pashto_text:
         roman_text += pashto_to_roman.get(char, char)
     
-    # If transliteration is too rough, use English reference with Urdu touch
-    if len(roman_text.strip()) < 3:
-        return english_reference
+    # Clean up extra spaces and formatting
+    roman_text = ' '.join(roman_text.split())
     
-    return roman_text
+    return roman_text if roman_text.strip() else pashto_text
 
 # ================================
 # MAIN APPLICATION
 # ================================
 
 def main():
+    # Initialize session state
+    if 'translation_output' not in st.session_state:
+        st.session_state.translation_output = ""
+    if 'urdu_input' not in st.session_state:
+        st.session_state.urdu_input = ""
+    
     # Load custom CSS
     load_css()
     
@@ -335,41 +329,19 @@ def main():
         st.markdown("### 📝 Input (Urdu)")
         urdu_input = st.text_area(
             "Enter Urdu text",
+            value=st.session_state.urdu_input,
             height=250,
-            placeholder="یہاں اردو متن درج کریں...",
+            placeholder="یہاں اردو متن درج کریں... (Enter Urdu text here...)",
             label_visibility="collapsed",
-            key="urdu_input"
+            key="urdu_input_area"
         )
     
     with col2:
         st.markdown("### 🔤 Output (Pashto in Roman)")
-        if 'translation_output' not in st.session_state:
-            st.session_state.translation_output = ""
-        
-        translation_output = st.text_area(
+        st.text_area(
             "Pashto translation in Roman script",
             value=st.session_state.translation_output,
             height=250,
             placeholder="Translation will appear here...",
             label_visibility="collapsed",
-            key="output_display",
-            disabled=True
-        )
-    
-    # Buttons
-    col_btn1, col_btn2, col_btn3 = st.columns([1, 1, 1])
-    
-    with col_btn2:
-        if st.button("🔄 Translate", use_container_width=True):
-            if urdu_input.strip():
-                with st.spinner("✨ Translating..."):
-                    result = urdu_to_pashto_roman(urdu_input)
-                    st.session_state.translation_output = result
-                    st.rerun()
-            else:
-                st.warning("⚠️ Please enter some Urdu text first!")
-    
-    with col_btn3:
-        st.markdown('<div class="clear-button">', unsafe_allow_html=True)
-        if st.button("🗑️ Clear", use_container_width=True):
-            st.session_state.translation_output = ""
+            key="output_