Update app.py

app.py

@@ -1,20 +1,25 @@
 import os
 import re
 import gradio as gr
+from typing import List, Tuple
 from dataclasses import dataclass
-from typing import List, Dict, Tuple
 
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# ---------------- Language detection ----------------
 ARABIC_RE = re.compile(r"[\u0600-\u06FF]")
 def detect_lang(text: str) -> str:
     return "ur" if ARABIC_RE.search(text) else "en"
 
+# ---------------- Guardrails ----------------
 @dataclass
 class Guardrails:
     refusal_msg_ur: str
     refusal_msg_en: str
     blocked_patterns: List[str]
     soft_patterns: List[str]
-    helplines: List[Dict[str, str]]
+    helplines: List[dict]
 
     @classmethod
     def from_yaml(cls, path: str):
@@ -22,10 +27,10 @@ class Guardrails:
         with open(path, "r", encoding="utf-8") as f:
             data = yaml.safe_load(f)
         return cls(
-            refusal_msg_ur=data["refusal_msg_ur"],
-            refusal_msg_en=data["refusal_msg_en"],
-            blocked_patterns=data["blocked_patterns"],
-            soft_patterns=data["soft_patterns"],
+            refusal_msg_ur=data.get("refusal_msg_ur", ""),
+            refusal_msg_en=data.get("refusal_msg_en", ""),
+            blocked_patterns=data.get("blocked_patterns", []),
+            soft_patterns=data.get("soft_patterns", []),
             helplines=data.get("helplines", []),
         )
 
@@ -41,238 +46,54 @@ class Guardrails:
 
 GUARD = Guardrails.from_yaml("guardrails.yaml")
 
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-import torch
-
+# ---------------- Model ----------------
 MODEL_ID = os.environ.get("SAFEPak_MODEL_ID", "Qwen/Qwen2-0.5B-Instruct")
 try:
-    TOKENIZER = AutoTokenizer.from_pretrained(MODEL_ID)
-    MODEL = AutoModelForCausalLM.from_pretrained(MODEL_ID, torch_dtype=torch.float32)
-    GEN = pipeline(
-        "text-generation",
-        model=MODEL,
-        tokenizer=TOKENIZER,
-        device_map="auto" if torch.cuda.is_available() else None,
-    )
-    TINY_FALLBACK = False
+    tok = AutoTokenizer.from_pretrained(MODEL_ID)
+    model = AutoModelForCausalLM.from_pretrained(MODEL_ID, torch_dtype=torch.float32)
 except Exception as e:
-    print("LLM load failed, tiny fallback active:", e)
-    TOKENIZER = None
-    GEN = None
-    TINY_FALLBACK = True
-
-def llm_reply(prompt: str, max_new_tokens: int = 120, temperature: float = 0.3) -> str:
-    if TINY_FALLBACK:
-        return ""
-    out = GEN(
-        prompt,
+    print("LLM load failed, fallback active:", e)
+    tok = model = None
+
+def llm_reply(prompt: str, max_new_tokens: int = 100, temperature: float = 0.3) -> str:
+    if not (tok and model):
+        # fallback جواب
+        return "safePak: میں آپ کی مدد کے لیے حاضر ہوں۔ براہ کرم مزید وضاحت دیں۔"
+    ids = tok.encode(prompt, return_tensors="pt")
+    out = model.generate(
+        ids,
         max_new_tokens=max_new_tokens,
-        do_sample=True,
         temperature=temperature,
-        pad_token_id=TOKENIZER.eos_token_id,
-    )[0]["generated_text"]
-    return out[len(prompt):].strip()
-
-SAFEPAK_SYSTEM = """You are a multilingual emergency disaster response assistant named “SafePak – 1122 Guide”.
-Provide immediate, clear, practical survival and rescue instructions for disasters, accidents, conflict, or panic in Pakistan-like regions.
-No clarifying questions in normal mode. Use concise checklists. Begin with a calming line if user seems distressed.
-Trigger modules by keywords (fire, gas, earthquake, flood, storm, injury, trapped, explosion, conflict, child).
-Offer offline-safe advice. No personal data storage. No medical diagnoses, only first aid."""
+        do_sample=True,
+        pad_token_id=tok.eos_token_id,
+    )
+    return tok.decode(out[0], skip_special_tokens=True)
 
