Update bot.py

bot.py

@@ -229,282 +229,16 @@ def setup_handlers(client):
             except Exception as e:
                 await conv.send_message(f"❌ **Error:** `{str(e)}`")
             finally:
-                try:
-                    await app.disconnect()
-                except:
-                    pass
+                if 'app' in locals():
+                    try: await app.disconnect()
+                    except: pass
 
     async def _load_words(event, is_callback=True):
-        all_lines = await asyncio.to_thread(read_file_sync)
-        if not all_lines: 
-            return await event.edit("Empty file.")
-            
-        CHUNK = 200000
-        current = all_lines[:CHUNK]
-        remaining = all_lines[CHUNK:]
-        
-        valid = await asyncio.to_thread(filter_words_sync, current)
-        r = await db.get_redis()
-        done = await r.smembers("frag:done")
-        to_q = list(valid - done)
-        
-        if to_q:
-            pipe = r.pipeline()
-            for i in range(0, len(to_q), 1000): 
-                pipe.lpush("frag:queue", *to_q[i:i+1000])
-            await pipe.execute()
-            
-        await asyncio.to_thread(write_file_sync, remaining)
-        await db.set_state(total=len(valid))
-        await event.edit(f"✅ Loaded {len(to_q):,} words.", buttons=get_main_menu())
-
-    @client.on(events.CallbackQuery())
-    async def button_handler(event):
-        if not is_admin(event): return
-        try: await event.answer()
-        except: pass
-        data = event.data.decode('utf-8')
-
-        if data == "back_main": 
-            await event.edit("**Fragment Control System**", buttons=get_main_menu())
-import os
-import re
-import time
-import io
-import asyncio
-from telethon import events, Button
-from PIL import Image, ImageDraw, ImageFont
-import db
-import scraper
-
-# --- AUTHENTICATION ---
-def get_admins():
-    return {int(x) for x in os.environ.get("ADMIN_IDS", "").split(",") if x}
-
-def is_admin(event):
-    return event.sender_id in get_admins()
-
-# --- PROFESSIONAL NAVIGATION ---
-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")]
-    ]
-
-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")]
-    ]
-
-# --- FRAGMENT-STYLE IMAGE GENERATOR ---
-def generate_fragment_card(username, status):
-    width, height = 1000, 450
-    bg_color = (17, 24, 39)        
-    card_bg = (31, 41, 55)        
-    text_white = (255, 255, 255)
-    text_gray = (156, 163, 175)
-    text_cyan = (56, 189, 248)    
-    divider_color = (55, 65, 81)
-    
-    status_map = {
-        "TAKEN": {"bg": (69, 39, 0), "text": (251, 191, 36), "desc": "Someone already claimed this username on Telegram. You can make an offer to the owner."},
-        "AVAILABLE": {"bg": (6, 78, 59), "text": (52, 211, 153), "desc": "This username is currently available! You can secure it immediately on Fragment."},
-        "FOR_SALE": {"bg": (30, 58, 138), "text": (147, 197, 253), "desc": "This username is listed for sale. You can purchase it using TON."},
-        "ON_AUCTION": {"bg": (30, 58, 138), "text": (147, 197, 253), "desc": "This username is currently in an active auction. Place your bid now."},
-        "SOLD": {"bg": (31, 41, 55), "text": (209, 213, 219), "desc": "This username was recently sold. It is currently locked to a new owner."},
-        "UNAVAILABLE": {"bg": (64, 64, 64), "text": (212, 212, 212), "desc": "This username is reserved or prohibited by Telegram and cannot be claimed."}
-    }
-    
-    cfg = status_map.get(status.upper(), status_map["UNAVAILABLE"])
-    img = Image.new('RGB', (width, height), color=bg_color)
-    draw = ImageDraw.Draw(img)
-
-    try:
-        font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
-        font_main = ImageFont.truetype(font_path, 48)
-        font_pill = ImageFont.truetype(font_path, 22)
-        font_card_head = ImageFont.truetype(font_path, 26)
-        font_card_body = ImageFont.truetype(font_path, 18)
-    except:
-        font_main = font_pill = font_card_head = font_card_body = ImageFont.load_default()
-
-    # Header
-    header_text = f"{username}.t.me"
-    draw.text((40, 40), header_text, fill=text_white, font=font_main)
-    tw = draw.textlength(header_text, font=font_main)
-    pill_x = 40 + tw + 25
-    draw.rounded_rectangle([pill_x, 52, pill_x + 135, 92], radius=8, fill=cfg["bg"])
-    draw.text((pill_x + 22, 59), status.replace("_", " ").capitalize(), fill=cfg["text"], font=font_pill)
-
-    # Left Card
-    draw.rounded_rectangle([40, 150, 480, 400], radius=10, fill=card_bg)
-    draw.text((105, 190), "Can I buy this username?", fill=text_white, font=font_card_head)
-    draw.line([40, 235, 480, 235], fill=divider_color, width=1)
-    draw.multiline_text((260, 315), cfg["desc"], fill=text_gray, font=font_card_body, align="center", anchor="mm", spacing=8)
-
-    # Right Card
-    draw.rounded_rectangle([510, 150, 960, 400], radius=10, fill=card_bg)
-    rows = [("Telegram Username", f"@{username}"), ("Web Address", f"t.me/{username}"), ("TON Web 3.0", f"{username}.t.me")]
-    y_pos = 185
-    for i, (label, value) in enumerate(rows):
-        draw.text((540, y_pos), label, fill=text_white, font=font_card_head)
-        draw.text((930, y_pos), value, fill=text_cyan, font=font_card_head, anchor="ra")
-        if i < 2:
-            y_div = y_pos + 65
-            draw.line([510, y_div, 960, y_div], fill=divider_color, width=1)
-            y_pos += 85
-
-    buf = io.BytesIO()
-    img.save(buf, format='PNG')
-    buf.seek(0)
-    buf.name = f"{username}.png"
-    return buf
-
-# --- SYSTEM DASHBOARD ---
-async def generate_status_msg():
-    state = await db.get_state()
-    counts = await db.get_counts()
-    concurrency = await db.get_concurrency()
-    qlen = await db.queue_size()
-    
-    total = int(state.get("total", 0))
-    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"
-    
-    bar_len = 20
-    filled = int(bar_len * (processed / total)) if total > 0 else 0
-    bar = "=" * filled + "-" * (bar_len - filled)
-    
-    return (
-        f"**SYSTEM DASHBOARD**\n━━━━━━━━━━━━━━━━━━━━━\n"
-        f"**State:** `{run_str}`\n**Queue:** `{qlen:,}` waiting\n"
-        f"**Speed:** `{concurrency}` workers\n**Progress:** `{pct:.2f}%`\n"
-        f"`[{bar}]` `{processed:,} / {total:,}`\n\n"
-        f"**METRICS**\n━━━━━━━━━━━━━━━━━━━━━\n"
-        f"Taken: `{counts.get('taken', 0):,}` | Unavail: `{counts.get('unavailable', 0):,}`\n"
-        f"Sale: `{counts.get('forsale', 0):,}` | Auction: `{counts.get('auction', 0):,}`\n"
-        f"Sold: `{counts.get('sold', 0):,}`"
-    )
-
-# --- BACKGROUND HELPERS ---
-def read_file_sync():
-    with open("words.txt", "r", encoding="utf-8") as f: 
-        return [line.strip().lower() for line in f if line.strip()]
-
-def write_file_sync(lines):
-    with open("words.txt", "w", encoding="utf-8") as f: 
-        f.write("\n".join(lines))
-
-def filter_words_sync(chunk):
-    return {w for w in chunk if re.match(r'^[a-z0-9_]{4,32}$', w)}
-
-# --- HANDLER SETUP ---
-def setup_handlers(client):
-    
-    @client.on(events.NewMessage(pattern='/start'))
-    async def start_cmd(event):
-        if is_admin(event): 
-            await event.respond("**Fragment Control System**", buttons=get_main_menu())
-
-    @client.on(events.NewMessage(pattern=r'/check\s+([a-zA-Z0-9_]+)'))
-    async def check_cmd(event):
-        if not is_admin(event): return
-        target = event.pattern_match.group(1).lower()
-        msg = await event.respond(f"🔍 Analyzing `@{target}`...")
-        
         try:
