Update app.py

app.py

@@ -1,11 +1,16 @@
 """
 ╔══════════════════════════════════════════════════════════════╗
 ║          Rahbar  |  رہبر                                     ║
-║  Pakistan's AI-Powered Civic Complaint System  v4.0          ║
+║  Pakistan's AI-Powered Civic Complaint System  v5.0          ║
 ╚══════════════════════════════════════════════════════════════╝
-Gradio 6 compatible — Professional English UI
-Fixes: GPS via gr.Button js=, camera via native gr.Image,
-       RAG chatbot, Groq Whisper STT, clean report headings
+Gradio 4/5/6 compatible
+Fixes v5.0:
+  - GPS button fully working (no js= hack, pure Python + JS injection)
+  - Map rendered via gr.HTML with working Leaflet (not sandboxed)
+  - Sindhi TTS fixed (uses 'ur' fallback since gtts has no 'sd')
+  - Chatbot with voice input + TTS output
+  - All UI text in Urdu (not Hindi/Punjabi heading labels)
+  - Gradio 6 compatible event wiring
 """
 
 import os, io, re, uuid, base64, datetime, urllib.parse
@@ -195,12 +200,12 @@ Always include GPS coordinates and photos with your complaint.""",
 ]
 
 
-# ─── RAG ENGINE — TF-IDF + Cosine Similarity ──────────────────
+# ─── RAG ENGINE ───────────────────────────────────────────────
 class RAGEngine:
     def __init__(self):
-        self.documents   = RAG_DOCUMENTS
-        self.vectorizer  = None
-        self.doc_matrix  = None
+        self.documents    = RAG_DOCUMENTS
+        self.vectorizer   = None
+        self.doc_matrix   = None
         self._initialized = False
 
     def initialize(self):
@@ -218,7 +223,7 @@ class RAGEngine:
                 analyzer='char_wb', ngram_range=(2, 5),
                 max_features=8000, sublinear_tf=True, min_df=1,
             )
-            self.doc_matrix  = self.vectorizer.fit_transform(corpus)
+            self.doc_matrix   = self.vectorizer.fit_transform(corpus)
             self._initialized = True
             return True
         except Exception as e:
@@ -248,9 +253,9 @@ class RAGEngine:
     def _keyword_fallback(self, query, top_k=3):
         q = query.lower()
         keywords = {
-            "Garbage":      ["garbage", "waste", "sanitation", "trash", "1139"],
-            "Pot Hole":     ["pothole", "pot hole", "road", "nha"],
-            "Pipe Leakage": ["water", "wasa", "pipe", "leakage", "contaminated"],
+            "Garbage":      ["garbage", "waste", "sanitation", "trash", "1139", "کچرا"],
+            "Pot Hole":     ["pothole", "pot hole", "road", "nha", "سڑک", "گڑھا"],
+            "Pipe Leakage": ["water", "wasa", "pipe", "leakage", "contaminated", "پانی", "پائپ"],
         }
         found_cat = None
         for cat, kws in keywords.items():
@@ -357,28 +362,33 @@ LEGAL_KB = {
 LOCALIZED = {
     "Garbage": {
         "English":          "Dumping garbage is a criminal offence under Punjab Waste Management Act 2014. Fine: Rs. 500-50,000. Helpline: 1139",
-        "اردو (Urdu)":      "کچرا پھینکنا پنجاب ویسٹ مینجمنٹ ایکٹ 2014 کے تحت جرم ہے۔ جرمانہ: 500 سے 50,000 روپے۔",
-        "پنجابی (Punjabi)": "کچرا سُٹنا پنجاب ویسٹ مینجمنٹ ایکٹ 2014 دے تحت جرم اے۔",
-        "سندھی (Sindhi)":   "ڪچرو اڇلائڻ پنجاب ويسٽ مئنيجمينٽ ايڪٽ 2014 تحت جرم آهي.",
+        "اردو (Urdu)":      "کچرا پھینکنا پنجاب ویسٹ مینجمنٹ ایکٹ 2014 کے تحت جرم ہے۔ جرمانہ: 500 سے 50,000 روپے۔ ہیلپ لائن: 1139",
+        "پنجابی (Punjabi)": "کچرا سُٹنا پنجاب ویسٹ مینجمنٹ ایکٹ 2014 دے تحت جرم اے۔ جرمانہ 500 توں 50,000 روپے۔ ہیلپ لائن: 1139",
+        "سندھی (Sindhi)":   "ڪچرو اڇلائڻ پنجاب ويسٽ مئنيجمينٽ ايڪٽ 2014 تحت جرم آهي. جرمانو: 500 کان 50,000 رپيا. مدد لاءِ: 1139",
     },
     "Pot Hole": {
         "English":          "Road repair is the government's legal obligation within 72 hours under Punjab LGA 2022. NHA: 051-9032800",
-        "اردو (Urdu)":      "سڑک کی مرمت پنجاب مقامی حکومت ایکٹ 2022 کے تحت 72 گھنٹوں میں حکومت کی ذمہ داری ہے۔",
-        "پنجابی (Punjabi)": "سڑک دی مرمت پنجاب LGA 2022 دے تحت 72 گھنٹیاں وچ سرکار دی ذمہ واری اے۔",
-        "سندھی (Sindhi)":   "سڙڪ جي مرمت پنجاب LGA 2022 تحت 72 ڪلاڪن ۾ حڪومت جي ذميواري آهي.",
+        "اردو (Urdu)":      "سڑک کی مرمت پنجاب مقامی حکومت ایکٹ 2022 کے تحت 72 گھنٹوں میں حکومت کی ذمہ داری ہے۔ NHA: 051-9032800",
+        "پنجابی (Punjabi)": "سڑک دی مرمت پنجاب LGA 2022 دے تحت 72 گھنٹیاں وچ سرکار دی ذمہ واری اے۔ NHA: 051-9032800",
+        "سندھی (Sindhi)":   "سڙڪ جي مرمت پنجاب LGA 2022 تحت 72 ڪلاڪن ۾ حڪومت جي ذميواري آهي. NHA: 051-9032800",
     },
     "Pipe Leakage": {
         "English":          "Pipe leakage must be repaired within 24 hours by WASA under Punjab Water Act 2019. WASA: 042-99200300",
-        "اردو (Urdu)":      "پائپ لیکیج کی مرمت پنجاب واٹر ایکٹ 2019 کے تحت WASA کی 24 گھنٹوں میں ذمہ داری ہے۔",
-        "پنجابی (Punjabi)": "پائپ لیکیج دی مرمت پنجاب واٹر ایکٹ 2019 دے تحت WASA دی 24 گھنٹیاں وچ ذمہ واری اے۔",
-        "سندھی (Sindhi)":   "پائپ ليڪيج جي مرمت پنجاب واٽر ايڪٽ 2019 تحت WASA جي 24 ڪلاڪن ۾ ذميواري آهي.",
+        "اردو (Urdu)":      "پائپ لیکیج کی مرمت پنجاب واٹر ایکٹ 2019 کے تحت WASA کی 24 گھنٹوں میں ذمہ داری ہے۔ WASA: 042-99200300",
+        "پنجابی (Punjabi)": "پائپ لیکیج دی مرمت پنجاب واٹر ایکٹ 2019 دے تحت WASA دی 24 گھنٹیاں وچ ذمہ واری اے۔ WASA: 042-99200300",
+        "سندھی (Sindhi)":   "پائپ ليڪيج جي مرمت پنجاب واٽر ايڪٽ 2019 تحت WASA جي 24 ڪلاڪن ۾ ذميواري آهي. WASA: 042-99200300",
     },
 }
 
 WASTE_CLASS_IDS = {24,25,26,27,28,32,33,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54}
+
+# ─── TTS language codes — Fixed Sindhi fallback ───────────────
+# gTTS does not support 'sd' (Sindhi), so we use 'ur' as closest fallback
 LANG_CODES = {
-    "English": "en", "اردو (Urdu)": "ur",
-    "پنجابی (Punjabi)": "pa", "سندھی (Sindhi)": "sd",
+    "English":          "en",
+    "اردو (Urdu)":      "ur",
+    "پنجابی (Punjabi)": "ur",   # Shahmukhi is same as Urdu phonetically for gTTS
+    "سندھی (Sindhi)":   "ur",   # gTTS has no Sindhi — Urdu is closest available
 }
 
 
@@ -451,9 +461,9 @@ def parse_gemini_response(text):
         m = re.search(pat, text, re.IGNORECASE)
         if m:
             val = m.group(1)
-            if key == "status":   r[key] = val.upper()
+            if key == "status":     r[key] = val.upper()
             elif key == "severity": r[key] = int(val)
-            else:                 r[key] = val
+            else:                   r[key] = val
     for pat, key in [
         (r"REASON:\s*(.+?)(?=REASON_URDU:|SEVERITY:|$)", "reason"),
         (r"REASON_URDU:\s*(.+?)(?=SEVERITY:|$)", "reason_urdu"),
@@ -469,9 +479,9 @@ def analyze_with_llama(issue, location, city, yolo_summary, severity, language="
     clean = issue.split(" ", 1)[-1]
     kb    = LEGAL_KB.get(clean, {})
     lang_map = {
-        "اردو (Urdu)":      "Respond entirely in Urdu.",
-        "پنجابی (Punjabi)": "Respond in Punjabi (Shahmukhi script).",
-        "سندھی (Sindhi)":   "Respond in Sindhi.",
+        "اردو (Urdu)":      "Respond entirely in Urdu script.",
+        "پنجابی (Punjabi)": "Respond in Punjabi Shahmukhi script (same as Urdu script but Punjabi language).",
+        "سندھی (Sindhi)":   "Respond in Sindhi script.",
     }
     lang_instruction = lang_map.get(language, "Respond in clear professional English.")
     if not GROQ_API_KEY:
@@ -520,9 +530,9 @@ def legal_chatbot_rag(user_message, history, language):
     retrieved_docs = rag_engine.retrieve(user_message, top_k=3)
     rag_context    = rag_engine.format_context(retrieved_docs)
     lang_map = {
-        "اردو (Urdu)":      "Respond entirely in Urdu.",
-        "پنجابی (Punjabi)": "Respond in Punjabi (Shahmukhi script).",
-        "سندھی (Sindhi)":   "Respond in Sindhi.",
+        "اردو (Urdu)":      "Respond entirely in Urdu script.",
+        "پنجابی (Punjabi)": "Respond in Punjabi Shahmukhi script.",
+        "سندھی (Sindhi)":   "Respond in Sindhi script.",
     }
     lang_instruction = lang_map.get(language, "Respond in clear professional English.")
     system_content = f"""You are Rahbar Legal Assistant — a civic rights advisor for Pakistani citizens.
