Update app.py

app.py

@@ -140,7 +140,6 @@ def module_response(user_msg: str, lang: str) -> str:
     calm_en = "Don't worry — take a deep breath; I'm with you."
 
     if lang == "ur":
-        # Fire
         if has_kw(user_msg, "fire"):
             bullets += [
                 "فوراً جھک جائیں، دھواں اوپر جاتا ہے۔",
@@ -150,7 +149,6 @@ def module_response(user_msg: str, lang: str) -> str:
                 "سیڑھیاں استعمال کریں؛ لفٹ ہرگز نہیں۔",
                 "بچوں/بزرگوں کو ساتھ رکھیں، حساب رکھیں۔",
             ]
-        # Gas Leak
         elif has_kw(user_msg, "gas"):
             bullets += [
                 "کسی بھی قسم کی چنگاری نہ بنائیں (سوئچ/لائٹر/فون فلیش نہیں).",
@@ -158,7 +156,6 @@ def module_response(user_msg: str, lang: str) -> str:
                 "گیس مین والو بند کریں۔",
                 "عمارت خالی کریں، محفوظ فاصلے پر جائیں۔",
             ]
-        # Earthquake
         elif has_kw(user_msg, "earthquake"):
             bullets += [
                 "جھکیں، ڈھکیں، اور پکڑیں (Drop, Cover, Hold).",
@@ -166,7 +163,6 @@ def module_response(user_msg: str, lang: str) -> str:
                 "زلزلہ رکتے ہی عمارت کا محفوظ راستہ دیکھ کر باہر جائیں۔",
                 "سیڑھیاں استعمال کریں؛ لفٹ نہیں۔",
             ]
-        # Flood / Storm
         elif has_kw(user_msg, "flood") or has_kw(user_msg, "storm"):
             bullets += [
                 "بجلی مین سوئچ بند کریں (اگر محفوظ ہو تو).",
@@ -174,35 +170,32 @@ def module_response(user_msg: str, lang: str) -> str:
                 "گاڑی کو پانی سے نہ گزاریں؛ 12 انچ پانی بھی خطرناک ہے۔",
                 "گو-بیگ، شناختی دستاویزات، دوائیں ساتھ رکھیں۔",
             ]
-        # Injury / bleeding
         elif has_kw(user_msg, "injury"):
             bullets += [
-                "خون بہہ رہا ہو تو دباؤ ڈال کر روکیں (clean cloth).",
+                "خون بہہ رہا ہو تو دباؤ ڈال کر روکیں (صاف کپڑا).",
                 "زخم کو صاف پانی سے دھوئیں؛ گہرا ہو تو فوری مدد لیں۔",
                 "سر/گردن/کمر چوٹ میں مریض کو نہ ہلائیں (spinal risk).",
             ]
-        # Explosion / conflict
         elif has_kw(user_msg, "explosion") or has_kw(user_msg, "conflict"):
             bullets += [
                 "زمین پر لیٹ جائیں، کور لیں، کھڑکیوں سے دور رہیں۔",
                 "فائرنگ کی سمت کے مخالف کور ڈھونڈیں (دیوار/پکی رکاوٹ).",
                 "فوراً محفوظ راستہ دیکھ کر نکلیں — رش اور شور سے بچیں۔",
             ]
-        # Trapped / child crying
         elif has_kw(user_msg, "trapped") or has_kw(user_msg, "child"):
             bullets += [
-                "خاموش اور آہستہ سانس — توانائی بچائیں۔",
-                "زور سے کھانسیں/آواز دیں صرف جب مدد قریب لگے۔",
-                "ہوا/روشنی کی سمت کا اندازہ رکھیں؛ دھواں ہو تو نیچے رہیں۔",
+                "آہستہ سانس، توانائی بچائیں۔",
+                "صرف جب مدد قریب لگے تب آواز دیں/اشارہ کریں۔",
+                "دھواں ہو تو نیچے رہیں؛ تازہ ہوا/روشنی کی سمت بڑھیں۔",
             ]
         else:
             bullets += [
-                "فوراً اپنے آس پاس کے خطرات پہچانیں (بجلی، گیس، دھواں، پانی).",
+                "فوراً اردگرد کے خطرات پہچانیں (بجلی، گیس، دھواں، پانی).",
                 "محفوظ جگہ/کور لیں، بچوں/بزرگوں کو ساتھ رکھیں۔",
                 "ایگزٹ راستہ ذہن میں رکھیں، لفٹ سے پرہیز کریں۔",
             ]
         lines = ["✅ یہاں آپ کیا کریں:"] + [f"- {b}" for b in bullets]
