Initial upload of Hotel OS Bot project

.dockerignore

@@ -0,0 +1,19 @@
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.Python
+env/
+venv/
+.env
+.env.*
+.pytest_cache/
+.vscode/
+.idea/
+*.swp
+*.swo
+*.swn
+data/
+*.db
+*.db-journal
+slips/

.env.example

@@ -0,0 +1,4 @@
+TELEGRAM_BOT_TOKEN=123456789:YOUR_BOT_TOKEN
+TELEGRAM_USER_ID=123456789
+GEMINI_API_KEY=AIzaSy...
+GEMINI_MODEL=gemini-pro

Dockerfile

@@ -0,0 +1,19 @@
+FROM python:3.12-slim
+
+RUN apt-get update && apt-get install -y \
+    tesseract-ocr tesseract-ocr-tha libgl1 \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+RUN mkdir -p /app/data && chmod 777 /app/data
+
+EXPOSE 8080
+
+CMD ["python", "-m", "app.main"]
+

README.md

@@ -1,11 +1,60 @@
----
-title: Bot Telegram
-emoji: 📈
-colorFrom: red
-colorTo: pink
-sdk: docker
-pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Hotel OS Bot
+
+A Telegram bot for managing hotel operations, including bookings, payments, and expenses.
+
+## Project Structure
+
+```
+hotel_os_bot/
+├── app/
+│   ├── __init__.py
+│   ├── main.py           # Bot + FastAPI + Polling
+│   ├── utils.py          # OCR, Verify, DB helpers
+│   └── resources/
+│       └── pms_blueprint.txt
+├── webapp/
+│   ├── index.html
+│   ├── style.css
+│   └── app.js
+├── scripts/
+│   ├── backup_db.py
+│   └── notify_pending.py
+├── data/
+├── requirements.txt
+├── Dockerfile
+├── .dockerignore
+├── start.sh
+├── space.yaml
+├── .env.example
+└── README.md
+```
+
+## How to Run
+
+1.  **Set up the environment:**
+    ```bash
+    cp .env.example .env
+    ```
+    Then, edit the `.env` file with your actual credentials (Telegram Bot Token, User ID, Gemini API Key).
+
+2.  **Install dependencies:**
+    ```bash
+    pip install -r requirements.txt
+    ```
+
+3.  **Run the application:**
+    ```bash
+    chmod +x start.sh
+    ./start.sh
+    ```
+    Alternatively, you can run the main module directly:
+    ```bash
+    python -m app.main
+    ```
+
+## Deploy to Hugging Face Spaces
+
+1.  Push the code to a GitHub repository.
+2.  Create a new Space on Hugging Face, selecting the "Docker" runtime.
+3.  Connect the Space to your GitHub repository.
+4.  In the Space settings, add the secrets from your `.env` file (`TELEGRAM_BOT_TOKEN`, `TELEGRAM_USER_ID`, `GEMINI_API_KEY`).

app/__init__.py

@@ -0,0 +1 @@
+# Empty file to make this a package

app/main.py