@@ -561,7 +571,7 @@ RAG Context (from Knowledge Base):
         answer = resp.choices[0].message.content.strip()
         if retrieved_docs:
             refs    = [f"[{d['title'][:35]}...]" for d in retrieved_docs[:2]]
-            answer += f"\n\n_Sources: {' | '.join(refs)}_"
+            answer += f"\n\n_ماخذ: {' | '.join(refs)}_"
     except Exception as e:
         answer = f"Error: {e}"
     return history + [
@@ -570,7 +580,28 @@ RAG Context (from Knowledge Base):
     ], ""
 
 
-# ─── TTS ───────────────────────────────────────────────────��──
+def chatbot_voice_input(audio_file, current_text):
+    """STT for chatbot voice input"""
+    if audio_file is None:
+        return current_text or ""
+    transcribed = stt(audio_file)
+    return transcribed
+
+
+def chatbot_tts_output(history, language):
+    """Generate TTS for last chatbot response"""
+    if not history:
+        return None
+    last_msg = history[-1]
+    if last_msg.get("role") == "assistant":
+        text = last_msg.get("content", "")
+        # Strip markdown refs
+        text = re.sub(r'_ماخذ:.*?_', '', text, flags=re.DOTALL).strip()
+        return make_tts(text[:600], language)
+    return None
+
+
+# ─── TTS — Fixed Sindhi support ───────────────────────────────
 def make_tts(text, language):
     try:
         from gtts import gTTS
@@ -580,14 +611,23 @@ def make_tts(text, language):
         tts.save(path)
         return path
     except Exception as e:
-        print(f"TTS error: {e}")
-        return None
+        print(f"TTS error for lang={language}: {e}")
+        # Fallback: try English
+        try:
+            from gtts import gTTS
+            tts  = gTTS(text=str(text)[:600], lang="en", slow=False)
+            path = f"/tmp/tts_fallback_{uuid.uuid4().hex[:8]}.mp3"
+            tts.save(path)
+            return path
+        except Exception as e2:
+            print(f"TTS fallback error: {e2}")
+            return None
 
 
 # ─── STT — Groq Whisper primary, SpeechRecognition fallback ──
 def stt(audio_file):
     if audio_file is None:
-        return "No audio provided. Please record or upload audio first."
+        return "آڈیو نہیں ملی۔ براہ کرم پہلے آڈیو ریکارڈ کریں یا اپلوڈ کریں۔"
 
     def ensure_wav(path):
         if path.lower().endswith(".wav"):
@@ -610,11 +650,11 @@ def stt(audio_file):
                     model="whisper-large-v3", file=f, response_format="text"
                 )
             text = result if isinstance(result, str) else result.text
-            return text.strip() or "No speech detected in audio."
+            return text.strip() or "آڈیو میں کوئی آواز نہیں ملی۔"
         except Exception as e:
             groq_err = str(e)
     else:
-        groq_err = "GROQ_API_KEY not set"
+        groq_err = "GROQ_API_KEY سیٹ نہیں ہے"
 
     try:
         import speech_recognition as sr
@@ -625,10 +665,10 @@ def stt(audio_file):
             audio_data = recognizer.record(src)
         return recognizer.recognize_google(audio_data)
     except Exception as e2:
-        return (f"Transcription failed.\n"
-                f"Groq Whisper error: {groq_err}\n"
-                f"Fallback error: {e2}\n"
-                f"Tip: Set GROQ_API_KEY for reliable Whisper transcription.")
+        return (f"ٹرانسکرپشن ناکام۔\n"
+                f"Groq Whisper خرابی: {groq_err}\n"
+                f"Fallback خرابی: {e2}\n"
+                f"تجویز: GROQ_API_KEY سیٹ کریں۔")
 
 
 # ─── LAW REFERENCE ────────────────────────────────────────────
@@ -637,18 +677,18 @@ def law_info(issue, language):
     kb     = LEGAL_KB.get(clean, {})
     local  = LOCALIZED.get(clean, {}).get(language, "")
     rights = "\n".join(f"  * {r}" for r in kb.get("citizen_rights", []))
-    out    = f"## Legal Reference: {issue}\n\n### Applicable Laws\n"
+    out    = f"## قانونی حوالہ: {issue}\n\n### قابل اطلاق قوانین\n"
     for law in kb.get("laws", []):
         out += f"  * {law}\n"
-    out += (f"\n### Fine / Penalty\n{kb.get('fine','N/A')}\n"
-            f"\n### Responsible Authority\n{kb.get('authority','N/A')}\n"
-            f"\n### Helpline\n**{kb.get('hotline','N/A')}**\n"
-            f"\n### Mandated Response Time\n{kb.get('response','N/A')}\n"
-            f"\n### Your Rights as a Citizen\n{rights}\n"
-            f"\n### Escalation Path\n{kb.get('escalation','N/A')}\n")
+    out += (f"\n### جرمانہ / سزا\n{kb.get('fine','N/A')}\n"
+            f"\n### ذمہ دار ادارہ\n{kb.get('authority','N/A')}\n"
+            f"\n### ہیلپ لائن\n**{kb.get('hotline','N/A')}**\n"
+            f"\n### لازمی جوابی مدت\n{kb.get('response','N/A')}\n"
+            f"\n### شہری کے حقوق\n{rights}\n"
+            f"\n### اضافی شکایت کا راستہ\n{kb.get('escalation','N/A')}\n")
     if local:
-        out += f"\n---\n### Localized Message ({language})\n> {local}\n"
-    out += f"\n---\n*Source: {kb.get('dataset_ref','Pakistani civic law databases')}*"
+        out += f"\n---\n### مقامی پیغام ({language})\n> {local}\n"
+    out += f"\n---\n*ماخذ: {kb.get('dataset_ref','Pakistani civic law databases')}*"
     return out
 
 
@@ -656,7 +696,7 @@ def law_info(issue, language):
 def get_admin_stats():
     total = len(complaint_log)
     if total == 0:
-        return "No complaints filed yet.", ""
+        return "ابھی تک کوئی شکایت درج نہیں ہوئی۔", ""
     counts = {"Garbage": 0, "Pot Hole": 0, "Pipe Leakage": 0}
     cities, severities = {}, []
     for c in complaint_log:
@@ -667,29 +707,29 @@ def get_admin_stats():
         severities.append(c.get("severity", 5))
     avg_sev  = sum(severities) / len(severities) if severities else 0
     top_city = max(cities, key=cities.get) if cities else "N/A"
-    stats_md = f"""## Admin Dashboard
-| Metric | Value |
+    stats_md = f"""## ایڈمن ڈیش بورڈ
+| معلومات | تعداد |
 |--------|-------|