-        lines.append("🕊️ گھبرائیں نہیں — میں آپ کے ساتھ ہوں۔")
+        lines.append(calm_ur)
         return "\n".join(lines)
     else:
         if has_kw(user_msg, "fire"):
@@ -210,7 +203,7 @@ def module_response(user_msg: str, lang: str) -> str:
                 "Stay low — smoke rises.",
                 "Cover nose/mouth with a wet cloth.",
                 "Turn off electricity/gas if safe.",
-                "Check door heat with back of hand; use alternate exit if hot.",
+                "Check doors for heat; use alternate exit if hot.",
                 "Use stairs, never elevators.",
                 "Keep children/elderly with you; headcount.",
             ]
@@ -225,14 +218,14 @@ def module_response(user_msg: str, lang: str) -> str:
             bullets += [
                 "Drop, Cover, and Hold On.",
                 "Stay away from windows/heavy furniture.",
-                "After shaking stops, exit via safe route.",
+                "After shaking stops, exit via a safe route.",
                 "Use stairs; avoid elevators.",
             ]
         elif has_kw(user_msg, "flood") or has_kw(user_msg, "storm"):
             bullets += [
                 "Turn off main power (if safe).",
                 "Move to higher ground.",
-                "Do not drive through water; 12 inches can stall/float cars.",
+                "Do not drive through water; ~12 inches can float cars.",
                 "Grab go-bag, IDs, medicines.",
             ]
         elif has_kw(user_msg, "injury"):
@@ -249,7 +242,7 @@ def module_response(user_msg: str, lang: str) -> str:
             ]
         elif has_kw(user_msg, "trapped") or has_kw(user_msg, "child"):
             bullets += [
-                "Stay calm, breathe slowly — conserve energy.",
+                "Breathe slowly — conserve energy.",
                 "Signal only when rescuers are near.",
                 "Stay low if smoke; move toward fresh air/light.",
             ]
@@ -260,11 +253,11 @@ def module_response(user_msg: str, lang: str) -> str:
                 "Know exits; avoid elevators.",
             ]
         lines = ["✅ Do this now:"] + [f"- {b}" for b in bullets]
-        lines.append("🕊️ Stay calm — I’m with you.")
+        lines.append(calm_en)
         return "\n".join(lines)
 
 # ----------------------------
-# Helplines formatter (always added for SOFT/BLOCK)
+# Helplines formatter
 # ----------------------------
 def helplines_block(lang: str) -> str:
     if not GUARD.helplines:
@@ -284,7 +277,6 @@ def helplines_block(lang: str) -> str:
 SIM_TRIGGER = re.compile(r"(practice with me|drill mode|drill|پریکٹس|ڈرل)", re.I)
 
 def start_sim_scenario(user_msg: str) -> Dict:
-    # pick scenario by keyword; default to earthquake
     if has_kw(user_msg, "fire"):
         scen = "fire"
     elif has_kw(user_msg, "flood"):
@@ -300,31 +292,31 @@ def sim_prompt(state: Dict, lang: str) -> str:
     if lang == "ur":
         if state["step"] == 1:
             if scen == "fire":
-                return "ڈرل شروع: کمرے میں آگ لگی ہے اور دھواں بڑھ رہا ہے۔ آپ سب سے پہلے کیا کریں گے؟"
+                return "ڈرل: کمرے میں آگ لگی ہے اور دھواں بڑھ رہا ہے۔ سب سے پہلا قدم کیا ہوگا؟"
             if scen == "flood":
