Update app.py

app.py

@@ -5,7 +5,7 @@ from dataclasses import dataclass
 
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM  
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # ---------------- Language detection ----------------
 ARABIC_RE = re.compile(r"[\u0600-\u06FF]")
@@ -141,6 +141,57 @@ def diversify_if_similar(prev: str, current: str, user_msg: str, lang: str) -> s
         return alt if alt else current
     return current
 
+# ---------------- Prebaked Urdu FAQs (exact templates you provided) ----------------
+# ہم regex-based ہلکا میچ استعمال کر رہے ہیں تاکہ اردو عبارت کے عام انداز پکڑ سکیں
+PREBAKED_UR: List[Dict[str,str]] = [
+    {
+        "q_re": r"(سیلاب).*(گھر).*(ہوں|ہو)",  # سیلاب آ گیا! میں گھر میں ہوں
+        "answer": "🚨 فوری اقدامات:\n- فوراً اونچی جگہ (2+ فلور) کی طرف ہٹیں۔\n- گہرے گھاٹ، ندیوں کے کنارے، یا چھوٹے دریاؤں کے قریب نہ رہیں۔\n- اگر گاڑی میں ہیں، تو فوراً گاڑی چھوڑ کر ہٹیں — گاڑی بہہ سکتی ہے۔\n\n📌 اگلے اقدامات:\n- 1122 یا 1500 پر فون کریں۔\n- اپنے فون کو ہائی ہولڈ رکھیں — پانی میں بھیگنے سے بچائیں ۔\n\n💡 کلیدی نکات:\n- پانی میں ہاتھ پاؤں نہ ڈالیں — گہرائی یا تیز دھارا کا خطرہ ہے۔\n- سیلاب کے بعد گھر میں داخل نہ ہوں جب تک کہ پانی نہ روکا گیا ہو۔"
+    },
+    {
+        "q_re": r"(زلزلہ).*(آ).*(رہا|رہی|گیا|گئی)|(جھٹک|ہل رہا)",
+        "answer": "🚨 فوری اقدامات:\n- 'ڈاون ڈاؤن' (Drop, Cover, Hold On):\n   - گریں (Drop)\n   - ٹیبل یا مضبوط دیوار کے نیچے چھپیں (Cover)\n   - سر کو ہاتھوں سے چھپائیں (Hold On)\n- کھڑے ہو کر دروازے یا کھڑکیوں کی طرف نہ بھاگیں — شیشے ٹوٹ سکتے ہیں۔\n\n📌 اگلے اقدامات:\n- زلزلہ ختم ہونے کے بعد فوراً گھر سے باہر نکلیں — گری سے گری ہوئی عمارت گر سکتی ہے۔\n- بجلی، گیس، پانی کے پائپ بند کریں (اگر گیس کی بو آ رہی ہو)۔\n\n💡 کلیدی نکات:\n- زلزلہ کے دوران کسی چیز کو ہلائیں یا چھوڑیں — ہلاکت خیز ہو سکتا ہے۔\n- گاڑی میں ہوں تو گاڑی کے اندر بیٹھیں — باہر نہ نکلیں۔"
+    },
+    {
+        "q_re": r"(آگ|جل).*(لگ|گئی|گیا)",
+        "answer": "🚨 فوری اقدامات:\n- فوراً گھر سے باہر نکلیں — دھواں سانس کے لیے خطرہ ہے۔\n- ہاتھ سے چہرہ ڈھانپیں اور زمین پر چلتے رہیں (دھواں اوپر ہوتا ہے)۔\n- کوئی بجلی، فون، یا شمع نہ چلائیں — چمک آتشباری کر سکتی ہے۔\n\n📌 اگلے اقدامات:\n- 1122 پر فون کریں — آگ کے بارے میں مطلع کریں۔\n- رشتہ داروں کو مطلع کریں کہ آپ کہاں ہیں۔\n\n💡 کلیدی نکات:\n- زندگی کو بچانا آگ کو بچانے سے زیادہ اہم ہے۔\n- آگ کے بعد گھر میں داخل نہ ہوں جب تک کہ ٹیم کا اجازت نہ ہو۔"
+    },
+    {
+        "q_re": r"(گیس).*(لیک|بو|رس|smell)",
+        "answer": "🚨 فوری اقدامات:\n- گیس کا سوئچ بند کریں (اسٹو کے بالکل نیچے)۔\n- کوئی بجلی، فون، یا شمع نہ چلائیں — چمک آتشباری کر سکتی ہے۔\n- فوراً گھر سے باہر نکلیں — گیس ہلکی سی بھی ہو سکتی ہے۔\n\n📌 اگلے اقدامات:\n- 1122 یا 1500 پر فون کریں — گیس کمپنی کو مطلع کریں۔\n- گھر میں داخل نہ ہوں جب تک کہ گیس چیک نہ ہو جائے۔\n\n💡 کلیدی نکات:\n- گیس لیک کا خطرہ ہر وقت موجود ہے — ہر ماہ چیک کروائیں۔\n- گیس ڈیٹیکٹر لگائیں — اگر ہے، تو اسے فوراً چالو کریں۔"
+    },
+    {
+        "q_re": r"(کلاؤڈ\s*برسٹ|cloud\s*burst)",
+        "answer": "🚨 فوری اقدامات:\n- فوراً اونچی جگہ (پہاڑی، چوٹی) کی طرف ہٹیں۔\n- گہرے گھاٹ، ندیوں کے کنارے، یا چھوٹے دریاؤں کے قریب نہ رہیں۔\n- اگر گاڑی میں ہیں، تو فوراً گاڑی چھوڑ کر ہٹیں — گاڑی بہہ سکتی ہے۔\n\n📌 اگلے اقدامات:\n- 1122 پر فون کریں — ہیلپ مانگیں۔\n- اپنے فون کو ہائی ہولڈ رکھیں — پانی میں نہ ڈالیں۔\n\n💡 کلیدی نکات:\n- کلاؤڈ برسٹ کے بعد سیلاب فوری ہوتا ہے — اس لیے وقت ضائع نہ کریں۔\n- اگر زمین ہل رہی ہو، تو یہ سیلاب کا اشارہ ہے — فوراً ہٹیں۔"
+    },
+    {
+        "q_re": r"(بجلی).*(گر|گری|گر گئی|گرگئی)|lightning|electr",
+        "answer": "🚨 فوری اقدامات:\n- فوراً بجلی کے سوئچ بند کریں (اگر ممکن ہو)۔\n- کوئی بجلی کا آلہ نہ چلائیں — شرارت یا چوٹ لگ سکتی ہے۔\n- ہاتھ پاؤں نہ ڈالیں بلکہ ہلکی روشنی (ٹارچ) استعمال کریں۔\n\n📌 اگلے اقدامات:\n- 1122 پر فون کریں — بجلی کمپنی کو مطلع کریں۔\n- ایمرجنسی ٹارچ یا فون کی روشنی استعمال کریں۔\n\n💡 کلیدی نکات:\n- بجلی گرنے کے بعد کسی بھی الیکٹریکل آلے کو چلانے سے گریز کریں۔\n- ہر 6 ماہ بجلی کے سوئچ اور کیبلز چیک کروائیں۔"
+    },
+    {
+        "q_re": r"(^|\s)SOS(\s|!|۔)|مدد\s*کرو",
+        "answer": "🚨 SOS فائر ہو گیا!\n- فوراً 1122 پر فون کریں — ہیلپ ڈیسک آپ کو مدد کرے گی۔\n- اگر فون کنکشن ہو، تو میں فوراً 1122 پر فون کرتا ہوں...\n- اپنے آپ کو ٹھیک رکھیں — آپ کے ساتھ ہے۔\n\n💡 کلیدی نکات:\n- SOS کا وقت ہے — فوراً 1122 پر فون کریں۔\n- ہر ہفتے ایمرجنسی بیگ چیک کریں: پانی، فوڈ، ٹارچ، فون چارجر، فارمیسی۔"
+    },
+    {
+        "q_re": r"(سیلاب).*(بعد|ختم|اتر)",
+        "answer": "🚨 ایمرجنسی کے بعد:\n- گھر میں داخل نہ ہوں جب تک کہ پانی نہ روکا گیا ہو۔\n- گیس، بجلی، پانی کے پائپ چیک کروائیں — خطرہ ہو سکتا ہے۔\n- اپنے فون کو چارج کریں — ہیلپ ڈیسک سے رابطہ کریں۔\n\n📌 اگلے اقدامات:\n- 1122 پر فون کریں — اپنی حالت بتائیں۔\n- رشتہ داروں کو مطلع کریں کہ آپ کہاں ہیں۔\n\n💡 کلیدی نکات:\n- پانی کے بعد گرد و غبار، کیمیکل، یا بیکٹیریا ہو سکتے ہیں — کپڑے بدلیں۔\n- ایمرجنسی بیگ میں پانی، فوڈ، فارمیسی، ٹارچ وغیرہ رکھیں"
+    },
+]
+
+def match_prebaked_ur(user_msg: str) -> str:
+    """اگر یوزر میسج اردو میں ہو اور کسی پری باکڈ ٹیمپلیٹ سے میچ کرے تو وہی جواب لوٹائیں۔"""
+    if detect_lang(user_msg) != "ur":
+        return ""
+    text = user_msg.strip()
+    for item in PREBAKED_UR:
+        pattern = item.get("q_re")
+        try:
+            if re.search(pattern, text, flags=re.IGNORECASE):
+                return item.get("answer","").strip()
+        except re.error:
+            continue
+    return ""
+
 # ---------------- Base checklist + Go-Bag ----------------
 def base_checklist_and_bag(user_msg: str, lang: str) -> Tuple[str, List[str]]:
     low = user_msg.lower()