@@ -0,0 +1,299 @@
+from __future__ import annotations
+
+import asyncio
+import json
+import logging
+import os
+import sqlite3
+from dataclasses import dataclass
+from datetime import date, datetime
+from decimal import Decimal
+from pathlib import Path
+from typing import Optional
+
+import google.generativeai as genai
+import pytesseract
+import uvicorn
+from fastapi import FastAPI
+from telegram import KeyboardButton, ReplyKeyboardMarkup, Update, WebAppInfo
+from telegram.ext import (
+    Application,
+    CommandHandler,
+    ContextTypes,
+    MessageHandler,
+    filters,
+)
+
+from .utils import (
+    Settings,
+    SlipExtractionResult,
+    extract_slip_information,
+    verify_booking_amount,
+    save_base64_image,
+    chunk_for_telegram,
+)
+
+# FastAPI app
+app = FastAPI()
+bot_app: Optional[Application] = None
+
+# Logging
+logging.basicConfig(
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    level=logging.INFO,
+)
+logger = logging.getLogger(__name__)
+
+
+def load_settings() -> Settings:
+    """Load settings from environment variables."""
+    def req_env(name: str) -> str:
+        val = os.getenv(name)
+        if not val:
+            raise RuntimeError(f"Missing env var: {name}")
+        return val
+
+    return Settings(
+        telegram_bot_token=req_env("TELEGRAM_BOT_TOKEN"),
+        authorized_user_id=int(req_env("TELEGRAM_USER_ID")),
+        gemini_api_key=req_env("GEMINI_API_KEY"),
+        gemini_model=os.getenv("GEMINI_MODEL", "gemini-pro"),
+        pms_blueprint_path=Path("app/resources/pms_blueprint.txt"),
+        database_path=Path("./data/hotel_os_bot.db"),
+        slip_storage_dir=Path("./data/slips"),
+        webapp_base_url=os.getenv("WEBAPP_BASE_URL", "http://localhost:8080/index.html"),
+        webapp_static_dir=Path("./webapp"),
+        webapp_host="0.0.0.0",
+        webapp_port=int(os.getenv("PORT", "8080")),
+        ocr_language=os.getenv("OCR_LANGUAGE", "eng+tha"),
+        tesseract_cmd=os.getenv("TESSERACT_CMD"),
+        amount_tolerance=Decimal("1.00"),
+    )
+
+
+def init_db(settings: Settings):
+    """Initialize SQLite database."""
+    settings.database_path.parent.mkdir(parents=True, exist_ok=True)
+    settings.slip_storage_dir.mkdir(parents=True, exist_ok=True)
+
+    with sqlite3.connect(settings.database_path) as conn:
+        conn.executescript("""
+            CREATE TABLE IF NOT EXISTS bookings (
+                id TEXT PRIMARY KEY,
+                guest_name TEXT,
+                total_due REAL,
+                currency TEXT,
+                status TEXT,
+                created_at TEXT DEFAULT CURRENT_TIMESTAMP
+            );
+
+            CREATE TABLE IF NOT EXISTS payment_slips (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                booking_id TEXT,
+                reference_code TEXT,
+                slip_image_path TEXT,
+                extracted_amount REAL,
+                extracted_date TEXT,
+                extracted_time TEXT,
+                status TEXT,
+                ocr_text TEXT,
+                created_at TEXT DEFAULT CURRENT_TIMESTAMP
+            );
+
+            CREATE TABLE IF NOT EXISTS expenses (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                amount REAL NOT NULL,
+                category TEXT NOT NULL,
+                note TEXT,
+                created_by TEXT,
+                created_at TEXT DEFAULT CURRENT_TIMESTAMP
+            );
+        """)
+
+
+def is_authorized(update: Update, context: ContextTypes.DEFAULT_TYPE) -> bool:
+    """Check if user is authorized."""
+    user = update.effective_user
+    if not user:
+        return False
+    auth_id = context.application.bot_data.get("authorized_user_id")
+    return auth_id is not None and user.id == auth_id
+
+
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """Handle /start command."""
+    if not is_authorized(update, context):
+        await update.message.reply_text("🔐 คุณไม่มีสิทธิ์ใช้งานบอทนี้")
+        return
+
+    stg: Settings = context.application.bot_data["settings"]
+    keyboard = ReplyKeyboardMarkup(
+        [[KeyboardButton("🛎️ เปิดระบบบริหารโรงแรม", web_app=WebAppInfo(url=stg.webapp_base_url))]],
+        resize_keyboard=True
+    )
+    await update.message.reply_text("ยินดีต้อนรับสู่ Hotel OS Bot!", reply_markup=keyboard)
+
+
+async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """Handle /help command."""
+    if not is_authorized(update, context):
+        return
+    await update.message.reply_text(
+        "/start – เปิดระบบ\n"
+        "/daily_report – รายงานประจำวัน\n"
+        "/receipt <ID> – ใบเสร็จ\n"
+        "พิมพ์ข้อความ → AI ตอบ"
+    )
+
+
+async def daily_report(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """Handle /daily_report command."""
+    if not is_authorized(update, context):
+        return
+
+    stg: Settings = context.application.bot_data["settings"]
+    today = date.today().isoformat()
+
+    with sqlite3.connect(stg.database_path) as conn:
+        bookings = conn.execute("SELECT COUNT(*) FROM bookings WHERE DATE(created_at) = ?", (today,)).fetchone()[0]
+        paid = conn.execute("SELECT SUM(extracted_amount) FROM payment_slips WHERE status = 'verified' AND DATE(created_at) = ?", (today,)).fetchone()[0] or 0
+        expense = conn.execute("SELECT SUM(amount) FROM expenses WHERE DATE(created_at) = ?", (today,)).fetchone()[0] or 0
+
+    await update.message.reply_text(
+        f"📊 รายงานวันที่ {today}:\n"
+        f"• การจอง: {bookings} ราย\n"
+        f"• ชำระ: {paid:,.2f} บาท\n"
+        f"• ค่าใช้จ่าย: {expense:,.2f} บาท\n"
+        f"• กำไร: {paid - expense:,.2f} บาท"
+    )
+
+
+async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """Handle text messages with AI."""
+    if not is_authorized(update, context) or not update.message.text:
+        return
+
+    model = context.application.bot_data.get("gemini_model")
+    if not model:
+        await update.message.reply_text("AI ยังไม่พร้อม")
+        return
+
+    try:
+        resp = await model.generate_content_async(update.message.text)
+        await update.message.reply_text(resp.text or "ขอโทษครับ ผมตอบไม่ได้")
+    except Exception as e:
+        logger.error(f"Gemini error: {e}")
+        await update.message.reply_text("เกิดข้อผิดพลาดกับ AI")
+
+
+async def web_app_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """Handle data from Web App."""
+    if not is_authorized(update, context) or not update.message.web_app_data:
+        return
+
+    try:
+        payload = json.loads(update.message.web_app_data.data)
+    except json.JSONDecodeError:
+        await update.message.reply_text("ข้อมูลผิดพลาด")
+        return
+
+    if payload.get("type") == "payment_slip":
+        await process_slip(update, context, payload)
+    elif payload.get("type") == "expense_entry":
+        await process_expense(update, context, payload)
+
+
+async def process_slip(update: Update, context: ContextTypes.DEFAULT_TYPE, payload: dict):
+    """Process payment slip from Web App."""
+    stg: Settings = context.application.bot_data["settings"]
+    booking_id = payload.get("booking_id")
+    file_b64 = payload.get("file_base64")
+
+    if not booking_id or not file_b64:
+        await update.message.reply_text("ข้อมูลไม่ครบ")
+        return
+
+    dest = stg.slip_storage_dir / f"{booking_id}_{int(datetime.utcnow().timestamp())}.jpg"
+
+    try:
+        await asyncio.to_thread(save_base64_image, file_b64, dest)
+        extraction = await asyncio.to_thread(extract_slip_information, dest, stg)
+        verification = await asyncio.to_thread(verify_booking_amount, stg, booking_id, extraction.amount)
+    except Exception as e:
+        logger.error(f"Slip processing error: {e}")
+        await update.message.reply_text(f"เกิดข้อผิดพลาด: {e}")
+        return
+
+    # Save to DB
+    with sqlite3.connect(stg.database_path) as conn:
+        conn.execute(
+            "INSERT INTO payment_slips (booking_id, slip_image_path, extracted_amount, status, ocr_text) VALUES (?, ?, ?, ?, ?)",
+            (booking_id, str(dest), float(extraction.amount) if extraction.amount else None, verification["status"], extraction.raw_text)
+        )
+
+    if verification["status"] == "verified":
+        await update.message.reply_text(f"✅ ตรวจสอบสลิปสำเร็จ ({booking_id})")
+    else:
+        await update.message.reply_text(f"⚠️ สลิปไม่ตรงกับข้อมูล: {verification['status']}")
+
+
+async def process_expense(update: Update, context: ContextTypes.DEFAULT_TYPE, payload: dict):
+    """Process expense entry from Web App."""
+    stg: Settings = context.application.bot_data["settings"]
+
+    try:
+        amount = Decimal(str(payload["amount"]))
+    except:
+        await update.message.reply_text("จำนวนเงินไม่ถูกต้อง")
+        return
+
+    with sqlite3.connect(stg.database_path) as conn:
+        conn.execute(
+            "INSERT INTO expenses (amount, category, note) VALUES (?, ?, ?)",
+            (float(amount), payload.get("category", "อื่นๆ"), payload.get("note", ""))
+        )
+
+    await update.message.reply_text(f"✅ บันทึกค่าใช้จ่าย {amount} บาท")
+
+
+@app.on_event("startup")
+async def startup():
+    """Initialize bot on startup."""
+    global bot_app
+
+    settings = load_settings()
+    init_db(settings)
+
+    if settings.tesseract_cmd:
+        pytesseract.pytesseract.tesseract_cmd = settings.tesseract_cmd
+
+    genai.configure(api_key=settings.gemini_api_key)
+    gemini_model = genai.GenerativeModel(settings.gemini_model)
+
+    bot_app = Application.builder().token(settings.telegram_bot_token).build()
+    bot_app.bot_data["settings"] = settings
+    bot_app.bot_data["authorized_user_id"] = settings.authorized_user_id
+    bot_app.bot_data["gemini_model"] = gemini_model
+
+    # Add handlers
+    bot_app.add_handler(CommandHandler("start", start))
+    bot_app.add_handler(CommandHandler("help", help_command))
+    bot_app.add_handler(CommandHandler("daily_report", daily_report))
+    bot_app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, echo))
+    bot_app.add_handler(MessageHandler(filters.StatusUpdate.WEB_APP_DATA, web_app_handler))
+
+    await bot_app.initialize()
+    await bot_app.start()
+
+    asyncio.create_task(bot_app.updater.start_polling())
+    logger.info("✅ Bot started in polling mode")
+
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}
+
+
+if __name__ == "__main__":
+    port = int(os.getenv("PORT", "8080")),
+    uvicorn.run(app, host="0.0.0.0", port=port)