-                return "ڈرل شروع: پانی تیزی سے گھر میں داخل ہو رہا ہے۔ آپ سب سے پہلے کیا کریں گے؟"
-            return "ڈرل شروع: زلزلے کے جھٹکے محسوس ہو رہے ہیں۔ آپ سب سے پہلے کیا کریں گے؟"
+                return "ڈرل: پانی تیزی سے گھر میں داخل ہو رہا ہے۔ سب سے پہلا قدم کیا ہوگا؟"
+            return "ڈرل: زلزلے کے جھٹکے محسوس ہو رہے ہیں۔ سب سے پہلا قدم کیا ہوگا؟"
         elif state["step"] == 2:
-            return "اچھا—اب دوسرا قدم کیا ہوگا؟ (مختصر ایک لائن میں بتائیں)"
+            return "اچھا—اب دوسرا قدم کیا ہوگا؟ (ایک لائن میں)"
         else:
-            return "آخری قدم کیا ہوگا جس سے آپ/گھر والے محفوظ رہیں گے؟"
+            return "آخری حفاظتی قدم کیا ہوگا؟"
     else:
         if state["step"] == 1:
             if scen == "fire":
-                return "Drill: There is a room fire and smoke is rising. What is your FIRST action?"
+                return "Drill: Room fire with rising smoke. What is your FIRST action?"
             if scen == "flood":
                 return "Drill: Water is rapidly entering the house. What is your FIRST action?"
-            return "Drill: You feel earthquake shaking. What is your FIRST action?"
+            return "Drill: Earthquake shaking. What is your FIRST action?"
         elif state["step"] == 2:
             return "Good — what is your SECOND step? (one short line)"
         else:
-            return "What is the FINAL step to keep you/your family safe?"
+            return "What is the FINAL step to keep everyone safe?"
 
 def sim_feedback(state: Dict, user_msg: str, lang: str) -> str:
     scen = state.get("scenario", "earthquake")
     tips_ur = {
-        "fire": ["جھک جائیں، دھوئیں سے نیچے رہیں", "گیلا کپڑا ناک/منہ پر", "لفٹ ہرگز نہیں — سیڑھیاں", "گرم دروازہ مت کھولیں"],
-        "flood": ["بجلی مین سوئچ بند", "اونچی جگہ/چھت پر جانا", "گاڑی/پانی سے گزرنے سے پرہیز", "گو-بیگ/شناختی دستاویزات ساتھ"],
+        "fire": ["جھک جائیں، دھوئیں سے نیچے رہیں", "گیلا کپڑا ناک/منہ پر", "لفٹ نہیں — سیڑھیاں", "گرم دروازہ مت کھولیں"],
+        "flood": ["بجلی مین سوئچ بند", "اونچی جگہ/چھت پر جانا", "پانی میں گاڑی نہ چلائیں", "گو-بیگ/شناختی دستاویزات ساتھ"],
         "earthquake": ["Drop, Cover, Hold", "کھڑکیوں/بھاری اشیاء سے دور", "جھٹکے رکنے پر سیڑھیوں سے باہر", "لفٹ نہیں"],
     }
     tips_en = {
@@ -333,7 +325,6 @@ def sim_feedback(state: Dict, user_msg: str, lang: str) -> str:
         "earthquake": ["Drop, Cover, Hold", "Stay away from windows/heavy objects", "Exit via stairs after shaking", "No elevators"],
     }
     good = tips_ur if lang == "ur" else tips_en
-    # very light scoring by keyword overlap
     score = 0
     low = user_msg.lower()
     for hint in good[scen]:
@@ -342,7 +333,7 @@ def sim_feedback(state: Dict, user_msg: str, lang: str) -> str:
                 score += 1
                 break
     if lang == "ur":