@@ -150,11 +201,11 @@ def base_checklist_and_bag(user_msg: str, lang: str) -> Tuple[str, List[str]]:
         checklist = ("✅ ابھی یہ کریں:\n"
                      "- جھک جائیں/Stay low — دھواں اوپر جاتا ہے\n"
                      "- ناک/منہ پر گیلا کپڑا رکھیں\n"
-                     "- اگر ممکن ہو تو بجلی/گیس مین بند کریں\n"
+                     "- اگر محفوظ ہو تو بجلی/گیس مین بند کریں\n"
                      "- گرم دروازہ مت کھولیں؛ دوسرا راستہ لیں\n"
                      "- سیڑھیاں استعمال کریں — لفٹ نہیں\n"
                      "- بچوں/بزرگوں کو ساتھ رکھیں\n"
-                     "🕊️ گھبرائیں نہیں — الله آپ کے ساتھ ہے۔"
+                     "🕊️ گھبرائیں نہیں — میں آپ کے ساتھ ہوں۔"
                      ) if lang=="ur" else (
                      "✅ Do this now:\n"
                      "- Stay low — smoke rises\n"
@@ -173,9 +224,9 @@ def base_checklist_and_bag(user_msg: str, lang: str) -> Tuple[str, List[str]]:
         checklist = ("✅ ابھی یہ کریں:\n"
                      "- Drop, Cover, Hold — جھکیں، ڈھکیں، پکڑیں\n"
                      "- کھڑکیوں/بھاری اشیاء سے دور رہیں\n"
-                     "- جھٹکے رکنے پر سیڑھیوں سے باہر نکلیں (لفٹ استعمال نہ کریں)\n"
+                     "- جھٹکے رکنے پر سیڑھیوں سے باہر نکلیں (لفٹ نہیں)\n"
                      "- کھلی جگہ میں رکیں\n"
-                     "🕊️ حوصلہ رکھیں — الله کریم خیر کا معاملہ فرمائیں گے۔"
+                     "🕊️ حوصلہ رکھیں — میں آپ کے ساتھ ہوں۔"
                      ) if lang=="ur" else (
                      "✅ Do this now:\n"
                      "- Drop, Cover, Hold On\n"
@@ -190,11 +241,11 @@ def base_checklist_and_bag(user_msg: str, lang: str) -> Tuple[str, List[str]]:
     elif any(k in low for k in ["سیلاب","flood","overflow","water entering","طوفان","storm","آندھی"]):
         evac = True
         checklist = ("✅ ابھی یہ کریں:\n"
-                     "- اگر ممکن  ہو تو بجلی مین سوئچ بند کریں\n"
-                     "- اونچی جگہ چھت پر چلے جائیں \n"
+                     "- اگر محفوظ ہو تو بجلی مین سوئچ بند کریں\n"
+                     "- اونچی جگہ/چھت پر منتقل ہوں\n"
                      "- پانی میں گاڑی نہ چلائیں\n"
                      "- بچوں/بزرگوں کو پہلے نکالیں\n"
-                     "🕊️ گھبرائیں نہیں — حوصلہ کریں آپ صحیح سلامت نکل سکتے ہیں۔"
+                     "🕊️ گھبرائیں نہیں — آپ محفوظ نکل سکتے ہیں۔"
                      ) if lang=="ur" else (
                      "✅ Do this now:\n"
                      "- Turn off main power if safe\n"
@@ -209,10 +260,10 @@ def base_checklist_and_bag(user_msg: str, lang: str) -> Tuple[str, List[str]]:
     elif any(k in low for k in ["gas","گیس","leak","smell gas","بو"]):
         checklist = ("✅ ابھی یہ کریں:\n"
                      "- کوئی چنگاری/سوئچ/لائٹر استعمال نہ کریں\n"
-                     "- کھڑکیاں/دروازے کھول کر ہوا کے گزرنے کا راستہ بنائیں \n"
+                     "- کھڑکیاں/دروازے کھول کر ہوا دار کریں\n"
                      "- گیس مین والو بند کریں\n"
                      "- محفوظ فاصلے پر چلے جائیں\n"
-                     "🕊️ پرسکون رہیں — اور الله سے سلامتی کی دعا کریں۔"
+                     "🕊️ پرسکون رہیں — میں رہنمائی کر رہا/رہی ہوں۔"
                      ) if lang=="ur" else (
                      "✅ Do this now:\n"
                      "- No sparks (no switches/lighters/flash)\n"
@@ -226,7 +277,7 @@ def base_checklist_and_bag(user_msg: str, lang: str) -> Tuple[str, List[str]]:
                ["Mask/wet cloth","IDs","Phone+power bank","Water","Medicines"])
     elif any(k in low for k in ["injury","زخمی","bleeding","cut","fracture","burn"]):
         checklist = ("✅ ابھی یہ کریں:\n"
-                     "- خون ہو تو مضبوطی سے دبائیں (صاف کپڑا)\n"
+                     "- خون ہو تو مضبوط دباؤ دیں (صاف کپڑا)\n"
                      "- زخم صاف پانی سے دھوئیں\n"
                      "- گہرا زخم/زیادہ خون ہو تو فوری مدد لیں\n"
                      "- گردن/کمر چوٹ میں مریض کو نہ ہلائیں\n"
@@ -303,6 +354,15 @@ def chat_fn(user_msg: str, chat_messages: List[Dict[str,str]], sim_state: dict):
             {"role":"assistant","content":msg}
         ], sim_state
 
+    # ✅ NEW: Prebaked Urdu templates first
+    pre_ur = match_prebaked_ur(user_msg)
+    if pre_ur:
+        final = pre_ur + "\n\n" + helplines_block("ur")
+        return chat_messages + [
+            {"role":"user","content":user_msg},
+            {"role":"assistant","content":final}
+        ], sim_state
+
     # Base checklist + Go-Bag
     base, _ = base_checklist_and_bag(user_msg, lang)