Update bot.py

bot.py

@@ -4,9 +4,9 @@ import time
 import io
 import asyncio
 from telethon import events, Button
-from PIL import Image, ImageDraw, ImageFont
+from PIL import Image, ImageDraw, ImageFont, ImageFilter
 import db
-import scraper # Required for the /check command
+import scraper
 
 def get_admins():
     return {int(x) for x in os.environ.get("ADMIN_IDS", "").split(",") if x}
@@ -14,75 +14,99 @@ def get_admins():
 def is_admin(event):
     return event.sender_id in get_admins()
 
-# --- PROFESSIONAL UI MENUS ---
+# --- PROFESSIONAL UI MENUS (Color-coded with emojis) ---
 def get_main_menu():
     return [
-        [Button.inline("Start Scan", b"start_scan"), Button.inline("Pause", b"pause_scan"), Button.inline("Stop", b"stop_scan")],
-        [Button.inline("Live Status", b"show_status"), Button.inline("Export Data", b"export_files")],
-        [Button.inline("System Settings", b"menu_settings")]
+        [Button.inline("▶️ Start Scan", b"start_scan"), Button.inline("⏸ Pause", b"pause_scan"), Button.inline("⏹ Stop", b"stop_scan")],
+        [Button.inline("📊 Live Status", b"show_status"), Button.inline("📥 Export Data", b"export_files")],
+        [Button.inline("⚙️ System Settings", b"menu_settings")]
     ]
 
 def get_settings_menu():
     return [
-        [Button.inline("Load Database", b"load_words")],
-        [Button.inline("Set Speed Limit", b"set_speed")],
-        [Button.inline("Wipe Database", b"reset_confirm")],
-        [Button.inline("« Back to Main Menu", b"back_main")]
+        [Button.inline("💾 Load Database", b"load_words")],
+        [Button.inline("⚡ Set Speed Limit", b"set_speed")],
+        [Button.inline("🔴 Wipe Database", b"reset_confirm")],
+        [Button.inline("« Back to Main Menu", b"back_main")]
     ]
 
-# --- IMAGE GENERATOR (LIQUID GLASS STYLE) ---
-def generate_status_image(username, status):
-    # Base dark theme colors
-    bg_color = (15, 23, 42)       # Slate 900
-    card_color = (30, 41, 59)     # Slate 800 (Glass card)
-    outline_color = (51, 65, 85)  # Slate 700
-    text_primary = (248, 250, 252) # Slate 50
+# --- PRO LIQUID GLASS IMAGE GENERATOR ---
+def generate_glass_image(username, status):
+    width, height = 800, 420
     
-    # Determine status color
-    if status == "TAKEN":
-        status_color = (239, 68, 68)   # Red
-    elif status == "UNAVAILABLE":
-        status_color = (156, 163, 175) # Gray
-    elif status == "FOR_SALE" or status == "ON_AUCTION":
-        status_color = (59, 130, 246)  # Blue
-    else:
-        status_color = (34, 197, 94)   # Green (Available)
-
-    # Create Image
-    img = Image.new('RGB', (800, 400), color=bg_color)
-    d = ImageDraw.Draw(img)
+    # 1. Base vibrant background (Deep Dark Theme)
+    base = Image.new('RGBA', (width, height), (15, 23, 42, 255))
+    draw_base = ImageDraw.Draw(base)
     
-    # Draw "Glass" Card (Rounded Rectangle)
-    d.rounded_rectangle([(50, 50), (750, 350)], radius=20, fill=card_color, outline=outline_color, width=3)
+    # Draw abstract colorful light orbs to simulate background glow
+    draw_base.ellipse([-100, -100, 350, 350], fill=(59, 130, 246, 255))   # Ocean Blue
+    draw_base.ellipse([450, 150, 950, 650], fill=(139, 92, 246, 255))    # Neon Purple
+    draw_base.ellipse([200, -150, 700, 250], fill=(14, 165, 233, 255))   # Cyan
     
-    # Status Indicator Dot
-    d.ellipse([(90, 90), (110, 110)], fill=status_color)
+    # Blur the orbs to create a smooth, glowing ambient background
+    base = base.filter(ImageFilter.GaussianBlur(50))
+    
+    # 2. Create the Frosted Glass Card
+    card_x1, card_y1, card_x2, card_y2 = 60, 60, 740, 360
+    glass = Image.new('RGBA', (width, height), (0, 0, 0, 0))
+    draw_glass = ImageDraw.Draw(glass)
+    
+    # Semi-transparent white card (Alpha 25) with a slightly brighter border
+    draw_glass.rounded_rectangle(
+        [card_x1, card_y1, card_x2, card_y2], 
+        radius=30, 
+        fill=(255, 255, 255, 25), 
+        outline=(255, 255, 255, 90), 
+        width=2
+    )
     
