Spaces:

thehnx
/

PepBielsaBot

Running

App Files Files Community

thehnx commited on Jun 11

Commit

e2c6c50

1 Parent(s): da49547

feat: add HF Spaces deployment (Dockerfile + /health endpoint for UptimeRobot)

Browse files

Files changed (4) hide show

Dockerfile +46 -0
README.md +45 -0
main.py +114 -1
requirements.txt +1 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,46 @@

+# ─────────────────────────────────────────────────────────────────────────────
+# PepBielsa Monitor Bot — Dockerfile for Hugging Face Spaces
+# ─────────────────────────────────────────────────────────────────────────────
+# HF Spaces runs Docker containers and exposes port 7860 to the web.
+# UptimeRobot pings https://<space>.hf.space/health every 5 min to keep it alive.
+# ─────────────────────────────────────────────────────────────────────────────
+FROM python:3.11-slim
+# Prevents Python from writing .pyc files and enables stdout/stderr logging.
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    PORT=7860
+WORKDIR /app
+# Install OS-level dependencies first (cached layer).
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        gcc \
+        libffi-dev \
+    && rm -rf /var/lib/apt/lists/*
+# Install Python dependencies (cached unless requirements.txt changes).
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+# Copy application source.
+COPY . .
+# Create directories the bot writes to at runtime.
+RUN mkdir -p temp_media
+# HF Spaces runs containers as a non-root user by default.
+# Create a dedicated user for security best-practices.
+RUN useradd -m -u 1000 botuser \
+    && chown -R botuser:botuser /app
+USER botuser
+# HF Spaces expects the container to listen on port 7860.
+EXPOSE 7860
+# Health-check so Docker/HF knows the service is alive.
+HEALTHCHECK --interval=30s --timeout=10s --start-period=15s --retries=3 \
+    CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:7860/health')" || exit 1
+CMD ["python", "main.py"]

README.md ADDED Viewed

	@@ -0,0 +1,45 @@

+---
+title: PepBielsa Monitor Bot
+emoji: ⚽
+colorFrom: green
+colorTo: blue
+sdk: docker
+app_port: 7860
+pinned: false
+license: mit
+---
+# ⚽ PepBielsa Monitor Bot
+A Telegram channel monitor and forwarder with AI-powered deduplication (Groq llama-3.1-8b-instant).
+## Health Check
+A lightweight web server runs alongside the bot at `/health` — used by UptimeRobot to prevent the Space from sleeping.
+```
+GET https://<your-space>.hf.space/health
+```
+## Environment Variables
+Set these as **Secrets** in your HF Space settings (`Settings → Repository secrets`):
+| Variable | Description |
+|----------|-------------|
+| `API_ID` | Telegram API ID |
+| `API_HASH` | Telegram API Hash |
+| `PHONE_NUMBER` | Your Telegram phone number |
+| `SESSION_STRING` | Generated by `session_export.py` |
+| `SOURCE_CHANNELS` | Comma-separated channel usernames |
+| `DESTINATION_CHANNEL` | Target channel username |
+| `GROQ_API_KEY` | Groq API key (free at console.groq.com) |
+| `HF_TOKEN` | HF write token (for persistent store) |
+| `HF_DATASET_REPO` | e.g. `yourname/pepbielsa-processed` |
+## UptimeRobot Setup
+1. Create a **HTTP(s)** monitor
+2. URL: `https://<your-space-name>.hf.space/health`
+3. Monitoring interval: **5 minutes**
+4. Keyword monitor (optional): set keyword to `Online`

main.py CHANGED Viewed

@@ -44,6 +44,12 @@ import time
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple
 from telethon import TelegramClient, events
 from telethon.errors import (
     ChannelPrivateError,
@@ -840,6 +846,94 @@ class TelegramMonitor:
         # DB is closed only on final shutdown (not between reconnects).
 # ═════════════════════════════════════════════════════════════════════════════
 # Entry-point
 # ═════════════════════════════════════════════════════════════════════════════
@@ -869,8 +963,27 @@ def main() -> None:
     monitor = TelegramMonitor(cfg)
     try:
-        asyncio.run(monitor.run_forever())
     except KeyboardInterrupt:
         pass
     finally:

 from pathlib import Path
 from typing import Dict, List, Optional, Tuple
+try:
+    from aiohttp import web as aiohttp_web
+    _AIOHTTP_AVAILABLE = True
+except ImportError:
+    _AIOHTTP_AVAILABLE = False
 from telethon import TelegramClient, events
 from telethon.errors import (
     ChannelPrivateError,
         # DB is closed only on final shutdown (not between reconnects).
+# ═════════════════════════════════════════════════════════════════════════════
+# Health-check web server (for UptimeRobot / HF Spaces keep-alive)
+# ═════════════════════════════════════════════════════════════════════════════
+_HEALTH_HTML = """\
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>PepBielsa Bot — Status</title>
+  <style>
+    *{{margin:0;padding:0;box-sizing:border-box}}
+    body{{font-family:-apple-system,BlinkMacSystemFont,'Segoe UI',Roboto,sans-serif;
+          background:#0f172a;color:#e2e8f0;display:flex;align-items:center;
+          justify-content:center;min-height:100vh}}
+    .card{{background:#1e293b;border:1px solid #334155;border-radius:16px;
+           padding:40px 48px;text-align:center;max-width:420px;width:90%}}
+    .dot{{width:14px;height:14px;background:#22c55e;border-radius:50%;
+          display:inline-block;margin-right:8px;
+          box-shadow:0 0 0 4px rgba(34,197,94,.25);animation:pulse 2s infinite}}
+    @keyframes pulse{{0%,100%{{box-shadow:0 0 0 4px rgba(34,197,94,.25)}}
+                      50%{{box-shadow:0 0 0 8px rgba(34,197,94,.1)}}}}
+    h1{{font-size:1.6rem;font-weight:700;margin-bottom:8px}}
+    p{{color:#94a3b8;font-size:.9rem;margin-top:12px}}
+    .badge{{display:inline-block;background:#22c55e22;color:#4ade80;
+            border:1px solid #22c55e55;border-radius:999px;
+            padding:4px 14px;font-size:.8rem;margin-top:20px}}
+  </style>
+</head>
+<body>
+  <div class="card">
+    <h1><span class="dot"></span>PepBielsa Bot</h1>
+    <p>Telegram channel monitor is active and forwarding.</p>
+    <div class="badge">&#x2713;&nbsp;Online</div>
+  </div>
+</body>
+</html>
+"""
+async def _health_handler(request) -> "aiohttp_web.Response":
+    """Return an HTML health page that satisfies UptimeRobot keyword monitors."""
+    return aiohttp_web.Response(
+        text=_HEALTH_HTML,
+        content_type="text/html",
+        charset="utf-8",
+        headers={
+            # Mimic a real browser server so UptimeRobot treats us as a web page.
+            "Server": "nginx/1.25.3",
+            "X-Content-Type-Options": "nosniff",
+            "Cache-Control": "no-cache, no-store, must-revalidate",
+        },
+    )
+async def run_health_server(port: int = 7860) -> None:
+    """Start the aiohttp health-check server on *port*.
+    HF Spaces expects the app to listen on port 7860 by default.
+    Override via the PORT environment variable.
+    """
+    if not _AIOHTTP_AVAILABLE:
+        log.warning(
+            "aiohttp not installed — health-check server disabled. "
+            "Add 'aiohttp>=3.9.0' to requirements.txt."
+        )
+        return
+    app = aiohttp_web.Application()
+    app.router.add_get("/", _health_handler)
+    app.router.add_get("/health", _health_handler)
+    runner = aiohttp_web.AppRunner(app)
+    await runner.setup()
+    site = aiohttp_web.TCPSite(runner, "0.0.0.0", port)
+    await site.start()
+    log.info("Health-check server listening on http://0.0.0.0:%d/health", port)
+    # Keep running forever (cancelled when the event loop shuts down).
+    try:
+        await asyncio.Event().wait()
+    except asyncio.CancelledError:
+        pass
+    finally:
+        await runner.cleanup()
 # ═════════════════════════════════════════════════════════════════════════════
 # Entry-point
 # ═════════════════════════════════════════════════════════════════════════════
     monitor = TelegramMonitor(cfg)
+    # Determine health-server port (HF Spaces default: 7860).
+    health_port = int(os.getenv("PORT", "7860"))
+    async def _run_all() -> None:
+        """Run the bot and the health-check server concurrently."""
+        bot_task = asyncio.create_task(monitor.run_forever())
+        health_task = asyncio.create_task(run_health_server(health_port))
+        try:
+            await asyncio.gather(bot_task, health_task)
+        except KeyboardInterrupt:
+            pass
+        finally:
+            bot_task.cancel()
+            health_task.cancel()
+            with asyncio.suppress(asyncio.CancelledError):
+                await bot_task
+            with asyncio.suppress(asyncio.CancelledError):
+                await health_task
     try:
+        asyncio.run(_run_all())
     except KeyboardInterrupt:
         pass
     finally:

requirements.txt CHANGED Viewed

@@ -1,5 +1,6 @@
 telethon>=1.36.0
 python-dotenv>=1.0.0
 aiofiles>=23.2.1
 groq>=0.9.0
 huggingface_hub>=0.23.0

 telethon>=1.36.0
 python-dotenv>=1.0.0
 aiofiles>=23.2.1
+aiohttp>=3.9.0
 groq>=0.9.0
 huggingface_hub>=0.23.0