-KW = {
-    "fire": ["fire", "room on fire", "smoke", "جل", "آگ", "جلدی", "smelling smoke"],
-    "gas": ["gas", "gas leak", "smell gas", "cylinder", "گیس", "لیک", "بو"],
-    "earthquake": ["earthquake", "tremor", "shaking", "زلزلہ", "جھٹکا", "jhatka"],
-    "flood": ["flood", "water entering", "overflow", "سیلاب", "بارش", "پانی آ گیا"],
-    "storm": ["storm", "cyclone", "high winds", "طوفان", "آندھی"],
-    "injury": ["injury", "bleeding", "cut", "fracture", "burn wound", "زخمی", "خون"],
-    "explosion": ["explosion", "blast", "bomb blast", "دھماکا"],
-    "conflict": ["gunfire", "firing", "crossfire", "terror", "bomb", "گولیاں"],
-    "trapped": ["trapped", "stuck", "buried", "rubble", "پھنس"],
-    "child": ["child", "baby", "بچہ", "رو رہا"],
-}
+# ---------------- Helpline rendering ----------------
+def _clean_phone_for_links(p: str) -> str:
+    return "+" + "".join(ch for ch in p if ch.isdigit())
 
-def has_kw(text: str, key: str) -> bool:
-    low = text.lower()
-    for token in KW.get(key, []):
-        if token in low:
-            return True
-    return False
+def phone_tel_link(p: str) -> str:
+    return f"tel:{_clean_phone_for_links(p)}"
 