-    # Text Setup (Using default font scaled up via ImageFont if custom isn't available)
+    # Composite the glass card directly over the glowing background
+    img = Image.alpha_composite(base, glass)
+    draw = ImageDraw.Draw(img)
+    
+    # 3. Determine Status Colors for the Badge
+    if status == "TAKEN":
+        status_color = (239, 68, 68, 255)      # Red
+    elif status == "UNAVAILABLE":
+        status_color = (156, 163, 175, 255)    # Gray
+    elif status in ["FOR_SALE", "ON_AUCTION"]:
+        status_color = (59, 130, 246, 255)     # Blue
+    elif status == "SOLD":
+        status_color = (245, 158, 11, 255)     # Orange
+    else:
+        status_color = (34, 197, 94, 255)      # Green
+        
+    # 4. Typography Loading
     try:
-        # Tries to load a standard Linux font, falls back to default
-        font_large = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 60)
-        font_small = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 30)
+        font_large = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 64)
+        font_med = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 32)
+        font_small = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 20)
     except IOError:
-        font_large = ImageFont.load_default()
-        font_small = ImageFont.load_default()
+        font_large = font_med = font_small = ImageFont.load_default()
 
     # Draw Text
-    d.text((130, 75), "FRAGMENT STATUS REPORT", fill=outline_color, font=font_small)
-    d.text((90, 160), f"@{username}", fill=text_primary, font=font_large)
+    draw.text((110, 100), "FRAGMENT INTELLIGENCE REPORT", fill=(226, 232, 240, 200), font=font_small)
+    draw.text((105, 140), f"@{username}", fill=(255, 255, 255, 255), font=font_large)
+    
+    # Draw Status Pill / Badge
+    # Measure text to dynamically size the pill
+    text_bbox = draw.textbbox((0, 0), status, font=font_med)
+    text_width = text_bbox[2] - text_bbox[0]
+    badge_w = text_width + 60
+    badge_h = 56
     
-    # Draw Status Pill
-    d.rounded_rectangle([(90, 260), (350, 310)], radius=25, fill=bg_color, outline=status_color, width=2)
-    d.text((120, 265), status, fill=status_color, font=font_small)
+    # Render pill background and text
+    draw.rounded_rectangle([110, 250, 110 + badge_w, 250 + badge_h], radius=28, fill=status_color)
+    draw.text((140, 258), status, fill=(255, 255, 255, 255), font=font_med)
 
-    # Save to buffer
+    # Save to memory buffer
     buf = io.BytesIO()
-    img.save(buf, format='PNG')
+    img.convert('RGB').save(buf, format='PNG')
     buf.seek(0)
     buf.name = f"{username}_status.png"
     return buf
 
-# --- MODERN STATUS DASHBOARD ---
+# --- ENCODING-SAFE MODERN DASHBOARD ---
 async def generate_status_msg():
     state = await db.get_state()
     counts = await db.get_counts()
@@ -93,16 +117,17 @@ async def generate_status_msg():
     processed = int(state.get("processed", 0))
     pct = (processed / total * 100) if total > 0 else 0
     
-    run_str = "RUNNING" if state.get("running") == "1" else "STOPPED"
-    if state.get("paused") == "1": run_str = "PAUSED"
+    run_str = "🟢 RUNNING" if state.get("running") == "1" else "⏹ STOPPED"
+    if state.get("paused") == "1": run_str = "⏸ PAUSED"
     
+    # Clean ASCII Progress bar (No weird encoding issues)
     bar_length = 20
     filled = int(bar_length * (processed / total)) if total > 0 else 0
-    bar = "â–ˆ" * filled + "â–‘" * (bar_length - filled)
+    bar = "=" * filled + "-" * (bar_length - filled)
     
     msg = (
         f"**SYSTEM DASHBOARD**\n"
-        f"────────────────────────\n"
+        f"━━━━━━━━━━━━━━━━━━━\n"
         f"**State:** `{run_str}`\n"
         f"**Queue:** `{qlen:,}` waiting\n"
         f"**Speed:** `{concurrency}` workers\n"
@@ -110,7 +135,7 @@ async def generate_status_msg():
         f"`[{bar}]`\n"
         f"`{processed:,} / {total:,}`\n\n"
         f"**DATABASE METRICS**\n"
-        f"────────────────────────\n"
+        f"━━━━━━━━━━━━━━━━━━━\n"
         f"Taken      : `{counts.get('taken', 0):,}`\n"
         f"Unavailable: `{counts.get('unavailable', 0):,}`\n"
         f"For Sale   : `{counts.get('forsale', 0):,}`\n"
@@ -138,48 +163,48 @@ def setup_handlers(client):
         if not is_admin(event): return
         await event.respond("**Fragment Control System**\nSelect an operation:", buttons=get_main_menu())
 
-    # --- NEW MANUAL CHECK FEATURE ---
+    # --- ADVANCED MANUAL CHECK FEATURE ---
     @client.on(events.NewMessage(pattern=r'/check\s+([a-zA-Z0-9_]+)'))
     async def check_cmd(event):
         if not is_admin(event): return
         username = event.pattern_match.group(1).lower()
         
-        msg = await event.respond(f"Scanning `{username}`...")
+        msg = await event.respond(f"🔍 Executing deep scan on `@{username}`...")
         
-        # Scrape data
         try:
+            # Uses the exact same robust scraper your queue workers use
             status = await scraper.check_fragment(username)
             if not status:
                 status = "UNKNOWN"
         except Exception:
             status = "ERROR"
 
-        # Generate sleek image
-        img_buffer = await asyncio.to_thread(generate_status_image, username, status)
+        # Generate the Liquid Glass UI Card
+        img_buffer = await asyncio.to_thread(generate_glass_image, username, status)
         
         # Send result
         await msg.delete()
         await client.send_file(
             event.chat_id, 
             file=img_buffer, 
-            caption=f"**Target:** `@{username}`\n**Status:** `{status}`"
+            caption=f"**Target:** `@{username}`\n**Real-time Status:** `{status}`"
         )
 
     @client.on(events.NewMessage(pattern='/upload'))
     async def upload_cmd(event):
         if not is_admin(event): return
         if not event.is_reply:
-            await event.respond("Reply to a `.txt` file with `/upload`.")
+            await event.respond("⚠️ Reply to a `.txt` file with `/upload`.")
             return
 
         reply_msg = await event.get_reply_message()
         if not reply_msg.document or reply_msg.file.ext != '.txt':
-            await event.respond("Target must be a `.txt` document.")
+            await event.respond("⚠️ Target must be a `.txt` document.")
             return
 
-        msg = await event.respond("Transferring to secure server storage...")
+        msg = await event.respond("📥 Transferring to secure server storage...")
         await reply_msg.download_media(file="words.txt")
-        await msg.edit("Transfer complete. Open **System Settings** -> **Load Database** to initialize.", buttons=get_main_menu())
+        await msg.edit("✅ Transfer complete. Open **System Settings** -> **Load Database** to initialize.", buttons=get_main_menu())
 
     async def _load_words(event, is_callback=True):
         async def update_msg(text, buttons=None):
@@ -191,11 +216,11 @@ def setup_handlers(client):
         try:
             all_lines = await asyncio.to_thread(read_file_sync)
         except Exception:
-            await update_msg("`words.txt` missing. Please use `/upload`.")
+            await update_msg("❌ `words.txt` missing. Please use `/upload`.")
             return
 
         if not all_lines:
-            await update_msg("Database is empty. Upload a new master list.")
+            await update_msg("⚠️ Database is empty. Upload a new master list.")
             return
 
         CHUNK_SIZE = 200000
@@ -216,14 +241,14 @@ def setup_handlers(client):
                 await pipe.execute()
             except Exception as e:
                 if "oom" in str(e).lower() or "maxmemory" in str(e).lower():
-                    await update_msg("Memory limit exceeded. Please Wipe Database first.")
+                    await update_msg("❌ Memory limit exceeded. Please Wipe Database first.")
                 return
         
         await asyncio.to_thread(write_file_sync, remaining_lines)
         await db.set_state(total=len(valid_words))
         skipped = len(valid_words) - len(to_queue)
         
-        msg = f"**Database Loaded**\n• Added: `{len(to_queue):,}`\n• Skipped: `{skipped:,}`\n• Disk Remaining: `{len(remaining_lines):,}`"
+        msg = f"**Database Loaded**\n• Added: `{len(to_queue):,}`\n• Skipped: `{skipped:,}`\n• Disk Remaining: `{len(remaining_lines):,}`"
         await update_msg(msg, buttons=get_main_menu())
 
     @client.on(events.CallbackQuery())
@@ -234,45 +259,43 @@ def setup_handlers(client):
 
         data = event.data.decode('utf-8')
 
-        # Menu Navigation
         if data == "back_main":
             await event.edit("**Fragment Control System**\nSelect an operation:", buttons=get_main_menu())
         elif data == "menu_settings":
             await event.edit("**System Settings**\nConfigure scanner variables:", buttons=get_settings_menu())
 
-        # Logic
         elif data == "load_words":
-            await event.edit("Processing chunk logic in background...")
+            await event.edit("⏳ Processing chunk logic in background...")
             await _load_words(event, is_callback=True)
             
         elif data == "start_scan":
             await db.set_state(running="1", paused="0", start_time=str(time.time()))
-            await event.edit("Scanner initialized and running.", buttons=get_main_menu())
+            await event.edit("▶️ Scanner initialized and running.", buttons=get_main_menu())
             
         elif data == "pause_scan":
             state = await db.get_state()
             if state.get("paused") == "1":
                 await db.set_state(paused="0")
-                await event.edit("Scanner resumed.", buttons=get_main_menu())
+                await event.edit("▶️ Scanner resumed.", buttons=get_main_menu())
             else:
                 await db.set_state(paused="1")
-                await event.edit("Scanner paused.", buttons=get_main_menu())
+                await event.edit("⏸ Scanner paused.", buttons=get_main_menu())
                 
         elif data == "stop_scan":
             await db.set_state(running="0", paused="0")
-            await event.edit("Scanner terminated.", buttons=get_main_menu())
+            await event.edit("⏹ Scanner terminated.", buttons=get_main_menu())
             
         elif data in ("show_status", "refresh_status"):
             msg = await generate_status_msg()
-            buttons = [[Button.inline("Refresh Data", b"refresh_status")], [Button.inline("« Back", b"back_main")]]
+            buttons = [[Button.inline("🔄 Refresh Data", b"refresh_status")], [Button.inline("« Back", b"back_main")]]
             try: await event.edit(msg, buttons=buttons)
             except Exception: pass 
             
         elif data == "set_speed":
             buttons = [
-                [Button.inline("Slow (10)", b"spd_10"), Button.inline("Normal (20)", b"spd_20")],
-                [Button.inline("Fast (30)", b"spd_30"), Button.inline("Max (50)", b"spd_50")],
-                [Button.inline("« Back", b"menu_settings")]
+                [Button.inline("🐢 Slow (10)", b"spd_10"), Button.inline("🚶 Normal (20)", b"spd_20")],
+                [Button.inline("🚗 Fast (30)", b"spd_30"), Button.inline("🚀 Max (50)", b"spd_50")],
+                [Button.inline("« Back", b"menu_settings")]
             ]
             await event.edit("**Thread Allocation**\nSelect maximum concurrent workers:", buttons=buttons)
             
@@ -280,12 +303,12 @@ def setup_handlers(client):
             try:
                 speed_limit = int(data.split("_")[1])
                 await db.set_concurrency(speed_limit)
-                await event.edit(f"Thread limit updated to `{speed_limit}`.", buttons=get_settings_menu())
+                await event.edit(f"✅ Thread limit updated to `{speed_limit}`.", buttons=get_settings_menu())
             except Exception:
                 pass
             
         elif data == "export_files":
-            await event.edit("Generating export packages...")
+            await event.edit("⏳ Generating export packages...")
             
             taken = sorted(await db.get_all_taken())
             unavail = sorted(await db.get_all_unavailable())
@@ -299,13 +322,13 @@ def setup_handlers(client):
             if sold: await client.send_file(event.chat_id, file=io.BytesIO("\n".join(sold).encode('utf-8')), caption="sold.txt")
             if forsale: await client.send_file(event.chat_id, file=io.BytesIO("\n".join(forsale).encode('utf-8')), caption="forsale.txt")
             
-            await event.edit("Export sequence complete.", buttons=get_main_menu())
+            await event.edit("✅ Export sequence complete.", buttons=get_main_menu())
             
         elif data == "reset_confirm":
-            buttons = [[Button.inline("CONFIRM WIPE", b"reset_do")], [Button.inline("Cancel", b"menu_settings")]]
+            buttons = [[Button.inline("⚠️ CONFIRM WIPE", b"reset_do")], [Button.inline("Cancel", b"menu_settings")]]
             await event.edit("**AUTHORIZATION REQUIRED**\n\nThis will purge the active Redis database. Confirm?", buttons=buttons)
             
         elif data == "reset_do":
-            await event.edit("Purging memory...")
+            await event.edit("⏳ Purging memory...")
             await db.flush_all()
-            await event.edit("Database wiped. Awaiting new chunk allocation.", buttons=get_settings_menu())
+            await event.edit("💣 Database wiped. Awaiting new chunk allocation.", buttons=get_settings_menu())