Update app.py

app.py

@@ -463,49 +463,134 @@ def execute_command(message):
 #         url="https://t.me/python3463_bot",
 #         status_code=status.HTTP_302_FOUND  # 302 is fine too
 #     )
-CACHE_TTL = 60 * 60  # 1 hour
-CACHE = {}
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import HTMLResponse, FileResponse
+import requests, os, time, pathlib
+
+# ──────────────────────────────
+#  Config
+# ──────────────────────────────
+# BOT_TOKEN     = os.getenv("BOT_TOKEN")            # required
+BOT_USERNAME  = os.getenv("BOT_USERNAME", "python3463_bot")
+AVATAR_TTL    = int(os.getenv("AVATAR_TTL", "3600"))  # seconds to keep avatar fresh
+DATA_DIR      = pathlib.Path("/tmp")              # change if you prefer
+AVATAR_PATH   = DATA_DIR / "bot_avatar.jpg"
+AVATAR_META   = {"ts": 0}                         # keeps timestamp of last download
+
+if not TOKEN:
+    raise RuntimeError("TOKEN environment variable not set")
+
+app = FastAPI(title="Telegram Bot Preview")
+
+
+# ──────────────────────────────
+#  Helper: refresh avatar if needed
+# ──────────────────────────────
+def refresh_avatar(force: bool = False) -> None:
+    """
+    Download the bot's avatar if it's missing or stale.
+    Saves it to AVATAR_PATH and updates AVATAR_META['ts'].
+    """
+    now = time.time()
+    if not force and AVATAR_PATH.exists() and now - AVATAR_META["ts"] < AVATAR_TTL:
+        return  # still fresh
 
-BOT_USERNAME = "python3463_bot"
-TME_URL      = f"https://t.me/{BOT_USERNAME}"
+    try:
+        # 1. get bot id
+        me = requests.get(f"https://api.telegram.org/bot{TOKEN}/getMe").json()
+        user_id = me["result"]["id"]
+
+        # 2. get latest profile photo
+        photos = requests.get(
+            f"https://api.telegram.org/bot{TOKEN}/getUserProfilePhotos",
+            params={"user_id": user_id, "limit": 1},
+            timeout=5,
+        ).json()
+
+        if photos["result"]["total_count"] == 0:
+            return  # bot has no avatar
+
+        file_id = photos["result"]["photos"][0][-1]["file_id"]  # biggest size
+
+        # 3. resolve file path
+        file_obj = requests.get(
+            f"https://api.telegram.org/bot{TOKEN}/getFile",
+            params={"file_id": file_id},
+            timeout=5,
+        ).json()
+        file_path = file_obj["result"]["file_path"]
+
+        # 4. download the actual image
+        file_url = f"https://api.telegram.org/file/bot{TOKEN}/{file_path}"
+        img_bytes = requests.get(file_url, timeout=10).content
+
+        DATA_DIR.mkdir(parents=True, exist_ok=True)
+        with open(AVATAR_PATH, "wb") as f:
+            f.write(img_bytes)
+
+        AVATAR_META["ts"] = now
+        print("Avatar refreshed")
+    except Exception as exc:
+        # If something goes wrong, keep the old one
+        print("Avatar refresh failed:", exc)
+
+
+# ──────────────────────────────
+#  Helper: bot description (once)
+# ──────────────────────────────
+def get_bot_description() -> str:
+    resp = requests.get(
+        f"https://api.telegram.org/bot{TOKEN}/getMyDescription"
+    ).json()
+    return resp["result"].get("description", "")
+
+
+# ──────────────────────────────
+#  Startup: fetch avatar once
+# ──────────────────────────────
+@app.on_event("startup")
+def on_startup():
+    refresh_avatar(force=True)
+
+
+# ──────────────────────────────
+#  Route: preview card
+# ──────────────────────────────
+@app.get("/", include_in_schema=False, response_class=HTMLResponse)
+def preview():
+    refresh_avatar()  # refresh if stale
+
+    avatar_src = "/avatar.jpg" if AVATAR_PATH.exists() else "https://telegram.org/img/t_logo.png"
+    description = get_bot_description()
 
-def get_avatar_url() -> str | None:
-    now = time.time()
-    if "photo" in CACHE and now - CACHE["ts"] < CACHE_TTL:
-        return CACHE["photo"]
-
-    html = requests.get(TME_URL, timeout=5).text
-    m = re.search(r'<meta property="og:image" content="([^"]+)"', html)
-    if m:
-        CACHE["photo"] = m.group(1)
-        CACHE["ts"] = now
-        return CACHE["photo"]
-    return None
-
-app = FastAPI()
-
-@app.get("/", include_in_schema=False)
-def root():
-    photo = get_avatar_url() or "/static/telegram.png"
     html = f"""
     <html>
       <head>
-        <title>Python Bot</title>
+        <title>{BOT_USERNAME}</title>
         <style>
-          body{{font-family:sans-serif;display:flex;justify-content:center;}}
-          .card{{max-width:420px;padding:24px;border-radius:12px;
-                 box-shadow:0 4px 12px rgba(0,0,0,.1);text-align:center;}}
-          img{{border-radius:50%;width:120px;height:120px;object-fit:cover;}}
-          a.btn{{background:#2AABEE;color:#fff;padding:12px 24px;border-radius:8px;
-                text-decoration:none;font-weight:bold;display:inline-block;margin-top:12px;}}
+          body {{
+            font-family: sans-serif; display: flex; justify-content: center;
+            padding: 40px; background: #f7f7f7;
+          }}
+          .card {{
+            max-width: 420px; background: #fff; padding: 24px; text-align: center;
+            border-radius: 12px; box-shadow: 0 4px 12px rgba(0,0,0,.1);
+          }}
+          .avatar {{
+            width: 120px; height: 120px; border-radius: 50%; object-fit: cover;
+          }}
+          .btn {{
+            display: inline-block; margin-top: 16px; padding: 12px 24px;
+            background: #2AABEE; color: #fff; border-radius: 8px;
+            text-decoration: none; font-weight: bold;
+          }}
         </style>
       </head>
       <body>
         <div class="card">
-          <img src="{photo}" alt="Bot Avatar">
-          <h2>Python Bot</h2>
-          <p><strong>@{BOT_USERNAME}</strong></p>
-          <p>The fastest way to practise Python 3.</p>
+          <img src="{avatar_src}" alt="avatar" class="avatar">
+          <h2>@{BOT_USERNAME}</h2>
+          <p>{description}</p>
           <a class="btn" href="https://t.me/{BOT_USERNAME}" target="_blank">Start Bot</a>
         </div>
       </body>
@@ -513,6 +598,18 @@ def root():
     """
     return HTMLResponse(html)
 
+
+# ──────────────────────────────
+#  Route: serve avatar file
+# ──────────────────────────────
+@app.get("/avatar.jpg", include_in_schema=False)
+def avatar():
+    refresh_avatar()  # make sure it's fresh
+    if not AVATAR_PATH.exists():
+        raise HTTPException(status_code=404, detail="Avatar not found")
+    return FileResponse(AVATAR_PATH, media_type="image/jpeg")
+
+
 @app.on_event("startup")
 def startup():
     # Launch the bot *after* Uvicorn has started