-| Total Complaints | **{total}** |
-| Average Severity | **{avg_sev:.1f}/10** |
-| Top City | **{top_city}** |
+| کل شکایات | **{total}** |
+| اوسط شدت | **{avg_sev:.1f}/10** |
+| سب سے زیادہ شکایات | **{top_city}** |
 
-### By Issue Type
-| Issue | Count |
+### نوع کے مطابق
+| مسئلہ | تعداد |
 |-------|-------|
-| Garbage | {counts['Garbage']} |
-| Pot Hole | {counts['Pot Hole']} |
-| Pipe Leakage | {counts['Pipe Leakage']} |
+| کچرا | {counts['Garbage']} |
+| سڑک کا گڑھا | {counts['Pot Hole']} |
+| پائپ لیکیج | {counts['Pipe Leakage']} |
 
-### By City
+### شہر کے مطابق
 """
     for city, cnt in sorted(cities.items(), key=lambda x: -x[1]):
         stats_md += f"| {city} | {cnt} |\n"
-    log_md = "## Recent Complaints\n\n"
+    log_md = "## حالیہ شکایات\n\n"
     for c in reversed(complaint_log[-10:]):
         log_md += (f"**{c['id']}** | {c['timestamp']} | {c['city']}, {c['location']} | "
-                   f"{c['issue']} | Severity {c['severity']}/10 | "
-                   f"Citizen: {c.get('name','N/A')} (CNIC: {c.get('cnic','N/A')})\n\n")
+                   f"{c['issue']} | شدت {c['severity']}/10 | "
+                   f"شہری: {c.get('name','N/A')} (CNIC: {c.get('cnic','N/A')})\n\n")
     return stats_md, log_md
 
 
@@ -700,7 +740,6 @@ def severity_icon(score):
     return "RED"
 
 
-# ─── Unicode bold for plain-text headings in gr.Textbox ───────
 def _ubold(text):
     result = []
     for ch in str(text):
@@ -791,13 +830,13 @@ def build_professional_report(complaint_id, timestamp, name, cnic, phone, city,
 def make_report(image, issue_type, city, location, name, cnic, phone,
                 description, language, enable_tts):
     if image is None:
-        return (None, "Please upload an image.", "", "", None, "", None)
+        return (None, "براہ کرم ایک تصویر اپلوڈ کریں۔", "", "", None, "", None)
     if not location.strip():
-        return (None, "Please enter a location.", "", "", None, "", None)
+        return (None, "براہ کرم مقام درج کریں۔", "", "", None, "", None)
     if not name.strip():
-        return (None, "Please enter your full name.", "", "", None, "", None)
+        return (None, "براہ کرم اپنا پورا نام درج کریں۔", "", "", None, "", None)
     if not cnic.strip():
-        return (None, "Please enter your CNIC number.", "", "", None, "", None)
+        return (None, "براہ کرم اپنا CNIC نمبر درج کریں۔", "", "", None, "", None)
 
     complaint_id = f"RB-{uuid.uuid4().hex[:8].upper()}"
     timestamp    = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -811,10 +850,10 @@ def make_report(image, issue_type, city, location, name, cnic, phone,
 
     if gemini_status == "REJECTED":
         return (annotated_img,
-                f"COMPLAINT REJECTED BY AI VERIFICATION\n\nReason: {gemini_reason}\n"
-                f"Confidence: {gemini_parsed.get('confidence','N/A')}\n\n"
-                f"Please upload a clear image showing the reported issue ({issue_type}).\n"
-                "Your complaint has NOT been logged.",
+                f"شکایت مسترد — AI تصدیق\n\nوجہ: {gemini_reason}\n"
+                f"اعتماد: {gemini_parsed.get('confidence','N/A')}\n\n"
+                f"براہ کرم مسئلے کی واضح تصویر اپلوڈ کریں ({issue_type})۔\n"
+                "آپ کی شکایت لاگ نہیں ہوئی۔",
                 "", "", None, complaint_id, None)
 
     if gemini_status == "UNKNOWN" and "GOOGLE_API_KEY not set" in gemini_raw:
@@ -841,16 +880,16 @@ def make_report(image, issue_type, city, location, name, cnic, phone,
         "name": name, "cnic": cnic, "phone": phone,
     })
 
-    wa_text = (f"Rahbar Complaint\nRef: {complaint_id}\nIssue: {issue_type}\n"
-               f"Location: {location}, {city}\nSeverity: {final_severity}/10\n"
-               f"Authority: {kb.get('authority','N/A')}\n"
-               f"Helpline: {kb.get('hotline','N/A')}\nFiled: {timestamp}")
-    wa_md = f"[Share on WhatsApp](https://wa.me/?text={urllib.parse.quote(wa_text[:1000])})"
+    wa_text = (f"رہبر شکایت\nRef: {complaint_id}\nمسئلہ: {issue_type}\n"
+               f"مقام: {location}, {city}\nشدت: {final_severity}/10\n"
+               f"ادارہ: {kb.get('authority','N/A')}\n"
+               f"ہیلپ لائن: {kb.get('hotline','N/A')}\nوقت: {timestamp}")
+    wa_md = f"[واٹس ایپ پر شیئر کریں](https://wa.me/?text={urllib.parse.quote(wa_text[:1000])})"
 
     report_tts_path = None
     if enable_tts:
-        tts_text = (f"Complaint {complaint_id} filed. Issue: {issue_type} "
-                    f"at {location}, {city}. Severity: {final_severity} out of 10. {local}")
+        tts_text = (f"شکایت {complaint_id} درج ہوئی۔ مسئلہ: {issue_type} "
+                    f"مقام: {location}, {city}۔ شدت: {final_severity} میں سے 10۔ {local}")
         report_tts_path = make_tts(tts_text, language)
 
     advice_tts_path = make_tts(llama_advice[:600], language)
@@ -858,10 +897,39 @@ def make_report(image, issue_type, city, location, name, cnic, phone,
 
 
 def update_areas(city):
-    areas = CITIES_AREAS.get(city, ["Enter area manually"])
+    areas = CITIES_AREAS.get(city, ["علاقہ درج کریں"])
     return gr.Dropdown(choices=areas, value=areas[0])
 
 
+# ─── GPS reverse geocode (server-side fallback) ───────────────
+def gps_reverse_geocode(lat_str, lng_str):
+    """Called from GPS JS via gr.Button, fills location textbox"""
+    try:
+        lat = float(lat_str)
+        lng = float(lng_str)
+    except Exception:
+        return ""
+    try:
+        import urllib.request, json
+        url = f"https://nominatim.openstreetmap.org/reverse?format=jsonv2&lat={lat}&lon={lng}&zoom=17&addressdetails=1"
+        req = urllib.request.Request(url, headers={"User-Agent": "Rahbar/5.0"})
+        with urllib.request.urlopen(req, timeout=5) as resp:
+            data = json.loads(resp.read())
+        a = data.get("address", {})
+        parts = []
+        if a.get("road") or a.get("pedestrian"):
+            parts.append(a.get("road") or a.get("pedestrian"))
+        if a.get("suburb") or a.get("neighbourhood"):
+            parts.append(a.get("suburb") or a.get("neighbourhood"))
+        if a.get("city_district") or a.get("county"):
+            parts.append(a.get("city_district") or a.get("county"))
+        if not parts and data.get("display_name"):
+            parts = data["display_name"].split(",")[:3]
+        return ", ".join(p.strip() for p in parts if p.strip()) or f"GPS:{lat:.5f},{lng:.5f}"
+    except Exception:
+        return f"GPS:{lat:.5f},{lng:.5f}"
+
+
 # ─── CSS ──────────────────────────────────────────────────────
 CSS = """
 @import url('https://fonts.googleapis.com/css2?family=Noto+Nastaliq+Urdu:wght@400;700&family=Playfair+Display:wght@700;900&family=DM+Sans:wght@300;400;500;600&family=DM+Mono:wght@400;500&display=swap');
@@ -923,6 +991,9 @@ label,.gradio-container .label-wrap span{color:var(--text-primary)!important}
 .gradio-container select,.gradio-container [data-testid="dropdown"]{background:var(--surface)!important;color:var(--text-primary)!important;border-color:var(--border-strong)!important}
 .gradio-container .message.user{background:var(--surface3)!important;color:var(--text-primary)!important}
 .gradio-container .message.bot{background:var(--surface2)!important;color:var(--text-primary)!important}
+#map_container{border-radius:12px;overflow:hidden;border:2px solid var(--border-strong);margin:8px 0}
+#rahbar_map{height:260px;width:100%;background:#e8f5e9}
+.gps-status{font-size:.82rem;min-height:28px;padding:7px 12px;border-radius:8px;background:var(--surface2);border:1px solid var(--border);color:var(--g5);font-weight:500;margin-bottom:8px}
 ::-webkit-scrollbar{width:6px;height:6px}
 ::-webkit-scrollbar-track{background:var(--surface2)}
 ::-webkit-scrollbar-thumb{background:var(--g6);border-radius:3px}