-        fb = "درست سمت میں ہیں۔ " if score >= 1 else "بہتر کریں: "
+        fb = "درست سمت میں ہیں۔ " if score >= 1 else "بہتری کی ضرورت: "
         checklist = "\n- " + "\n- ".join(good[scen])
         return fb + "محفوظ قدم یہ ہیں:" + checklist
     else:
@@ -369,20 +360,19 @@ def chat_fn(user_msg: str, chat_history: List[Tuple[str, str]], sim_state: Dict)
         reply = sim_prompt(sim_state, lang)
         return chat_history + [(user_msg, reply)], sim_state
 
-    # Simulation step flow
+    # Simulation flow
     if sim_state.get("active"):
-        # give feedback and advance
         feedback = sim_feedback(sim_state, user_msg, lang)
         sim_state["step"] += 1
         if sim_state["step"] > 3:
             sim_state = {"active": False, "step": 0, "scenario": None}
-            end_line = "ڈرل مکمل — شاباش! آپ نے بہتر فیصلہ سازی دیکھی۔ محفوظ رہیں۔" if lang == "ur" else "Drill complete — well done. Stay safe."
+            end_line = "ڈرل مکمل — شاباش! محفوظ رہیں۔" if lang == "ur" else "Drill complete — well done. Stay safe."
             return chat_history + [(user_msg, feedback + "\n\n" + end_line)], sim_state
         else:
             prompt_q = sim_prompt(sim_state, lang)
             return chat_history + [(user_msg, feedback + "\n\n" + prompt_q)], sim_state
 
-    # Normal emergency routing (no questions)
+    # Normal routing (no clarifying questions)
     if status == "BLOCK":
         msg = GUARD.refusal_msg_ur if lang == "ur" else GUARD.refusal_msg_en
         helpb = helplines_block(lang)
@@ -390,14 +380,12 @@ def chat_fn(user_msg: str, chat_history: List[Tuple[str, str]], sim_state: Dict)
             msg += "\n\n" + helpb
         return chat_history + [(user_msg, msg)], sim_state
 
-    # Immediate checklist from modules
     checklist = module_response(user_msg, lang)
 
-    # Compose with system prompt for any extra refinement (short)
+    # Optional short LLM tail (kept small for CPU)
     prompt = build_prompt(user_msg, lang)
-    model_tail = llm_reply(prompt, max_new_tokens=80, temperature=0.5)
+    _ = llm_reply(prompt, max_new_tokens=80, temperature=0.5)  # computed but not appended to keep replies concise
 
-    # Soft case: prepend a gentle note + add helplines
     if status == "SOFT":
         note = "احتیاط: یہ حساس موضوع ہے۔\n\n" if lang == "ur" else "Note: This is a sensitive topic.\n\n"
         checklist = note + checklist
@@ -405,7 +393,7 @@ def chat_fn(user_msg: str, chat_history: List[Tuple[str, str]], sim_state: Dict)
         if helpb:
             checklist += "\n\n" + helpb
 
-    final = checklist  # keep concise; tail is optional if needed
+    final = checklist
     return chat_history + [(user_msg, final)], sim_state
 
 # ----------------------------
@@ -418,6 +406,7 @@ with gr.Blocks(title="SafePak – 1122 Guide") as demo:
 
     chat = gr.Chatbot(height=480)
     msg = gr.Textbox(label="Message / پیغام", placeholder="Type here… / یہاں لکھیں…", lines=2)
+    send = gr.Button("Send / بھیجیں")
     clear = gr.Button("Clear")
 
     def on_submit(u, h, s):
@@ -425,6 +414,7 @@ with gr.Blocks(title="SafePak – 1122 Guide") as demo:
         return new_h, "", new_s
 
     msg.submit(on_submit, [msg, chat, state], [chat, msg, state])
+    send.click(on_submit, [msg, chat, state], [chat, msg, state])
     clear.click(lambda: ([], {"active": False, "step": 0, "scenario": None}), [], [chat, state])
 
 if __name__ == "__main__":