Spaces:
Runtime error
Runtime error
File size: 3,938 Bytes
2a7171f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 | """Telegram bot — runs as a daemon thread sharing the same ledger instance."""
import os
import asyncio
import logging
import threading
from telegram import Update
from telegram.ext import (
ApplicationBuilder, CommandHandler, MessageHandler,
ContextTypes, filters,
)
from ledger import Ledger
from agent import batch_response, execute
logger = logging.getLogger(__name__)
MAX_HISTORY = 20 # messages (10 turns) kept per user
# ── handlers ──────────────────────────────────────────────────────────────────
async def on_start(update: Update, context: ContextTypes.DEFAULT_TYPE):
await update.message.reply_text(
"👋 *Finance Manager*\n\n"
"Just tell me about your expenses naturally:\n"
"• _Spent $12 on lunch_\n"
"• _Paid $1200 rent yesterday_\n"
"• _Undo_ — removes the last entry\n\n"
"Commands:\n"
"/summary — spending by category\n"
"/clear — reset conversation history",
parse_mode="Markdown",
)
async def on_summary(update: Update, context: ContextTypes.DEFAULT_TYPE):
ledger: Ledger = context.bot_data["ledger"]
by_cat = ledger.by_category()
total = ledger.total()
if not by_cat:
await update.message.reply_text("No entries yet. Start logging expenses!")
return
lines = [f"💰 *Total: ${total:.2f}*\n"]
lines += [
f"• {cat}: ${amt:.2f}"
for cat, amt in sorted(by_cat.items(), key=lambda x: -x[1])
]
await update.message.reply_text("\n".join(lines), parse_mode="Markdown")
async def on_clear(update: Update, context: ContextTypes.DEFAULT_TYPE):
context.user_data["history"] = []
await update.message.reply_text("Conversation history cleared.")
async def on_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
ledger: Ledger = context.bot_data["ledger"]
token = os.getenv("HF_TOKEN", "")
if not token:
await update.message.reply_text("HF_TOKEN is not configured.")
return
history: list[dict] = context.user_data.get("history", [])
text = update.message.text
reply, action = await asyncio.to_thread(batch_response, text, history, ledger, token)
if action:
confirmation = execute(action, ledger, text)
if confirmation:
reply += f"\n\n{confirmation}"
# Persist last N messages for context
context.user_data["history"] = (
history + [
{"role": "user", "content": text},
{"role": "assistant", "content": reply},
]
)[-MAX_HISTORY:]
await update.message.reply_text(reply, parse_mode="Markdown")
# ── entry point ───────────────────────────────────────────────────────────────
def start(ledger: Ledger):
"""Start the Telegram bot in a daemon thread. No-op if token not set."""
bot_token = os.getenv("TELEGRAM_BOT_TOKEN")
if not bot_token:
logger.info("TELEGRAM_BOT_TOKEN not set — Telegram bot disabled.")
return
async def _run():
app = ApplicationBuilder().token(bot_token).build()
app.bot_data["ledger"] = ledger
app.add_handler(CommandHandler("start", on_start))
app.add_handler(CommandHandler("summary", on_summary))
app.add_handler(CommandHandler("clear", on_clear))
app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, on_message))
logger.info("Telegram bot polling started.")
async with app:
await app.start()
await app.updater.start_polling()
await asyncio.Event().wait() # run until process exits
threading.Thread(target=lambda: asyncio.run(_run()), daemon=True).start()
|