@@ -940,220 +1011,201 @@ HEADER_HTML = """
   <span class="badge badge-ai">YOLOv8</span>
   <span class="badge badge-ai">Llama 3.3</span>
   <span class="badge badge-ai">RAG Chatbot</span>
-  <span class="badge badge-pk">4 Languages</span>
+  <span class="badge badge-pk">4 زبانیں</span>
   <span class="badge badge-live">LIVE</span>
 </div>
 """
 
-# ─── MAP WIDGET (inside gr.HTML = sandboxed iframe) ───────────
-# Draws the Leaflet map. GPS logic runs from parent page via gr.Button(js=).
-# Map receives coordinates via window.postMessage and moves the pin.
+# ─── MAP HTML — Leaflet loaded directly (no iframe sandbox) ───
+# This is injected into the Gradio page DOM directly.
+# The GPS button Python handler updates lat_state/lng_state,
+# and a JS watcher moves the map pin accordingly.
 MAP_HTML = """
-<link rel="stylesheet" href="https://unpkg.com/leaflet@1.9.4/dist/leaflet.css"/>
-<script src="https://unpkg.com/leaflet@1.9.4/dist/leaflet.js"></script>
-
-<div style="margin:6px 0 4px">
-  <div id="gpsStatusDiv" style="
-    font-size:.82rem;min-height:26px;margin-bottom:8px;font-weight:500;
-    padding:7px 12px;color:#25a06b;border-radius:8px;
-    background:#f4f8f5;border:1px solid #b8d9c5;">
-    Map ready. Use the GPS button above or click the map to place a pin.
-  </div>
-  <div id="rahbarMapDiv" style="
-    height:240px;width:100%;border-radius:12px;
-    border:2px solid #1f7a52;margin-bottom:6px;
-    background:#e8f5e9;position:relative;overflow:hidden;">
+<div id="map_container">
+  <div id="gps_status_bar" class="gps-status">
+    نقشہ تیار ہے۔ GPS بٹن دبائیں یا نقشے پر کلک کریں۔
   </div>
-  <p style="font-size:.71rem;color:#5a8a6e;margin:2px 0 8px">
-    Click map or drag pin to adjust location &middot; GPS button auto-fills your address
+  <div id="rahbar_map"></div>
+  <p style="font-size:.71rem;color:var(--text-muted);margin:4px 0 0;padding:0 4px">
+    نقشے پر کلک کریں یا پن گھسیٹیں — GPS بٹن آپ کا پتہ خود بھرے گا
   </p>
 </div>
 
+<link rel="stylesheet" href="https://unpkg.com/leaflet@1.9.4/dist/leaflet.css"/>
 <script>
-(function(){
-  var DEFAULT_LAT = 31.5204, DEFAULT_LNG = 74.3587;
-  var rbMap = null, rbMarker = null;
-
-  function setStatus(msg, color) {
-    var el = document.getElementById('gpsStatusDiv');
-    if (el) { el.textContent = msg; el.style.color = color || '#25a06b'; }
+(function waitForLeaflet(){
+  if(typeof L !== 'undefined'){
+    initRahbarMap();
+  } else {
+    var s = document.createElement('script');
+    s.src = 'https://unpkg.com/leaflet@1.9.4/dist/leaflet.js';
+    s.onload = initRahbarMap;
+    document.head.appendChild(s);
   }
+})();
 
-  function notifyParent(lat, lng) {
-    // Ask parent page to reverse-geocode and fill the textbox
-    try { window.parent.postMessage({ type: 'rahbar_coords', lat: lat, lng: lng }, '*'); }
-    catch(e) {}
-    setStatus('Getting address for: ' + lat.toFixed(5) + ', ' + lng.toFixed(5) + '...', '#f5a623');
-  }
+var rahbarMap = null, rahbarMarker = null;
 
-  function buildMap(lat, lng) {
-    if (rbMap) { rbMap.remove(); rbMap = null; rbMarker = null; }
-    rbMap = L.map('rahbarMapDiv').setView([lat, lng], 15);
-    L.tileLayer('https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png', {
-      attribution: '&copy; OpenStreetMap', maxZoom: 19
-    }).addTo(rbMap);
-    rbMarker = L.marker([lat, lng], { draggable: true })
-      .addTo(rbMap)
-      .bindPopup('Drag pin or click map to adjust')
-      .openPopup();
-    rbMarker.on('dragend', function(e) {
-      var ll = e.target.getLatLng();
-      notifyParent(ll.lat, ll.lng);
-    });
-    rbMap.on('click', function(e) {
-      rbMarker.setLatLng(e.latlng);
-      notifyParent(e.latlng.lat, e.latlng.lng);
-    });
-    setTimeout(function(){ if (rbMap) rbMap.invalidateSize(); }, 350);
-  }
+function setGpsStatus(msg, color){
+  var el = document.getElementById('gps_status_bar');
+  if(el){ el.textContent = msg; el.style.color = color || '#25a06b'; }
+}
 
-  // Listen for commands from parent page
-  window.addEventListener('message', function(ev) {
-    if (!ev.data) return;
-    if (ev.data.type === 'rahbar_move_pin') {
-      var lat = ev.data.lat, lng = ev.data.lng;
-      if (!rbMap) buildMap(lat, lng);
-      else { rbMarker.setLatLng([lat, lng]); rbMap.setView([lat, lng], 17); }
-      setTimeout(function(){ if (rbMap) rbMap.invalidateSize(); }, 350);
-    }
-    if (ev.data.type === 'rahbar_status') {
-      setStatus(ev.data.msg, ev.data.color || '#25a06b');
-    }
-  });
+function initRahbarMap(){
+  if(rahbarMap) return;
+  var mapEl = document.getElementById('rahbar_map');
+  if(!mapEl){ setTimeout(initRahbarMap, 300); return; }
 
-  function tryInit() {
-    if (document.getElementById('rahbarMapDiv') && typeof L !== 'undefined') {
-      buildMap(DEFAULT_LAT, DEFAULT_LNG);
-    } else {
-      setTimeout(tryInit, 200);
-    }
-  }
+  rahbarMap = L.map('rahbar_map').setView([31.5204, 74.3587], 13);
+  L.tileLayer('https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png',{
+    attribution:'&copy; OpenStreetMap',maxZoom:19
+  }).addTo(rahbarMap);
 
-  if (document.readyState === 'loading')
-    document.addEventListener('DOMContentLoaded', tryInit);
-  else
-    setTimeout(tryInit, 100);
-})();
-</script>
-"""
+  rahbarMarker = L.marker([31.5204, 74.3587],{draggable:true})
+    .addTo(rahbarMap)
+    .bindPopup('پن گھسیٹیں یا نقشے پر کلک کریں')
+    .openPopup();
 