-            status = await scraper.check_fragment(target)
-            if not status: status = "UNAVAILABLE"
+            all_lines = await asyncio.to_thread(read_file_sync)
         except Exception:
-            status = "ERROR"
+            return await event.edit("❌ `words.txt` not found. Upload it first.")
             
-        img = await asyncio.to_thread(generate_fragment_card, target, status)
-        await msg.delete()
-        await client.send_file(event.chat_id, file=img, caption=f"**Report:** `@{target}`")
-
-    @client.on(events.NewMessage(pattern='/upload'))
-    async def upload_cmd(event):
-        if not is_admin(event): return
-        if not event.is_reply: 
-            return await event.respond("Reply to a `.txt` file.")
-        reply = await event.get_reply_message()
-        if not reply.document: 
-            return await event.respond("Not a document.")
-        
-        wait = await event.respond("Downloading...")
-        await reply.download_media(file="words.txt")
-        await wait.edit("✅ Uploaded. Go to Settings -> Load Database.", buttons=get_main_menu())
-
-    @client.on(events.NewMessage(pattern='/getstring'))
-    async def pyrogram_string_cmd(event):
-        if not is_admin(event): return
-        
-        async with client.conversation(event.chat_id, timeout=300) as conv:
-            try:
-                await conv.send_message("🛠 **Pyrogram String Session Generator**\n\nEnter your `API_ID`:")
-                api_id_msg = await conv.get_response()
-                api_id = int(api_id_msg.text.strip())
-
-                await conv.send_message("Enter your `API_HASH`:")
-                api_hash_msg = await conv.get_response()
-                api_hash = api_hash_msg.text.strip()
-
-                await conv.send_message("Enter the **Phone Number** (e.g., +1234567890):")
-                phone_msg = await conv.get_response()
-                phone = phone_msg.text.replace(" ", "").strip()
-
-                await conv.send_message("⏳ Requesting OTP from Telegram...")
-                
-                from pyrogram import Client
-                from pyrogram.errors import SessionPasswordNeeded
-
-                app = Client("temp_session", api_id=api_id, api_hash=api_hash, in_memory=True)
-                await app.connect()
-                
-                sent_code = await app.send_code(phone)
-                
-                await conv.send_message(
-                    "📥 **Code Sent!**\n\n"
-                    "⚠️ **CRITICAL:** You MUST enter the code with spaces between numbers.\n"
-                    "Example: `1 2 3 4 5`"
-                )
-                code_msg = await conv.get_response()
-                otp_code = code_msg.text.replace(" ", "").strip()
-                
-                try:
-                    await app.sign_in(phone, sent_code.phone_code_hash, otp_code)
-                except SessionPasswordNeeded:
-                    await conv.send_message("🔐 **2FA Password Required.**\nEnter your Cloud Password:")
-                    pwd_msg = await conv.get_response()
-                    await app.check_password(pwd_msg.text.strip())
-                
-                string_session = await app.export_session_string()
-                
-                await conv.send_message(
-                    f"✅ **String Session Generated!**\n\n"
-                    f"`{string_session}`\n\n"
-                    f"⚠️ Keep this secret. Anyone with this can access your account."
-                )
-                
-            except asyncio.TimeoutError:
-                await conv.send_message("❌ **Timeout.** You took too long. Run `/getstring` again.")
-            except Exception as e:
-                await conv.send_message(f"❌ **Error:** `{str(e)}`")
-            finally:
-                try:
-                    await app.disconnect()
-                except:
-                    pass
-
-    async def _load_words(event, is_callback=True):
-        all_lines = await asyncio.to_thread(read_file_sync)
         if not all_lines: 
             return await event.edit("Empty file.")
             
@@ -559,51 +293,30 @@ def setup_handlers(client):
         elif data.startswith("spd_"):
             await db.set_concurrency(int(data.split("_")[1]))
             await event.edit("✅ Speed Updated", buttons=get_settings_menu())
+            
         elif data == "export_files":
-            await event.edit("⏳ Exporting...")
-            for s in ["taken", "unavailable", "sold", "forsale"]:
-                lst = sorted(list(await (getattr(db, f"get_all_{s}")())))
-                if lst: 
-                    await client.send_file(event.chat_id, io.BytesIO("\n".join(lst).encode()), caption=f"{s}.txt")
-            await event.edit("✅ Done", buttons=get_main_menu())
-        elif data == "reset_confirm": 
-            await event.edit("⚠️ WIPE ALL?", buttons=[[Button.inline("CONFIRM", b"reset_do")], [Button.inline("Cancel", b"menu_settings")]])
-        elif data == "reset_do":
-            await db.flush_all()
-            await event.edit("💣 Purged.", buttons=get_settings_menu())
-        elif data == "menu_settings": 
-            await event.edit("**System Settings**", buttons=get_settings_menu())
-        elif data == "load_words": 
-            await event.edit("⏳ Loading chunk...")
-            await _load_words(event)
-        elif data == "start_scan": 
-            await db.set_state(running="1", paused="0")
-            await event.edit("▶️ Scanner Running", buttons=get_main_menu())
-        elif data == "pause_scan":
-            s = await db.get_state()
-            p = "0" if s.get("paused") == "1" else "1"
-            await db.set_state(paused=p)
-            await event.edit("⏸ Scanner Paused" if p=="1" else "▶️ Resumed", buttons=get_main_menu())
-        elif data == "stop_scan":
-            await db.set_state(running="0", paused="0")
-            await event.edit("⏹ Scanner Stopped", buttons=get_main_menu())
-        elif data in ("show_status", "refresh_status"):
-            await event.edit(await generate_status_msg(), buttons=[[Button.inline("🔄 Refresh", b"refresh_status")], [Button.inline("« Back", b"back_main")]])
-        elif data == "set_speed":
-            await event.edit("Select Speed:", buttons=[[Button.inline("10", b"spd_10"), Button.inline("20", b"spd_20")], [Button.inline("30", b"spd_30"), Button.inline("50", b"spd_50")], [Button.inline("« Back", b"menu_settings")]])
-        elif data.startswith("spd_"):
-            await db.set_concurrency(int(data.split("_")[1]))
-            await event.edit("✅ Speed Updated", buttons=get_settings_menu())
-        elif data == "export_files":
-            await event.edit("⏳ Exporting...")
-            for s in ["taken", "unavailable", "sold", "forsale"]:
+            await event.edit("⏳ Exporting data... Generating text files.")
+            exported_count = 0
+            
+            # BUG FIX: Added 'auction' to the list so we don't miss auction exports!
+            for s in ["taken", "unavailable", "sold", "forsale", "auction"]:
+                # Fetches the corresponding function dynamically from db.py
                 lst = sorted(list(await (getattr(db, f"get_all_{s}")())))
-                if lst: 
-                    await client.send_file(event.chat_id, io.BytesIO("\n".join(lst).encode()), caption=f"{s}.txt")
-            await event.edit("✅ Done", buttons=get_main_menu())
+                if lst:
+                    # BUG FIX: Added the explicit .name attribute
+                    f = io.BytesIO("\n".join(lst).encode('utf-8'))
+                    f.name = f"{s}.txt"
+                    
+                    await client.send_file(event.chat_id, file=f)
+                    exported_count += 1
+                    
+            if exported_count > 0:
+                await event.edit(f"✅ Export complete! Sent {exported_count} files.", buttons=get_main_menu())
+            else:
+                await event.edit("⚠️ No data to export yet. Start a scan first!", buttons=get_main_menu())
+                
         elif data == "reset_confirm": 
             await event.edit("⚠️ WIPE ALL?", buttons=[[Button.inline("CONFIRM", b"reset_do")], [Button.inline("Cancel", b"menu_settings")]])
         elif data == "reset_do":
             await db.flush_all()
             await event.edit("💣 Purged.", buttons=get_settings_menu())
-