-def module_response(user_msg: str, lang: str) -> Tuple[List[str], str]:
-    bullets = []
-    scenario = "general"
-    if lang == "ur":
-        calm = "🕊️ گھبرائیں نہیں — ایک گہری سانس لیں، میں آپ کے ساتھ ہوں۔"
-        if has_kw(user_msg, "fire"):
-            scenario = "fire"
-            bullets = [
-                "جھک جائیں — دھواں اوپر جاتا ہے۔",
-                "ناک/منہ پر گیلا کپڑا رکھیں۔",
-                "اگر محفوظ ہو تو بجلی/گیس مین بند کریں۔",
-                "گرم دروازہ مت کھولیں؛ دوسرا راستہ لیں۔",
-                "سیڑھیاں استعمال کریں — لفٹ ہرگز نہیں۔",
-                "بچوں/بزرگوں کو ساتھ رکھیں، گنتی کریں۔",
-                calm,
-            ]
-        elif has_kw(user_msg, "gas"):
-            scenario = "gas"
-            bullets = [
-                "چنگاری نہ بنائیں (سوئچ/لائٹر/فون فلیش نہیں)۔",
-                "کھڑکیاں/دروازے کھول کر ہوا دار کریں۔",
-                "گیس کا مین والو بند کریں۔",
-                "عمارت خالی کریں، محفوظ فاصلے پر جائیں۔",
-                calm,
-            ]
-        elif has_kw(user_msg, "earthquake"):
-            scenario = "earthquake"
-            bullets = [
-                "Drop, Cover, Hold — جھکیں، ڈھکیں، پکڑیں۔",
-                "کھڑکیوں/بھاری اشیاء سے دور رہیں۔",
-                "جھٹکے رکنے پر سیڑھیوں سے باہر نکلیں۔",
-                "کھلی جگہ میں رکیں؛ لفٹ استعمال نہ کریں۔",
-                calm,
-            ]
-        elif has_kw(user_msg, "flood") or has_kw(user_msg, "storm"):
-            scenario = "flood"
-            bullets = [
-                "اگر محفوظ ہو تو بجلی مین سوئچ بند کریں۔",
-                "اونچی جگہ/چھت پر منتقل ہوں۔",
-                "پانی میں گاڑی نہ چلائیں۔",
-                "اہم کاغذات اور دوائیں ساتھ رکھیں۔",
-                calm,
-            ]
-        elif has_kw(user_msg, "injury"):
-            scenario = "injury"
-            bullets = [
-                "خون بہہ رہا ہو تو مضبوط دباؤ دیں (صاف کپڑا)۔",
-                "زخم صاف پانی سے دھوئیں؛ گہرا ہو تو مدد لیں۔",
-                "گردن/کمر چوٹ پر مریض کو مت ہلائیں۔",
-                calm,
-            ]
-        elif has_kw(user_msg, "explosion") or has_kw(user_msg, "conflict"):
-            scenario = "conflict"
-            bullets = [
-                "زمین پر لیٹیں، مضبوط کور لیں، کھڑکیوں سے دور رہیں۔",
-                "گولیاں مخالف سمت میں کور لیتے ہوئے حرکت کریں۔",
-                "محفوظ راستہ دیکھ کر پرسکون انداز میں نکلیں۔",
-                calm,
-            ]
-        elif has_kw(user_msg, "trapped") or has_kw(user_msg, "child"):
-            scenario = "trapped"
-            bullets = [
-                "آہستہ سانس — توانائی بچائیں۔",
-                "صرف جب مدد قریب لگے تب آواز/اشارہ کریں۔",
-                "دھواں ہو تو نیچے رہیں اور تازہ ہوا کی سمت جائیں۔",
-                calm,
-            ]
-        else:
-            bullets = [
-                "قریب ترین خطرات پہچانیں (بجلی، گیس، دھواں، پانی)۔",
-                "محفوظ کور/خارج ہونے کا راستہ منتخب کریں۔",
-                "بچوں/بزرگوں کو ساتھ رکھیں؛ لفٹ سے پرہیز کریں۔",
-                calm,
-            ]
-    else:
-        calm = "🕊️ Stay calm — take a deep breath; I’m with you."
-        if has_kw(user_msg, "fire"):
-            scenario = "fire"
-            bullets = [
-                "Stay low — smoke rises.",
-                "Wet cloth over nose/mouth.",
-                "Cut power/gas if safe.",
-                "Do not open hot doors.",
-                "Use stairs, never elevators.",
-                "Keep kids/elderly with you; headcount.",
-                calm,
-            ]
-        elif has_kw(user_msg, "gas"):
-            scenario = "gas"
-            bullets = [
-                "No sparks (no switches/lighters/flash).",
-                "Ventilate by opening windows/doors.",
-                "Shut the main gas valve.",
-                "Evacuate to a safe distance.",
-                calm,
-            ]
-        elif has_kw(user_msg, "earthquake"):
-            scenario = "earthquake"
-            bullets = [
-                "Drop, Cover, Hold On.",
-                "Stay away from windows/heavy furniture.",
-                "Exit via stairs after shaking stops.",
-                "Wait in open safe area; no elevators.",
-                calm,
-            ]
-        elif has_kw(user_msg, "flood") or has_kw(user_msg, "storm"):
-            scenario = "flood"
-            bullets = [
-                "Turn off main power if safe.",
-                "Move to higher ground/roof.",
-                "Do not drive through water.",
-                "Keep IDs/medicines with you.",
-                calm,
-            ]
-        elif has_kw(user_msg, "injury"):
-            scenario = "injury"
-            bullets = [
-                "Apply firm direct pressure to stop bleeding.",
-                "Rinse with clean water; seek help if deep.",
-                "Suspected spine injury — do not move person.",
-                calm,
-            ]
-        elif has_kw(user_msg, "explosion") or has_kw(user_msg, "conflict"):
-            scenario = "conflict"
-            bullets = [
-                "Get low, take solid cover, away from windows.",
-                "Move opposite to gunfire using cover.",
-                "Exit safely; avoid crowds and noise.",
-                calm,
-            ]
-        elif has_kw(user_msg, "trapped") or has_kw(user_msg, "child"):
-            scenario = "trapped"
-            bullets = [
-                "Breathe slowly — conserve energy.",
-                "Signal only when rescuers seem near.",
-                "Stay low if smoke; move toward fresh air/light.",
-                calm,
-            ]
-        else:
-            bullets = [
-                "Identify immediate hazards (electricity, gas, smoke, water).",
-                "Choose safe cover/exit route.",
-                "Keep kids/elderly close; avoid elevators.",
-                calm,
-            ]
-    return bullets, scenario
+def phone_wa_link(p: str) -> str:
+    num = _clean_phone_for_links(p).lstrip("+")
+    return f"https://wa.me/{num}"
 