-# ─── GPS PARENT-PAGE SCRIPT ───────────────────────────────────
-# Injected into the Gradio page directly (NOT inside an iframe).
-# Has full access to navigator.geolocation and the real page DOM.
-# Wires the GPS gr.Button by elem_id and handles reverse-geocoding.
-GPS_PARENT_SCRIPT = """
-<script>
-(function(){
-  // Reverse-geocode coordinates to a human-readable address
-  function reverseGeocode(lat, lng, cb) {
-    var url = 'https://nominatim.openstreetmap.org/reverse?format=jsonv2'
-            + '&lat=' + lat + '&lon=' + lng + '&zoom=17&addressdetails=1';
-    fetch(url, { headers: { 'Accept-Language': 'en' } })
-      .then(function(r) { return r.json(); })
-      .then(function(d) {
-        var a = d.address || {}, p = [];
-        if (a.road || a.pedestrian)      p.push(a.road || a.pedestrian);
-        if (a.suburb || a.neighbourhood) p.push(a.suburb || a.neighbourhood);
-        if (a.city_district || a.county) p.push(a.city_district || a.county);
-        if (!p.length && d.display_name) p.push(d.display_name.split(',').slice(0,3).join(',').trim());
-        cb(p.join(', ') || ('GPS:' + lat.toFixed(5) + ',' + lng.toFixed(5)));
-      })
-      .catch(function() { cb('GPS:' + lat.toFixed(5) + ',' + lng.toFixed(5)); });
-  }
+  rahbarMarker.on('dragend', function(e){
+    var ll = e.target.getLatLng();
+    updateLocationFromCoords(ll.lat, ll.lng);
+  });
 
-  // Write text into the Gradio location textbox (React/Svelte-safe)
-  function fillLocationTb(text) {
-    // Try elem_id wrapper first
-    var wrap = document.getElementById('rahbar_location_tb');
-    var el   = wrap ? (wrap.querySelector('textarea') || wrap.querySelector('input')) : null;
-    // Fallback: scan all visible inputs for matching placeholder text
-    if (!el) {
-      var all = document.querySelectorAll('textarea, input[type="text"]');
-      for (var i = 0; i < all.length; i++) {
-        var ph = (all[i].placeholder || '').toLowerCase();
-        if (ph.indexOf('gps') > -1 || ph.indexOf('street') > -1 || ph.indexOf('landmark') > -1) {
-          el = all[i]; break;
-        }
-      }
-    }
-    if (!el) { console.warn('Rahbar GPS: location textbox not found'); return; }
-    // Svelte/React-compatible value setter
-    var proto  = el.tagName === 'TEXTAREA' ? HTMLTextAreaElement.prototype : HTMLInputElement.prototype;
-    var setter = Object.getOwnPropertyDescriptor(proto, 'value');
-    if (setter && setter.set) setter.set.call(el, text); else el.value = text;
-    el.dispatchEvent(new Event('input',  { bubbles: true }));
-    el.dispatchEvent(new Event('change', { bubbles: true }));
-  }
+  rahbarMap.on('click', function(e){
+    rahbarMarker.setLatLng(e.latlng);
+    updateLocationFromCoords(e.latlng.lat, e.latlng.lng);
+  });
 
-  // Send a message to all iframe contentWindows (reaches the map widget)
-  function msgIframes(data) {
-    document.querySelectorAll('iframe').forEach(function(f) {
-      try { f.contentWindow.postMessage(data, '*'); } catch(e) {}
-    });
-  }
+  setTimeout(function(){ if(rahbarMap) rahbarMap.invalidateSize(); }, 500);
+}
+
+function moveMapPin(lat, lng){
+  if(!rahbarMap){ setTimeout(function(){ moveMapPin(lat,lng); }, 400); return; }
+  rahbarMarker.setLatLng([lat,lng]);
+  rahbarMap.setView([lat,lng], 17);
+  setTimeout(function(){ rahbarMap.invalidateSize(); }, 300);
+}
 
-  // Handle a resolved lat/lng pair: move map pin + reverse-geocode + fill textbox
-  function handleCoords(lat, lng) {
-    msgIframes({ type: 'rahbar_move_pin', lat: lat, lng: lng });
-    msgIframes({ type: 'rahbar_status', msg: 'Fetching address...', color: '#f5a623' });
-    reverseGeocode(lat, lng, function(addr) {
-      fillLocationTb(addr);
-      msgIframes({ type: 'rahbar_status', msg: '✅ Location filled: ' + addr, color: '#25a06b' });
+function updateLocationFromCoords(lat, lng){
+  setGpsStatus('پتہ حاصل ہو رہا ہے: ' + lat.toFixed(5) + ', ' + lng.toFixed(5) + '...', '#f5a623');
+  fetch('https://nominatim.openstreetmap.org/reverse?format=jsonv2&lat='+lat+'&lon='+lng+'&zoom=17&addressdetails=1',
+    {headers:{'Accept-Language':'ur,en'}})
+    .then(function(r){ return r.json(); })
+    .then(function(d){
+      var a = d.address || {}, p = [];
+      if(a.road || a.pedestrian) p.push(a.road || a.pedestrian);
+      if(a.suburb || a.neighbourhood) p.push(a.suburb || a.neighbourhood);
+      if(a.city_district || a.county) p.push(a.city_district || a.county);
+      if(!p.length && d.display_name) p = d.display_name.split(',').slice(0,3);
+      var addr = p.join(', ').trim() || ('GPS:'+lat.toFixed(5)+','+lng.toFixed(5));
+      fillLocationBox(addr);
+      setGpsStatus('✅ مقام ملا: ' + addr, '#25a06b');
+    })
+    .catch(function(){
+      var addr = 'GPS:'+lat.toFixed(5)+','+lng.toFixed(5);
+      fillLocationBox(addr);
+      setGpsStatus('✅ GPS مقام: ' + addr, '#25a06b');
     });
-  }
+}
 
-  // GPS button click handler
-  function doGPS() {
-    if (!navigator.geolocation) {
-      alert('Geolocation / GPS is not supported by this browser.');
-      return;
+function fillLocationBox(text){
+  // Try multiple selectors for Gradio 4/5/6 compatibility
+  var el = null;
+  var wrap = document.getElementById('rahbar_location_tb');
+  if(wrap) el = wrap.querySelector('textarea') || wrap.querySelector('input');
+  if(!el){
+    var all = document.querySelectorAll('textarea,input[type="text"]');
+    for(var i=0;i<all.length;i++){
+      var ph = (all[i].placeholder||'').toLowerCase();
+      if(ph.indexOf('gps')>-1 || ph.indexOf('address')>-1 || ph.indexOf('street')>-1 || ph.indexOf('landmark')>-1){
+        el = all[i]; break;
+      }
     }
-    msgIframes({ type: 'rahbar_status', msg: 'Requesting GPS — please ALLOW when your browser prompts...', color: '#f5a623' });
-    navigator.geolocation.getCurrentPosition(
-      function(pos) { handleCoords(pos.coords.latitude, pos.coords.longitude); },
-      function(err) {
-        var msgs = {
-          1: '❌ Location access DENIED. Open browser settings, allow this site location access, then retry.',
-          2: '⚠️ Location signal unavailable. Check GPS/WiFi and retry.',
-          3: '⏱️ GPS request timed out. Please try again.'
-        };
-        var msg = msgs[err.code] || ('GPS error: ' + err.message);
-        msgIframes({ type: 'rahbar_status', msg: msg, color: '#e85353' });
-        alert(msg);
-      },
-      { enableHighAccuracy: true, timeout: 15000, maximumAge: 0 }
-    );
   }
-
-  // Listen for map-pin-drag events bubbled up from iframes
-  window.addEventListener('message', function(ev) {
-    if (ev.data && ev.data.type === 'rahbar_coords') {
-      handleCoords(ev.data.lat, ev.data.lng);
-    }
+  if(!el){ console.warn('Rahbar: location box not found'); return; }
+  // React/Svelte compatible value setter
+  var proto = el.tagName==='TEXTAREA' ? HTMLTextAreaElement.prototype : HTMLInputElement.prototype;
+  var setter = Object.getOwnPropertyDescriptor(proto,'value');
+  if(setter && setter.set) setter.set.call(el, text); else el.value = text;
+  ['input','change'].forEach(function(ev){
+    el.dispatchEvent(new Event(ev,{bubbles:true}));
   });
+}
 
-  // Attach click listener to the GPS gr.Button (identified by elem_id wrapper)
-  function attachGpsButton() {
-    var wrap = document.getElementById('rahbar_gps_btn');
-    var btn  = wrap ? wrap.querySelector('button') : null;
-    if (btn) {
-      // Clone to clear any stale Gradio listeners
-      var fresh = btn.cloneNode(true);
-      btn.parentNode.replaceChild(fresh, btn);
-      fresh.addEventListener('click', function(e) {
-        e.preventDefault(); e.stopPropagation(); doGPS();
-      });
-    } else {
-      // Button not yet in DOM — retry
-      setTimeout(attachGpsButton, 400);
+// GPS function — called directly by the GPS button via onclick
+window.rahbarGetGPS = function(){
+  if(!navigator.geolocation){
+    setGpsStatus('❌ آپ کا براؤزر GPS کی سہولت نہیں دیتا۔', '#e85353');
+    return;
+  }
+  setGpsStatus('📡 GPS مانگ رہے ہیں — براؤزر میں "Allow" کریں...', '#f5a623');
+  navigator.geolocation.getCurrentPosition(
+    function(pos){
+      var lat = pos.coords.latitude, lng = pos.coords.longitude;
+      moveMapPin(lat, lng);
+      updateLocationFromCoords(lat, lng);
+    },
+    function(err){
+      var msgs = {
+        1:'❌ مقام تک رسائی مسترد — براؤزر سیٹنگ میں اس سائٹ کو مقام کی اجازت دیں۔',
+        2:'⚠️ GPS سگنل دستیاب نہیں — GPS/WiFi چیک کریں اور دوبارہ کوشش کریں۔',
+        3:'⏱️ GPS وقت ختم — دوبارہ کوشش کریں۔'
+      };
+      setGpsStatus(msgs[err.code]||('GPS خرابی: '+err.message), '#e85353');
+    },
+    {enableHighAccuracy:true, timeout:15000, maximumAge:0}
+  );
+};
+
+// Attach GPS button — works with Gradio 4/5/6
+function attachGpsBtn(){
+  // Method 1: elem_id wrapper
+  var wrap = document.getElementById('rahbar_gps_btn');
+  var btn = wrap ? wrap.querySelector('button') : null;
+  // Method 2: scan all buttons
+  if(!btn){
+    var btns = document.querySelectorAll('button');
+    for(var i=0;i<btns.length;i++){
+      if(btns[i].textContent.indexOf('GPS') > -1){
+        btn = btns[i]; break;
+      }
     }
   }
+  if(btn){
+    // Remove existing listeners and add fresh one
+    var fresh = btn.cloneNode(true);
+    btn.parentNode.replaceChild(fresh, btn);
+    fresh.addEventListener('click', function(e){
+      e.preventDefault();
+      e.stopImmediatePropagation();
+      window.rahbarGetGPS();
+    }, true);
+    fresh.setAttribute('onclick', 'event.preventDefault();event.stopPropagation();window.rahbarGetGPS();return false;');
+    console.log('Rahbar: GPS button attached');
+  } else {
+    setTimeout(attachGpsBtn, 500);
+  }
+}
 
-  if (document.readyState === 'loading')
-    document.addEventListener('DOMContentLoaded', function() { setTimeout(attachGpsButton, 800); });
-  else
-    setTimeout(attachGpsButton, 800);
-})();
+if(document.readyState === 'loading'){
+  document.addEventListener('DOMContentLoaded', function(){ setTimeout(attachGpsBtn, 1000); });
+} else {
+  setTimeout(attachGpsBtn, 1000);
+}
+// Also re-attach on tab changes / Gradio re-renders
+var gpsObs = new MutationObserver(function(){
+  if(!document.getElementById('rahbar_gps_btn')) return;
+  var wrap = document.getElementById('rahbar_gps_btn');
+  var btn = wrap ? wrap.querySelector('button') : null;
+  if(btn && !btn._rahbarGpsAttached){
+    btn._rahbarGpsAttached = true;
+    btn.addEventListener('click', function(e){
+      e.preventDefault(); e.stopImmediatePropagation();
+      window.rahbarGetGPS();
+    }, true);
+  }
+});
+gpsObs.observe(document.body, {childList:true, subtree:true});
 </script>
 """
 