app/resources/pms_blueprint.txt

@@ -0,0 +1,12 @@
+ระบบ PMS (Property Management System) ของ Hotel OS ประกอบด้วย:
+
+1. การจองห้อง (Bookings) – id, guest_name, total_due, currency, status
+2. การชำระเงิน – รับสลิป, OCR, ตรวจสอบยอด, บันทึกใน payment_slips
+3. การบันทึกค่าใช้จ่าย (Expenses) – amount, category, note, created_by
+4. รายงานสรุป – จำนวนการจอง, ยอดรวมที่ชำระ, รายจ่ายทั้งหมด
+
+Gemini AI จะอ้างอิงข้อมูลนี้เมื่อตอบคำถามเกี่ยวกับ:
+• สถานะการจอง
+• ยอดค้างชำระ
+• วิธีชำระเงิน
+• รายจ่ายโรงแรม

app/utils.py

@@ -0,0 +1,128 @@
+import base64
+import re
+import sqlite3
+from datetime import datetime, date, time
+from decimal import Decimal, InvalidOperation
+from pathlib import Path
+from typing import Optional, Dict
+from dataclasses import dataclass
+
+import pytesseract
+from PIL import Image
+
+
+@dataclass
+class Settings:
+    telegram_bot_token: str
+    authorized_user_id: int
+    gemini_api_key: str
+    gemini_model: str
+    pms_blueprint_path: Path
+    database_path: Path
+    slip_storage_dir: Path
+    webapp_base_url: str
+    webapp_static_dir: Path
+    webapp_host: str
+    webapp_port: int
+    ocr_language: str
+    tesseract_cmd: Optional[str]
+    amount_tolerance: Decimal
+
+
+@dataclass
+class SlipExtractionResult:
+    raw_text: str
+    amount: Optional[Decimal]
+    payment_date: Optional[date]
+    payment_time: Optional[time]
+
+
+def save_base64_image(b64: str, dest: Path) -> None:
+    """Decode base64-encoded image and save to file."""
+    try:
+        data = base64.b64decode(b64)
+        dest.write_bytes(data)
+    except Exception as exc:
+        raise RuntimeError(f"ไม่สามารถบันทึกรูปภาพได้: {exc}")
+
+
+def extract_slip_information(image_path: Path, settings: Settings) -> SlipExtractionResult:
+    """Use Tesseract OCR to extract payment info from slip."""
+    try:
+        img = Image.open(image_path)
+    except Exception as exc:
+        raise RuntimeError(f"เปิดไฟล์ภาพไม่ได้: {exc}")
+
+    raw_text = pytesseract.image_to_string(img, lang=settings.ocr_language)
+
+    # Extract amount
+    amount: Optional[Decimal] = None
+    amt_match = re.search(r'(\d{1,3}(?:[,\s]\d{3})*(?:\.\d{2})|\d+\.\d{2})', raw_text)
+    if amt_match:
+        amt_raw = amt_match.group(1).replace(',', '').replace(' ', '')
+        try:
+            amount = Decimal(amt_raw)
+        except InvalidOperation:
+            pass
+
+    # Extract date
+    payment_date: Optional[date] = None
+    date_match = re.search(r'(\d{1,2}[\/\-\.]\d{1,2}[\/\-\.]\d{2,4})', raw_text)
+    if date_match:
+        for fmt in ("%d/%m/%Y", "%d-%m-%Y", "%Y-%m-%d", "%d.%m.%Y"):
+            try:
+                payment_date = datetime.strptime(date_match.group(1), fmt).date()
+                break
+            except ValueError:
+                continue
+
+    # Extract time
+    payment_time: Optional[time] = None
+    time_match = re.search(r'(\d{1,2}:\d{2})(?:\s?(AM|PM))?', raw_text, re.IGNORECASE)
+    if time_match:
+        try:
+            t_str = time_match.group(1)
+            ampm = time_match.group(2)
+            if ampm:
+                payment_time = datetime.strptime(f"{t_str} {ampm}", "%I:%M %p").time()
+            else:
+                payment_time = datetime.strptime(t_str, "%H:%M").time()
+        except ValueError:
+            pass
+
+    return SlipExtractionResult(
+        raw_text=raw_text,
+        amount=amount,
+        payment_date=payment_date,
+        payment_time=payment_time,
+    )
+
+
+def verify_booking_amount(
+    settings: Settings,
+    booking_id: str,
+    extracted_amount: Optional[Decimal],
+) -> Dict[str, str]:
+    """Verify extracted amount against booking."""
+    with sqlite3.connect(settings.database_path) as conn:
+        row = conn.execute("SELECT total_due FROM bookings WHERE id = ?", (booking_id,)).fetchone()
+
+    if not row:
+        return {"status": "booking_not_found"}
+
+    expected = Decimal(str(row[0]))
+
+    if extracted_amount is None:
+        return {"status": "amount_missing"}
+
+    if abs(expected - extracted_amount) <= settings.amount_tolerance:
+        return {"status": "verified", "expected_amount": str(expected)}
+    else:
+        return {"status": "amount_mismatch", "expected_amount": str(expected)}
+
+
+def chunk_for_telegram(text: str, limit: int = 4096) -> list[str]:
+    """Split text into Telegram-compatible chunks."""
+    if len(text) <= limit:
+        return [text]
+    return [text[i:i + limit] for i in range(0, len(text), limit)]