-def gobag_for(scenario: str, lang: str) -> List[str]:
-    if lang == "ur":
-        base = ["شناختی کارڈ/کاغذات", "نقدی", "موبائل + پاور بینک", "پینے کا پانی", "خشک کھانا", "دوائیں", "بچوں/بزرگوں کی ضروری اشیاء", "ہلکی چادر/کپڑے", "ٹارچ"]
-        if scenario == "fire":
-            return base + ["ماسک/گیلا کپڑا", "فائر سیفٹی آگاہی"]
-        if scenario == "flood":
-            return base + ["واٹر پروف بیگز", "چپل/بوٹس"]
-        if scenario == "earthquake":
-            return base + ["وسل", "چھوٹی فرسٹ ایڈ کٹ"]
-        if scenario == "gas":
-            return base + ["ریسپریٹری ماسک (اگر دستیاب)", "ماچس/لائٹر الگ بیگ میں"]
-        if scenario == "conflict":
-            return base + ["low-profile کپڑے", "ضروری نمبر نوٹ بک میں"]
-        return base
-    else:
-        base = ["IDs/documents", "Cash", "Phone + power bank", "Drinking water", "Dry food", "Medicines", "Baby/elderly needs", "Light blanket/clothes", "Torch/flashlight"]
-        if scenario == "fire":
-            return base + ["Mask/wet cloth", "Basic fire safety info"]
-        if scenario == "flood":
-            return base + ["Waterproof pouches", "Sandals/boots"]
-        if scenario == "earthquake":
-            return base + ["Whistle", "Small first-aid kit"]
-        if scenario == "gas":
-            return base + ["Respirator mask if available", "Matches/lighter stored separately"]
-        if scenario == "conflict":
-            return base + ["Low-profile clothing", "Emergency numbers notebook"]
-        return base
+def render_quick_helplines_md(lang: str = "ur") -> str:
+    if not GUARD.helplines:
+        return ""
+    title = "### فوری ہیلپ لائنز" if lang == "ur" else "### Quick Helplines"
+    rows = [title]
+    for h in GUARD.helplines:
+        name = h.get("name","").strip()
+        phone = h.get("phone","").strip()
+        if not (name and phone):
+            continue
+        tel = phone_tel_link(phone)
+        wa  = phone_wa_link(phone)
+        rows.append(f"**{name}**  \n{phone}\n[📞 Call]({tel})    [💬 WhatsApp]({wa})")
+    return "\n\n".join(rows)
 
 def helplines_block(lang: str) -> str:
     if not GUARD.helplines:
@@ -283,156 +104,87 @@ def helplines_block(lang: str) -> str:
         nm = h.get("name", "").strip()
         ph = h.get("phone", "").strip()
         if nm and ph:
-            lines.append(f"{nm}:\n{ph}")
-    return "\n".join(lines)
-
-SIM_TRIGGER = re.compile(r"(practice with me|drill mode|drill|پریکٹس|ڈرل)", re.I)
-
-def start_sim_scenario(user_msg: str) -> Dict:
-    if has_kw(user_msg, "fire"):
-        scen = "fire"
-    elif has_kw(user_msg, "flood"):
-        scen = "flood"
-    elif has_kw(user_msg, "earthquake"):
-        scen = "earthquake"
-    else:
-        scen = "earthquake"
-    return {"active": True, "step": 1, "scenario": scen}
-
-def sim_prompt(state: Dict, lang: str) -> str:
-    scen = state.get("scenario", "earthquake")
-    if lang == "ur":
-        if state["step"] == 1:
-            if scen == "fire":
-                return "ڈرل: کمرے میں آگ اور دھواں ہے — سب سے پہلا قدم کیا ہوگا؟"
-            if scen == "flood":
-                return "ڈرل: پانی تیزی سے گھر میں داخل ہو رہا ہے — سب سے پہلا قدم کیا ہوگا؟"
-            return "ڈرل: زلزلے کے جھٹکے محسوس ہو رہے ہیں — سب سے پہلا قدم کیا ہوگا؟"
-        elif state["step"] == 2:
-            return "اچھا—اب دوسرا قدم کیا ہوگا؟ (ایک لائن میں)"
-        else:
-            return "آخری حفاظتی قدم کیا ہوگا؟"
-    else:
-        if state["step"] == 1:
-            if scen == "fire":
-                return "Drill: Room fire with smoke — what is your FIRST action?"
-            if scen == "flood":
-                return "Drill: Water entering rapidly — 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 everyone safe?"
-
-def sim_feedback(state: Dict, user_msg: str, lang: str) -> str:
-    scen = state.get("scenario", "earthquake")
-    tips_ur = {
-        "fire": ["جھک جائیں، دھوئیں سے نیچے رہیں", "گیلا کپڑا ناک/منہ پر", "لفٹ نہیں — سیڑھیاں", "گرم دروازہ مت کھولیں"],
-        "flood": ["بجلی مین سوئچ بند", "اونچی جگہ/چھت پر جانا", "پانی میں گاڑی نہ چلائیں", "گو-بیگ/شناختی دستاویزات"],
-        "earthquake": ["Drop, Cover, Hold", "کھڑکیوں/بھاری اشیاء سے دور", "جھٹکے رکنے پر سیڑھیوں سے باہر", "لفٹ نہیں"],
-    }
-    tips_en = {
-        "fire": ["Stay low under smoke", "Wet cloth over nose/mouth", "Use stairs, not elevators", "Do not open hot doors"],
-        "flood": ["Turn off main power", "Move to higher ground/roof", "Avoid driving through water", "Grab go-bag/IDs"],
-        "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
-    score = 0
-    low = user_msg.lower()
-    for hint in good[scen]:
-        for tok in re.findall(r"\w+", hint.lower()):
-            if tok in low:
-                score += 1
-                break
-    if lang == "ur":
-        fb = "درست سمت میں ہیں۔ " if score >= 1 else "بہتری کی ضرورت: "
-        checklist = "\n- " + "\n- ".join(good[scen])
-        return fb + "محفوظ قدم یہ ہیں:" + checklist
-    else:
-        fb = "Good direction. " if score >= 1 else "Improve: "
-        checklist = "\n- " + "\n- ".join(good[scen])
-        return fb + "Follow these safe steps:" + checklist
-
-def build_prompt(user_msg: str, lang: str) -> str:
-    return f"<system>\n{SAFEPAK_SYSTEM}\n</system>\n<user>\n{user_msg}\n</user>"
+            tel = phone_tel_link(ph)
+            wa  = phone_wa_link(ph)
+            lines.append(f"**{nm}**\n{ph}\n[📞 Call]({tel})  |  [💬 WhatsApp]({wa})")
+    return "\n\n".join(lines)
+
+# ---------------- Core prompts ----------------
+SAFEPAK_SYSTEM = """
+You are SafePak – 1122 Guide, a multilingual emergency disaster assistant.
+Always reply in user's language.
+Give medium length, action-oriented checklists (5–7 bullets) + Go-Bag tips if evacuation is implied.
+Never panic. End with an emotional calming line.
+"""
+
+def format_response(base: str, user_msg: str, lang: str) -> str:
+    refine_prompt = (
+        f"<system>\n{SAFEPAK_SYSTEM}\n</system>\n"
+        f"<user>\n{user_msg}\n\nDraft response:\n{base}\n</user>"
+    )
+    return llm_reply(refine_prompt, max_new_tokens=100, temperature=0.3)
 
-def chat_fn(user_msg: str, chat_history: List[Tuple[str, str]], sim_state: Dict):
+# ---------------- Chat logic ----------------
+def chat_fn(user_msg: str, chat_history: List[Tuple[str, str]], sim_state: dict):
     if not user_msg:
         return chat_history, sim_state
-
     lang = detect_lang(user_msg)
-    status, _ = GUARD.check(user_msg)
-
-    if SIM_TRIGGER.search(user_msg):
-        sim_state = start_sim_scenario(user_msg)
-        reply = sim_prompt(sim_state, lang)
-        return chat_history + [(user_msg, reply)], sim_state
-
-    if sim_state.get("active"):
-        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."
-            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
+    status, reason = GUARD.check(user_msg)
 
+    # BLOCK
     if status == "BLOCK":
         msg = GUARD.refusal_msg_ur if lang == "ur" else GUARD.refusal_msg_en
-        helpb = helplines_block(lang)
-        if helpb:
-            msg += "\n\n" + helpb
         return chat_history + [(user_msg, msg)], sim_state
 
-    bullets, scenario = module_response(user_msg, lang)
-    bag = gobag_for(scenario, lang)
-    if lang == "ur":
-        base = "✅ ابھی یہ کریں:\n" + "\n".join(f"- {b}" for b in bullets if b) + "\n\n🎒 ساتھ کیا لیں (Go-Bag):\n" + "\n".join(f"- {i}" for i in bag)
+    # base checklist by keyword
+    checklist = ""
+    low = user_msg.lower()
+    if any(k in low for k in ["fire","آگ","smoke"]):
+        checklist = "Fire safety steps"
+    elif any(k in low for k in ["زلزلہ","earthquake","tremor"]):
+        checklist = "Earthquake safety steps"
+    elif any(k in low for k in ["سیلاب","flood","overflow"]):
+        checklist = "Flood survival steps"
+    elif any(k in low for k in ["gas","گیس","leak"]):
+        checklist = "Gas leak safety steps"
+    elif any(k in low for k in ["injury","زخمی","bleeding"]):
+        checklist = "First aid for injury"
     else:
-        base = "✅ Do this now:\n" + "\n".join(f"- {b}" for b in bullets if b) + "\n\n🎒 Take with you (Go-Bag):\n" + "\n".join(f"- {i}" for i in bag)
-
-    if status == "SOFT":
-        note = "احتیاط: یہ حساس موضوع ہے۔\n\n" if lang == "ur" else "Note: This is a sensitive topic.\n\n"
-        base = note + base
-
-    refined = ""
-    if not TINY_FALLBACK:
-        refine_prompt = (
-            f"<system>\n{SAFEPAK_SYSTEM}\n"
-            f"Return ONLY a medium-length emergency guide in the user's language: 5–7 action bullets, "
-            f"then a Go-Bag sublist (4–8 items). No preface, no extra prose. Be clear, practical, life-saving.\n</system>\n"
-            f"<user>\nOriginal message:\n{user_msg}\n\nBase content to refine:\n{base}\n</user>"
-        )
-        refined = llm_reply(refine_prompt, max_new_tokens=120, temperature=0.3)
+        checklist = "General emergency guidance"
 
-    final = refined.strip() if refined else base
+    final = format_response(checklist, user_msg, lang)
 
+    # SOFT warning
     if status == "SOFT":
-        helpb = helplines_block(lang)
-        if helpb:
-            final += "\n\n" + helpb
+        if lang == "ur":
+            final = "⚠️ محتاط رہیں: یہ موضوع حساس ہے۔\n\n" + final
+        else:
+            final = "⚠️ Note: Sensitive topic.\n\n" + final
 
+    final += "\n\n" + helplines_block(lang)
     return chat_history + [(user_msg, final)], sim_state
 
+# ---------------- UI ----------------
 with gr.Blocks(title="SafePak – 1122 Guide") as demo:
     gr.Markdown("## SafePak – 1122 Guide\nMultilingual emergency assistant (CPU-only).")
 
-    state = gr.State({"active": False, "step": 0, "scenario": None})
+    # ثابت اوپر Helplines panel
+    gr.Markdown(render_quick_helplines_md("ur"))
 
-    chat = gr.Chatbot(height=520)
-    msg = gr.Textbox(label="Message / پیغام", placeholder="Type here… / یہاں لکھیں…", lines=2)
+    state = gr.State({"active": False, "step": 0, "scenario": None})
+    chatbot = gr.Chatbot(height=420)
+    msg = gr.Textbox(label="پیغام / Message", placeholder="یہاں لکھیں…", lines=2)
     send = gr.Button("Send / بھیجیں")
     clear = gr.Button("Clear")
 
-    def on_submit(u, h, s):
-        new_h, new_s = chat_fn(u, h, s)
-        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])
+    def _on_submit(u,h,s): return chat_fn(u,h,s)
+    msg.submit(_on_submit, [msg,chatbot,state],[chatbot,state])
+    send.click(_on_submit, [msg,chatbot,state],[chatbot,state])
+    clear.click(lambda: ([], {"active":False,"step":0,"scenario":None}), [], [chatbot,state])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue(concurrency_count=1, max_size=32).launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )