Update app.py

app.py

@@ -553,5 +553,122 @@ with gr.Blocks(title="Text2Receipt") as demo:
         outputs=[chatbot, doc_output, recs_output, st_parse, st_answers, note_input],
     )
 
+# ═════════════════════════════════════════════════════════════════════════════
+# 8.  Discord bot — runs as a background thread inside the Space  (+5% bonus)
+# ═════════════════════════════════════════════════════════════════════════════
+# The bot shares the SAME model loaded by _lazy_init(), so no extra memory.
+# Token is read from the DISCORD_TOKEN secret (Space → Settings → Secrets).
+# If the secret is absent, the bot is silently skipped and the UI still works.
+import threading, asyncio
+
+def _format_doc_for_discord(completed: dict) -> str:
+    """Render a completed fiscal document as Discord-friendly monospace text."""
+    c       = completed
+    issuer  = c.get("issuer", {})
+    client  = c.get("client", {})
+    lines   = c.get("lines",  [])
+    vat_pct = int(round(c.get("vat_rate", 0) * 100))
+
+    line_rows = "\n".join(
+        f"  • {ln['description']} × {ln['quantity']} → ₪{ln['line_total']:,.2f}"
+        for ln in lines
+    )
+    alloc = (f"\n🔖 מספר הקצאה: {c.get('allocation_number')}"
+             if c.get("allocation_required") else "")
+
+    return (
+        f"```\n"
+        f"{'━'*40}\n"
+        f"  {c.get('doc_type_he','מסמך')}   |   מס׳ {c.get('serial_number','')}\n"
+        f"  {c.get('issue_date','')}\n"
+        f"{'━'*40}\n"
+        f"  מנפיק : {issuer.get('name','')}  (ח.פ. {issuer.get('tax_id','')})\n"
+        f"  לקוח   : {client.get('name','')}"
+        f"{'  [עסק]' if client.get('is_business') else ''}\n"
+        f"{'━'*40}\n"
+        f"{line_rows}\n"
+        f"{'━'*40}\n"
+        f"  סכום נטו : ₪{c.get('subtotal',0):>12,.2f}\n"
+        f"  מע\"מ {vat_pct:2d}% : ₪{c.get('vat_amount',0):>12,.2f}\n"
+        f"  סה\"כ     : ₪{c.get('total',0):>12,.2f}\n"
+        f"  תשלום : {core.PAYMENT_HE.get(c.get('payment_method',''),'—')}\n"
+        f"{'━'*40}\n"
+        f"```"
+        f"{alloc}"
+    )
+
+
+def _discord_pipeline(raw_text: str) -> str:
+    """Full pipeline for Discord: raw Hebrew note → formatted document string.
+    Uses safe defaults (no agentic loop in chat — Discord is single-shot)."""
+    _lazy_init()
+    parse = model_parse(raw_text)
+    if parse is None:
+        return "❌ לא הצלחתי לחלץ את הפרטים. נסה לנסח מחדש."
+    final = apply_answers(parse, {})           # apply_answers fills safe defaults
+    try:
+        completed = core.complete(DEMO_ISSUER, final, _STATE["rng"])
+    except Exception as e:
+        return f"❌ שגיאה בעיבוד: {e}"
+    return _format_doc_for_discord(completed)
+
+
+def _start_discord_bot():
+    """Launch the Discord bot in a background thread if DISCORD_TOKEN is set."""
+    token = os.environ.get("DISCORD_TOKEN", "").strip()
+    if not token:
+        print("ℹ DISCORD_TOKEN not set — Discord bot skipped (UI still runs).")
+        return
+
+    try:
+        import discord
+        from discord.ext import commands
+    except ImportError:
+        print("⚠ discord.py not installed — add 'discord.py' to requirements.txt")
+        return
+
+    intents = discord.Intents.default()
+    intents.message_content = True
+    bot = commands.Bot(command_prefix="!", intents=intents)
+
+    @bot.event
+    async def on_ready():
+        print(f"✅ Discord bot online: {bot.user}")
+
+    @bot.command(name="receipt")
+    async def receipt_cmd(ctx, *, note: str):
+        await ctx.message.add_reaction("⏳")
+        try:
+            loop   = asyncio.get_event_loop()
+            result = await loop.run_in_executor(None, _discord_pipeline, note)
+            await ctx.reply(result)
+        except Exception as e:
+            await ctx.reply(f"❌ שגיאה: {e}")
+        finally:
+            try:
+                await ctx.message.remove_reaction("⏳", bot.user)
+                await ctx.message.add_reaction("✅")
+            except Exception:
+                pass
+
+    @bot.command(name="help_t2r")
+    async def help_cmd(ctx):
+        await ctx.reply(
+            "**Text2Receipt Bot** 🧾\n"
+            "המר הערת הכנסה עברית למסמך פיסקלי ישראלי.\n\n"
+            "שימוש: `!receipt <הערה בעברית>`\n"
+            "דוגמה: `!receipt קיבלתי 500 שקל ממשה על ייעוץ`"
+        )
+
+    def _run():
+        asyncio.run(bot.start(token))
+
+    threading.Thread(target=_run, daemon=True).start()
+    print("🤖 Discord bot thread started.")
+
+# ═══════════════════════════════════��═════════════════════════════════════════
+# 9.  Launch
+# ═════════════════════════════════════════════════════════════════════════════
 if __name__ == "__main__":
+    _start_discord_bot()          # starts only if DISCORD_TOKEN secret is set
     demo.launch(css=CSS)