requirements.txt

@@ -0,0 +1,6 @@
+python-telegram-bot[fastapi]==20.8
+fastapi==0.115.0
+uvicorn[standard]==0.30.1
+pillow==10.4.0
+pytesseract==0.3.13
+google-generativeai==0.7.2

scripts/backup_db.py

@@ -0,0 +1 @@
+# This script can be used to backup the database.

scripts/notify_pending.py

@@ -0,0 +1 @@
+# This script can be used to notify about pending tasks.

space.yaml

@@ -0,0 +1,2 @@
+runtime:
+  type: docker

start.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+export PORT=${PORT:-8080}
+mkdir -p ./data ./data/slips
+python -m app.main

webapp/app.js

@@ -0,0 +1,41 @@
+window.onload = function () {
+    const tg = window.Telegram?.WebApp;
+    if (!tg) {
+        console.warn('Not in Telegram Web App');
+        return;
+    }
+
+    // Payment Slip Form
+    document.getElementById('slip-form').addEventListener('submit', async (e) => {
+        e.preventDefault();
+        const formData = new FormData(e.target);
+        const file = formData.get('slip_image');
+
+        const reader = new FileReader();
+        reader.onload = () => {
+            const base64 = reader.result.split(',')[1];
+            tg.sendData(JSON.stringify({
+                type: "payment_slip",
+                booking_id: formData.get('booking_id'),
+                reference: formData.get('reference') || "",
+                file_base64: base64,
+                file_name: file.name
+            }));
+            tg.close();
+        };
+        reader.readAsDataURL(file);
+    });
+
+    // Expense Form
+    document.getElementById('expense-form').addEventListener('submit', (e) => {
+        e.preventDefault();
+        const data = new FormData(e.target);
+        tg.sendData(JSON.stringify({
+            type: "expense_entry",
+            amount: data.get('amount'),
+            category: data.get('category') || "อื่นๆ",
+            note: data.get('note') || ""
+        }));
+        tg.close();
+    });
+};