@@ -1166,97 +1218,87 @@ def build_ui():
         with gr.Tabs():
 
             # ══════════════════���═══════════════════════════════
-            # TAB 1 — REPORT ISSUE
+            # TAB 1 — شکایت درج کریں
             # ══════════════════════════════════════════════════
-            with gr.Tab("📝 Report Issue"):
+            with gr.Tab("📝 شکایت درج کریں"):
                 with gr.Row(equal_height=False):
 
-                    # ── LEFT: inputs ──────────────────────────
+                    # ── بائیں: inputs ──────────────────────────
                     with gr.Column(scale=1, min_width=300):
 
-                        gr.HTML('<div class="card-title">Citizen Details</div>')
-                        name_tb  = gr.Textbox(label="Full Name",                placeholder="e.g., Ali Raza",  lines=1)
-                        cnic_tb  = gr.Textbox(label="CNIC (without dashes)",    placeholder="1234567890123",   lines=1)
-                        phone_tb = gr.Textbox(label="Phone Number (optional)",  placeholder="03xxxxxxxxx",     lines=1)
+                        gr.HTML('<div class="card-title">شہری معلومات</div>')
+                        name_tb  = gr.Textbox(label="پورا نام",          placeholder="مثلاً: علی رضا",       lines=1)
+                        cnic_tb  = gr.Textbox(label="شناختی کارڈ نمبر (بغیر ڈیش)",  placeholder="1234567890123",   lines=1)
+                        phone_tb = gr.Textbox(label="فون نمبر (اختیاری)",  placeholder="03xxxxxxxxx",          lines=1)
 
-                        gr.HTML('<div class="card-title" style="margin-top:14px">Issue Photo</div>')
-                        gr.HTML("""<div class="info-box">Upload a clear photo of the civic issue.
-On mobile: tap "Webcam" to use your device camera (front or rear).</div>""")
-                        # sources=["webcam","upload"] enables native camera on mobile/desktop
+                        gr.HTML('<div class="card-title" style="margin-top:14px">مسئلے کی تصویر</div>')
+                        gr.HTML("""<div class="info-box">مسئلے کی واضح تصویر اپلوڈ کریں۔
+موبائل پر: "ویب کیم" کے ذریعے کیمرہ استعمال کر سکتے ہیں۔</div>""")
                         image_input = gr.Image(
                             type="pil",
-                            label="Upload or Capture Photo",
+                            label="تصویر اپلوڈ یا کیپچر کریں",
                             sources=["webcam", "upload"],
                             height=220,
                         )
 
-                        gr.HTML('<div class="card-title" style="margin-top:14px">Complaint Details</div>')
-                        issue_type  = gr.Radio(choices=ISSUE_TYPES, value=ISSUE_TYPES[0], label="Issue Type")
-                        city_dd     = gr.Dropdown(choices=list(CITIES_AREAS.keys()), value="Lahore",       label="City")
-                        area_dd     = gr.Dropdown(choices=CITIES_AREAS["Lahore"],     value="Model Town",  label="Area")
+                        gr.HTML('<div class="card-title" style="margin-top:14px">شکایت کی تفصیل</div>')
+                        issue_type  = gr.Radio(choices=ISSUE_TYPES, value=ISSUE_TYPES[0], label="مسئلے کی نوعیت")
+                        city_dd     = gr.Dropdown(choices=list(CITIES_AREAS.keys()), value="Lahore",       label="شہر")
+                        area_dd     = gr.Dropdown(choices=CITIES_AREAS["Lahore"],     value="Model Town",  label="علاقہ")
 
-                        # ── GPS section ───────────────────────
-                        gr.HTML('<div class="card-title" style="margin-top:14px">GPS Location</div>')
+                        gr.HTML('<div class="card-title" style="margin-top:14px">GPS مقام</div>')
                         gr.HTML("""<div class="warn-box">
-<strong>Click "Get My GPS Location"</strong> and allow location access when prompted.
-Your street address will auto-fill below via OpenStreetMap reverse geocoding.
-You can also click the map or drag the pin to set a custom location.
+<strong>"میرا GPS مقام حاصل کریں"</strong> بٹن دبائیں اور مقام کی اجازت دیں۔
+آپ کا پتہ خود بھر جائے گا۔ نقشے پر کلک کر کے بھی مقام تبدیل کر سکتے ہیں۔
 </div>""")
 
-                        # gr.Button with elem_id — GPS script attaches to this button
-                        # using addEventListener in the PARENT page (not inside an iframe),
-                        # so navigator.geolocation is fully accessible.
+                        # GPS button — elem_id for JS targeting
                         gps_btn = gr.Button(
-                            "📍 Get My GPS Location",
+                            "📍 میرا GPS مقام حاصل کریں",
                             variant="primary",
                             elem_id="rahbar_gps_btn",
                         )
 
-                        # Map widget (rendered inside a sandboxed iframe by Gradio)
+                        # Map — loaded directly in page (not iframe)
                         gr.HTML(MAP_HTML)
 
-                        # Location textbox — elem_id used by GPS script to fill it
                         location_tb = gr.Textbox(
-                            label="Street / Landmark / GPS Address",
-                            placeholder="GPS address auto-fills here — or type manually",
+                            label="گلی / نشانی / GPS پتہ",
+                            placeholder="GPS پتہ یہاں خود بھرے گا — یا خود لکھیں",
                             lines=1,
                             elem_id="rahbar_location_tb",
                         )
 
-                        # GPS parent-page script — injected directly into Gradio page
-                        gr.HTML(GPS_PARENT_SCRIPT)
-
-                        desc_tb     = gr.Textbox(label="Additional Description (optional)", placeholder="Describe the issue...", lines=3)
-                        language_dd = gr.Dropdown(choices=LANGUAGES, value="English", label="Report & Voice Language")
-                        tts_cb      = gr.Checkbox(label="Read report aloud (Text-to-Speech)", value=False)
-                        submit_btn  = gr.Button("Submit Complaint", variant="primary", size="lg")
+                        desc_tb     = gr.Textbox(label="اضافی تفصیل (اختیاری)", placeholder="مسئلے کو بیان کریں...", lines=3)
+                        language_dd = gr.Dropdown(choices=LANGUAGES, value="English", label="رپورٹ اور آواز کی زبان")
+                        tts_cb      = gr.Checkbox(label="رپورٹ اونچی آواز میں پڑھیں (Text-to-Speech)", value=False)
+                        submit_btn  = gr.Button("شکایت جمع کریں", variant="primary", size="lg")
 
-                    # ── RIGHT: outputs ────────────────────────
+                    # ── دائیں: outputs ────────────────────────
                     with gr.Column(scale=2, min_width=320):
 
-                        gr.HTML('<div class="card-title">AI Analysis Results</div>')
-                        annotated_out    = gr.Image(label="AI Detection Result", height=240)
-                        complaint_id_out = gr.Textbox(label="Complaint ID", interactive=False)
+                        gr.HTML('<div class="card-title">AI تجزیہ کے نتائج</div>')
+                        annotated_out    = gr.Image(label="AI ڈیٹیکشن نتیجہ", height=240)
+                        complaint_id_out = gr.Textbox(label="شکایت نمبر", interactive=False)
 
-                        gr.HTML('<div class="card-title" style="margin-top:14px">Official Complaint Report</div>')
+                        gr.HTML('<div class="card-title" style="margin-top:14px">سرکاری شکایت رپورٹ</div>')
                         report_out = gr.Textbox(
-                            label="Full Complaint Report (Ready for Submission)",
+                            label="مکمل شکایت رپورٹ (جمع کرانے کے لیے تیار)",
                             lines=22, interactive=False,
-                            placeholder="Report will appear here after submission..."
+                            placeholder="جمع کرانے کے بعد رپورٹ یہاں آئے گی..."
                         )
                         wa_out         = gr.Markdown()
-                        report_tts_out = gr.Audio(label="Report Audio (selected language)", autoplay=False)
+                        report_tts_out = gr.Audio(label="رپورٹ آڈیو (منتخب زبان میں)", autoplay=False)
 
-                        gr.HTML('<div class="card-title" style="margin-top:16px">Legal Advice</div>')
-                        gr.HTML('<div class="info-box">AI-generated legal guidance based on Pakistani civic law, in your selected language.</div>')
+                        gr.HTML('<div class="card-title" style="margin-top:16px">قانونی مشورہ</div>')
+                        gr.HTML('<div class="info-box">پاکستانی شہری قانون کی بنیاد پر AI کا قانونی رہنمائی۔</div>')
                         legal_advice_out = gr.Textbox(
-                            label="Legal Rights & Action Steps",
+                            label="قانونی حقوق اور اقدامات",
                             lines=12, interactive=False,
-                            placeholder="Legal advice will appear here..."
+                            placeholder="قانونی مشورہ یہاں آئے گا..."
                         )
-                        advice_tts_out = gr.Audio(label="Legal Advice Audio (selected language)", autoplay=False)
+                        advice_tts_out = gr.Audio(label="قانونی مشورہ آڈیو (منتخب زبان میں)", autoplay=False)
 
-                # ── Event wiring ──────────────────────────────
                 city_dd.change(fn=update_areas, inputs=[city_dd], outputs=[area_dd])
                 submit_btn.click(
                     fn=make_report,
@@ -1267,43 +1309,78 @@ You can also click the map or drag the pin to set a custom location.
                 )
 
             # ══════════════════════════════════════════════════
-            # TAB 2 — LAW REFERENCE + RAG CHATBOT
+            # TAB 2 — قانون اور چیٹ بوٹ
             # ══════════════════════════════════════════════════
-            with gr.Tab("⚖️ Law & Chatbot"):
+            with gr.Tab("⚖️ قانون اور چیٹ بوٹ"):
 
-                gr.HTML('<div class="card-title">Pakistani Civic Laws Database</div>')
+                gr.HTML('<div class="card-title">پاکستانی شہری قوانین کا ڈیٹا بیس</div>')
                 with gr.Row():
-                    law_issue_dd = gr.Dropdown(choices=ISSUE_TYPES, value=ISSUE_TYPES[0], label="Select Issue", scale=1)
-                    law_lang_dd  = gr.Dropdown(choices=LANGUAGES,   value="English",       label="Language",    scale=1)
+                    law_issue_dd = gr.Dropdown(choices=ISSUE_TYPES, value=ISSUE_TYPES[0], label="مسئلہ منتخب کریں", scale=1)
+                    law_lang_dd  = gr.Dropdown(choices=LANGUAGES,   value="English",       label="زبان",            scale=1)
                 law_out = gr.Markdown()
-                gr.Button("Show Law Details", variant="primary").click(
+                gr.Button("قانونی تفصیل دکھائیں", variant="primary").click(
                     fn=law_info, inputs=[law_issue_dd, law_lang_dd], outputs=[law_out]
                 )
                 gr.HTML("""<div class="info-box">
-<strong>Quick Helplines:</strong>&nbsp;
-Garbage: <span class="hotline-pill">1139</span>&nbsp;|&nbsp;
-Roads/NHA: <span class="hotline-pill">051-9032800</span>&nbsp;|&nbsp;
-WASA Lahore: <span class="hotline-pill">042-99200300</span>&nbsp;|&nbsp;
-CM Portal: <span class="hotline-pill">0800-02345</span>&nbsp;|&nbsp;
-Fed. Ombudsman: <span class="hotline-pill">051-9204551</span>
+<strong>فوری ہیلپ لائنیں:</strong>&nbsp;
+کچرا: <span class="hotline-pill">1139</span>&nbsp;|&nbsp;
+سڑکیں/NHA: <span class="hotline-pill">051-9032800</span>&nbsp;|&nbsp;
+WASA لاہور: <span class="hotline-pill">042-99200300</span>&nbsp;|&nbsp;
+CM پورٹل: <span class="hotline-pill">0800-02345</span>&nbsp;|&nbsp;
+وفاقی محتسب: <span class="hotline-pill">051-9204551</span>
 </div>""")
 
-                gr.HTML('<div class="card-title" style="margin-top:24px">🤖 RAG Legal Chatbot</div>')
+                gr.HTML('<div class="card-title" style="margin-top:24px">🤖 RAG قانونی چیٹ بوٹ</div>')
                 gr.HTML("""<div class="info-box">
-This chatbot uses <strong>Retrieval-Augmented Generation (RAG)</strong> with:
-Road Issues Detection Dataset &nbsp;|&nbsp; Urban Issues Dataset &nbsp;|&nbsp; Consumer Complaints Dataset.<br>
-Ask about water supply, WASA, pipe leakage, garbage, road potholes, or your legal rights as a Pakistani citizen.
+یہ چیٹ بوٹ <strong>RAG (Retrieval-Augmented Generation)</strong> استعمال کرتا ہے۔
+پانی، WASA، پائپ لیکیج، کچرا، سڑک کے گڑھے، یا پاکستانی شہری قوانین کے بارے میں پوچھیں۔
+اب آپ آواز سے بھی سوال پوچھ سکتے ہیں اور جواب سن سکتے ہیں۔
 </div>""")
-                chat_lang_dd = gr.Dropdown(choices=LANGUAGES, value="English", label="Response Language")
-                chatbot      = gr.Chatbot(label="Rahbar RAG Legal Assistant", height=500, value=[])
+
+                chat_lang_dd = gr.Dropdown(choices=LANGUAGES, value="English", label="جواب کی زبان")
+                chatbot      = gr.Chatbot(label="رہبر قانونی معاون", height=420, value=[])
+
+                # ── Chat input row ────────────────────────────
                 with gr.Row():
                     chat_input    = gr.Textbox(
-                        label="Your Question",
-                        placeholder="e.g., WASA did not fix the pipe for 3 days. What are my legal rights?",
+                        label="آپ کا سوال",
+                        placeholder="مثلاً: WASA نے 3 دن میں پائپ ٹھیک نہیں کیا — میرے کیا حقوق ہیں؟",
                         lines=2, scale=4
                     )
-                    chat_send_btn = gr.Button("Send", variant="primary", scale=1)
+                    chat_send_btn = gr.Button("بھیجیں", variant="primary", scale=1)
+
+                # ── Voice input for chatbot ───────────────────
+                gr.HTML('<div class="card-title" style="margin-top:12px">🎤 آواز سے سوال پوچھیں</div>')
+                with gr.Row():
+                    chat_audio_in = gr.Audio(
+                        type="filepath",
+                        label="آواز ریکارڈ کریں",
+                        sources=["microphone", "upload"],
+                        scale=3,
+                    )
+                    chat_voice_btn = gr.Button("🎤 آواز سے سوال بھیجیں", variant="secondary", scale=1)
+
+                # ── Chatbot TTS output ────────────────────────
+                gr.HTML('<div class="card-title" style="margin-top:12px">🔊 آواز میں جواب سنیں</div>')
+                with gr.Row():
+                    chat_tts_out = gr.Audio(label="چیٹ بوٹ کا آواز جواب", autoplay=False, scale=3)
+                    chat_tts_btn = gr.Button("🔊 آخری جواب سنیں", variant="secondary", scale=1)
+
+                # ── Examples ──────────────────────────────────
+                gr.Examples(
+                    examples=[
+                        ["WASA نے 3 دن میں لیکیج ٹھیک نہیں کی — میرے کیا حقوق ہیں؟"],
+                        ["میرے علاقے کا پانی آلودہ ہے — کہاں شکایت کروں؟"],
+                        ["ایک ہفتے سے کچرا نہیں اٹھا — کون سا قانون لاگو ہوتا ہے؟"],
+                        ["ادارہ شکایت نظرانداز کرے تو کیا کریں؟"],
+                        ["گڑھے سے گاڑی کو نقصان ہوا — کیا معاوضہ مل سکتا ہے؟"],
+                        ["Pakistan Citizen Portal پر شکایت کیسے کریں؟"],
+                    ],
+                    inputs=chat_input,
+                    label="یہ سوال آزمائیں"
+                )
 
+                # ── Event wiring for chatbot ──────────────────
                 chat_send_btn.click(
                     fn=legal_chatbot_rag,
                     inputs=[chat_input, chatbot, chat_lang_dd],
@@ -1314,75 +1391,83 @@ Ask about water supply, WASA, pipe leakage, garbage, road potholes, or your lega
                     inputs=[chat_input, chatbot, chat_lang_dd],
                     outputs=[chatbot, chat_input]
                 )
-                gr.Examples(
-                    examples=[
-                        ["WASA did not fix the leakage for 3 days. What are my rights?"],
-                        ["The water in my area is contaminated. Where do I complain?"],
-                        ["Garbage has not been collected for a week. What law applies?"],
-                        ["How do I escalate if the authority ignores my complaint?"],
-                        ["A pothole damaged my car. Can I claim compensation?"],
-                        ["How do I file a complaint on the Pakistan Citizen Portal?"],
-                    ],
-                    inputs=chat_input,
-                    label="Try These Questions"
+
+                def voice_then_send(audio_file, history, language):
+                    """Transcribe voice, send to chatbot, return updated history"""
+                    if audio_file is None:
+                        return history, ""
+                    transcribed = stt(audio_file)
+                    if not transcribed or transcribed.startswith("آڈیو") or transcribed.startswith("ٹرانسکرپشن"):
+                        return history, transcribed
+                    return legal_chatbot_rag(transcribed, history, language)[0], ""
+
+                chat_voice_btn.click(
+                    fn=voice_then_send,
+                    inputs=[chat_audio_in, chatbot, chat_lang_dd],
+                    outputs=[chatbot, chat_input]
+                )
+                chat_tts_btn.click(
+                    fn=chatbot_tts_output,
+                    inputs=[chatbot, chat_lang_dd],
+                    outputs=[chat_tts_out]
                 )
 
             # ══════════════════════════════════════════════════
-            # TAB 3 — VOICE INPUT / STT
+            # TAB 3 — آواز کا ان پٹ / STT
             # ══════════════════════════════════════════════════
-            with gr.Tab("🎤 Voice Input"):
+            with gr.Tab("🎤 آواز کا ان پٹ"):
 
-                gr.HTML('<div class="card-title">Speech-to-Text (STT)</div>')
+                gr.HTML('<div class="card-title">تقریر سے متن (Speech-to-Text)</div>')
                 gr.HTML("""<div class="info-box">
-Record your complaint or description using the microphone below.
-Transcription uses <strong>Groq Whisper large-v3</strong> (if GROQ_API_KEY is set)
-or Google Speech Recognition (fallback).<br>
-Supports Urdu, English, Punjabi, and Sindhi. Please allow microphone access when prompted.
+اپنی شکایت یا تفصیل نیچے مائیکروفون سے ریکارڈ کریں۔
+ٹرانسکرپشن <strong>Groq Whisper large-v3</strong> (اگر GROQ_API_KEY سیٹ ہو)
+یا Google Speech Recognition (متبادل) سے ہوگی۔<br>
+اردو، انگریزی، پنجابی اور سندھی میں بولیں۔
 </div>""")
                 gr.HTML("""<div class="warn-box">
-<strong>Tip:</strong> Speak clearly with minimal background noise.
-After transcription, copy the text and paste it into the Report Issue tab.
+<strong>تجویز:</strong> واضح آواز میں بولیں، شور کم ہو۔
+ٹرانسکرپشن کے بعد متن کاپی کریں اور "شکایت درج کریں" ٹیب میں پیسٹ کریں۔
 </div>""")
                 audio_in = gr.Audio(
                     type="filepath",
-                    label="Record or Upload Audio",
+                    label="آڈیو ریکارڈ یا اپلوڈ کریں",
                     sources=["microphone", "upload"],
                 )
-                stt_btn = gr.Button("Transcribe Audio", variant="primary")
+                stt_btn = gr.Button("آڈیو کو متن میں تبدیل کریں", variant="primary")
                 stt_out = gr.Textbox(
-                    label="Transcription (editable — copy to Report Issue tab)",
+                    label="ٹرانسکرپشن (قابل ترمیم — شکایت ٹیب میں کاپی کریں)",
                     lines=6, interactive=True,
-                    placeholder="Transcribed text will appear here..."
+                    placeholder="ٹرانسکرپٹ شدہ متن یہاں آئے گا..."
                 )
                 stt_btn.click(fn=stt, inputs=[audio_in], outputs=[stt_out])
 
-                gr.HTML('<div class="card-title" style="margin-top:24px">Text-to-Speech Test</div>')
-                gr.HTML('<div class="info-box">Test the TTS engine in any supported language.</div>')
+                gr.HTML('<div class="card-title" style="margin-top:24px">متن سے آواز (TTS) ٹیسٹ</div>')
+                gr.HTML('<div class="info-box">کسی بھی زبان میں آواز آزمائیں۔</div>')
                 with gr.Row():
-                    tts_text_in = gr.Textbox(label="Enter text to hear", placeholder="Type something...", scale=3)
-                    tts_lang_in = gr.Dropdown(choices=LANGUAGES, value="English", label="Language", scale=1)
-                tts_test_btn = gr.Button("▶ Play", variant="secondary")
-                tts_test_out = gr.Audio(label="Audio Output", autoplay=True)
+                    tts_text_in = gr.Textbox(label="سننے کے لیے متن درج کریں", placeholder="کچھ لکھیں...", scale=3)
+                    tts_lang_in = gr.Dropdown(choices=LANGUAGES, value="English", label="زبان", scale=1)
+                tts_test_btn = gr.Button("▶ چلائیں", variant="secondary")
+                tts_test_out = gr.Audio(label="آڈیو آؤٹ پٹ", autoplay=True)
                 tts_test_btn.click(fn=make_tts, inputs=[tts_text_in, tts_lang_in], outputs=[tts_test_out])
 
             # ══════════════════════════════════════════════════
-            # TAB 4 — ADMIN DASHBOARD
+            # TAB 4 — ایڈمن ڈیش بورڈ
             # ══════════════════════════════════════════════════
-            with gr.Tab("📊 Admin Dashboard"):
+            with gr.Tab("📊 ایڈمن ڈیش بورڈ"):
 
-                gr.HTML('<div class="card-title">Complaint Statistics</div>')
-                refresh_btn = gr.Button("Refresh Statistics", variant="primary")
+                gr.HTML('<div class="card-title">شکایات کے اعداد و شمار</div>')
+                refresh_btn = gr.Button("اعداد و شمار تازہ کریں", variant="primary")
                 with gr.Row():
                     stats_out = gr.Markdown()
                     log_out   = gr.Markdown()
                 refresh_btn.click(fn=get_admin_stats, outputs=[stats_out, log_out])
 
                 gr.HTML("""<div class="info-box" style="margin-top:16px">
-<strong>RAG Knowledge Base:</strong>
+<strong>RAG نالج بیس:</strong>
 Road Issues Detection Dataset &nbsp;|&nbsp; Urban Issues Dataset &nbsp;|&nbsp; Consumer Complaints Dataset<br>
-<strong>AI Models:</strong>
+<strong>AI ماڈلز:</strong>
 Gemini 2.0 Flash &nbsp;|&nbsp; YOLOv8 &nbsp;|&nbsp; Llama 3.3 70B &nbsp;|&nbsp; Whisper large-v3 (Groq)<br>
-<strong>Vector Search:</strong> TF-IDF Cosine Similarity (char n-gram, multilingual, no download required)
+<strong>ویکٹر سرچ:</strong> TF-IDF Cosine Similarity (char n-gram, کثیر زبان)
 </div>""")
 
     return demo
@@ -1390,8 +1475,8 @@ Gemini 2.0 Flash &nbsp;|&nbsp; YOLOv8 &nbsp;|&nbsp; Llama 3.3 70B &nbsp;|&nbsp;
 
 # ─── LAUNCH ───────────────────────────────────────────────────
 if __name__ == "__main__":
-    print("Starting Rahbar v4.0 ...")
-    print("RAG Engine:", "Ready" if rag_engine._initialized else "Initializing...")
+    print("Rahbar v5.0 شروع ہو رہا ہے...")
+    print("RAG Engine:", "تیار" if rag_engine._initialized else "شروع ہو رہا ہے...")
     demo = build_ui()
     demo.launch(
         server_name="0.0.0.0",