webapp/index.html

@@ -0,0 +1,37 @@
+<!DOCTYPE html>
+<html lang="th">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Hotel OS</title>
+    <link rel="stylesheet" href="style.css">
+    <script src="https://telegram.org/js/telegram-web-app.js"></script>
+</head>
+<body>
+    <h1>🏨 Hotel OS</h1>
+
+    <section>
+        <h2>📄 ส่งสลิปชำระเงิน</h2>
+        <form id="slip-form">
+            <label>Booking ID: <input type="text" name="booking_id" required /></label>
+            <label>Reference: <input type="text" name="reference" /></label>
+            <label>รูปสลิป: <input type="file" accept="image/*" name="slip_image" required /></label>
+            <button type="submit">ส่งสลิป</button>
+        </form>
+    </section>
+
+    <hr>
+
+    <section>
+        <h2>💰 บันทึกค่าใช้จ่าย</h2>
+        <form id="expense-form">
+            <label>จำนวนเงิน: <input type="number" step="0.01" name="amount" required /></label>
+            <label>หมวดหมู่: <input type="text" name="category" placeholder="เช่น ทำความสะอาด" /></label>
+            <label>หมายเหตุ: <input type="text" name="note" /></label>
+            <button type="submit">บันทึก</button>
+        </form>
+    </section>
+
+    <script src="app.js"></script>
+</body>
+</html>

webapp/style.css

@@ -0,0 +1,51 @@
+body {
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+    margin: 20px;
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    color: #fff;
+}
+
+h1 { color: #fff; text-align: center; }
+h2 { color: #f0f0f0; }
+
+section {
+    background: rgba(255,255,255,0.1);
+    backdrop-filter: blur(10px);
+    padding: 20px;
+    border-radius: 15px;
+    margin-bottom: 20px;
+}
+
+label {
+    display: block;
+    margin: 10px 0;
+    font-weight: 500;
+}
+
+input, button {
+    padding: 10px;
+    font-size: 1rem;
+    border-radius: 8px;
+    border: none;
+    width: 100%;
+    box-sizing: border-box;
+}
+
+input {
+    background: rgba(255,255,255,0.9);
+    color: #333;
+}
+
+button {
+    background: #27ae60;
+    color: #fff;
+    cursor: pointer;
+    font-weight: bold;
+    margin-top: 10px;
+    transition: 0.3s;
+}
+
+button:hover {
+    background: #1e8449;
+    transform: scale(1.02);
+}