Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import logging
 import nest_asyncio
 import random
@@ -14,12 +15,16 @@ from telethon import TelegramClient, events, Button
 from telethon.sessions import StringSession
 from telethon.tl import functions
 
+# ───── Logging ─────
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s | %(message)s",
+    stream=sys.stdout,
+)
+log = logging.getLogger("app")
+
 # ───── Setup ─────
 nest_asyncio.apply()
-logging.getLogger("telethon").setLevel(logging.CRITICAL)
-logging.getLogger("apscheduler").setLevel(logging.CRITICAL)
-logging.getLogger("werkzeug").setLevel(logging.ERROR)
-
 API_ID = int(os.getenv("API_ID", 3704772))
 API_HASH = os.getenv("API_HASH", "b8e50a035abb851c0dd424e14cac4c06")
 BOT_TOKEN = os.getenv("BOT_TOKEN")
@@ -35,14 +40,15 @@ userbots = {}
 
 scheduler = AsyncIOScheduler(timezone=pytz.timezone("Asia/Kolkata"))
 
-# ───── Reset Daily Created Count ─────
+# ───── Reset Daily Counts ─────
 def reset_daily_counts():
     accounts.update_many({}, {"$set": {"created": 0}})
     asyncio.create_task(bot.send_message(ADMIN_ID, "🔄 Daily group creation count reset."))
+    log.info("🔄 Daily reset of group creation count")
 
 scheduler.add_job(reset_daily_counts, CronTrigger(hour=0, minute=0))
 
-# ───── Schedule Group Creation ─────
+# ───── Schedule Random Group Creation ─────
 def schedule_random_group_creation(userbot, acc_doc):
     hour = random.randint(0, 23)
     minute = random.randint(0, 59)
@@ -54,7 +60,8 @@ def schedule_random_group_creation(userbot, acc_doc):
                 return
             await auto_create_groups(userbot, acc_doc)
         except Exception as e:
-            await bot.send_message(ADMIN_ID, f"❌ Error creating groups for {acc_doc['name']}: {e}")
+            await bot.send_message(ADMIN_ID, f"❌ Error for {acc_doc['name']}: {e}")
+            log.info(f"❌ Group creation error for {acc_doc['name']}: {e}")
 
     scheduler.add_job(job, trigger)
 
@@ -63,18 +70,20 @@ async def auto_create_groups(userbot, acc_doc):
     created = 0
     try:
         for i in range(random.randint(3, 5)):
-            group_title = f"Auto Group {i + 1}"
-            await userbot(functions.messages.CreateChatRequest(users=[], title=group_title))
+            title = f"Auto Group {i + 1}"
+            await userbot(functions.messages.CreateChatRequest(users=[], title=title))
             created += 1
             await asyncio.sleep(1)
         accounts.update_one({"_id": acc_doc["_id"]}, {"$inc": {"created": created}})
         info = await userbot.get_me()
         await bot.send_message(
             ADMIN_ID,
-            f"✅ Created {created} groups\n👤 Name: {info.first_name}\n🔗 Username: @{info.username or 'N/A'}\n📊 Total Created: {acc_doc['created'] + created}"
+            f"✅ Created {created} groups\n👤 {info.first_name}\n🔗 @{info.username or 'N/A'}\n📊 Total: {acc_doc['created'] + created}"
         )
+        log.info(f"👥 {created} groups created for {acc_doc['name']}")
     except Exception as e:
-        await bot.send_message(ADMIN_ID, f"❌ Failed group creation for {acc_doc['name']}: {e}")
+        await bot.send_message(ADMIN_ID, f"❌ Group creation failed: {e}")
+        log.info(f"❌ Group creation failed for {acc_doc['name']}: {e}")
 
 # ───── Start Userbot ─────
 async def start_userbot(session_str, owner_id):
@@ -97,10 +106,12 @@ async def start_userbot(session_str, owner_id):
         userbots[info.id] = userbot
         schedule_random_group_creation(userbot, doc)
         asyncio.create_task(userbot.run_until_disconnected())
-        await bot.send_message(ADMIN_ID, f"✅ New account added: {info.first_name} (@{info.username or 'N/A'})")
+        await bot.send_message(ADMIN_ID, f"✅ Added: {info.first_name} (@{info.username or 'N/A'})")
+        log.info(f"✅ Added account: {info.first_name} (@{info.username or 'N/A'})")
         return info
     except Exception as e:
-        await bot.send_message(ADMIN_ID, f"❌ Failed to add account: {e}")
+        await bot.send_message(ADMIN_ID, f"❌ Failed to add: {e}")
+        log.info(f"❌ Failed to add account: {e}")
         return None
 
 # ───── Bot Commands ─────
@@ -108,14 +119,14 @@ async def start_userbot(session_str, owner_id):
 async def start_cmd(event):
     if event.sender_id != ADMIN_ID:
         return
-    await event.reply("👋 Welcome! Use:\n/addacc — Add string session\n/delacc — Delete account\n/total — Total accounts")
+    await event.reply("👋 Welcome!\n/addacc — Add account\n/delacc — Delete\n/total — Count")
 
 @bot.on(events.NewMessage(pattern="/addacc"))
 async def addacc_cmd(event):
     if event.sender_id != ADMIN_ID:
         return
     async with bot.conversation(event.chat_id, timeout=60) as conv:
-        await conv.send_message("✍️ Send your **String Session**:")
+        await conv.send_message("✍️ Send your String Session:")
         session_msg = await conv.get_response()
         session_str = session_msg.text.strip()
         info = await start_userbot(session_str, event.sender_id)
@@ -129,7 +140,7 @@ async def total_accounts_cmd(event):
     if event.sender_id != ADMIN_ID:
         return
     count = accounts.count_documents({})
-    await event.respond(f"📊 Total connected accounts: **{count}**")
+    await event.respond(f"📊 Total connected: **{count}**")
 
 @bot.on(events.NewMessage(pattern="/delacc"))
 async def delacc_cmd(event):
@@ -137,10 +148,9 @@ async def delacc_cmd(event):
         return
     user_accs = list(accounts.find({"owner_id": ADMIN_ID}))
     if not user_accs:
-        await event.reply("😶 No accounts found.")
+        await event.reply("😶 No accounts.")
         return
-    buttons = []
-    row = []
+    buttons, row = [], []
     for idx, acc in enumerate(user_accs, 1):
         row.append(Button.inline(f"{acc['name']} (@{acc.get('username', 'N/A')})", data=f"del_{acc['user_id']}"))
         if idx % 2 == 0:
@@ -148,7 +158,7 @@ async def delacc_cmd(event):
             row = []
     if row:
         buttons.append(row)
-    await event.respond("Select account to delete:", buttons=buttons)
+    await event.respond("Select to delete:", buttons=buttons)
 
 @bot.on(events.CallbackQuery(pattern=b"del_"))
 async def handle_delete(event):
@@ -157,9 +167,9 @@ async def handle_delete(event):
     user_id = int(event.data.decode().split("_")[1])
     acc_doc = accounts.find_one({"user_id": user_id})
     if not acc_doc:
-        await event.answer("Account not found.", alert=True)
+        await event.answer("Not found.", alert=True)
         return
-    await event.edit(f"⚠️ Confirm deletion of {acc_doc['name']}?", buttons=[
+    await event.edit(f"⚠️ Confirm delete {acc_doc['name']}?", buttons=[
         [Button.inline("✅ Confirm", data=f"confdel_{user_id}"), Button.inline("❌ Cancel", data=b"cancel")]
     ])
 
@@ -174,10 +184,11 @@ async def confirm_delete(event):
             if user_id in userbots:
                 await userbots[user_id].disconnect()
                 del userbots[user_id]
-            await event.edit(f"✅ Deleted account: {acc_doc['name']}")
-            await bot.send_message(ADMIN_ID, f"🗑️ Account deleted: {acc_doc['name']} (@{acc_doc.get('username', 'N/A')})")
+            await event.edit(f"✅ Deleted: {acc_doc['name']}")
+            await bot.send_message(ADMIN_ID, f"🗑️ Deleted: {acc_doc['name']} (@{acc_doc.get('username', 'N/A')})")
+            log.info(f"🗑️ Deleted account: {acc_doc['name']} (@{acc_doc.get('username', 'N/A')})")
         except Exception as e:
-            await event.edit(f"Deleted from DB, but error in stopping: {e}")
+            await event.edit(f"Deleted from DB, error stopping: {e}")
     else:
         await event.edit("⚠️ Not found.")
 
@@ -187,8 +198,9 @@ async def cancel_cb(event):
         return
     await event.edit("❌ Cancelled.")
 
-# ───── Load Existing Userbots ─────
+# ───── Load All Bots ─────
 async def load_all_userbots():
+    log.info("🔄 Loading all stored userbots...")
     for acc in accounts.find():
         try:
             ub = TelegramClient(StringSession(acc["session"]), API_ID, API_HASH)
@@ -196,32 +208,33 @@ async def load_all_userbots():
             userbots[acc["user_id"]] = ub
             schedule_random_group_creation(ub, acc)
             asyncio.create_task(ub.run_until_disconnected())
-        except Exception:
-            await bot.send_message(ADMIN_ID, f"❌ Failed to start account {acc['name']}")
+            log.info(f"✅ Loaded: {acc['name']} (@{acc.get('username', 'N/A')})")
+        except Exception as e:
+            await bot.send_message(ADMIN_ID, f"❌ Failed to start {acc['name']}")
+            log.info(f"❌ Failed to start {acc['name']}: {e}")
 
 # ───── Flask App ─────
 app = Flask(__name__)
-cli = type('cli', (), {'show_server_banner': lambda *a, **kw: None})
-
 @app.route("/")
 def home():
     return "✅ Channel-making bot is running."
 
 def flask_thread():
-    app.run(host="0.0.0.0", port=7680,use_reloader=False, use_debugger=False)
+    app.run(host="0.0.0.0", port=7860, use_reloader=False)
 
-# ───── Start Everything ─────
-async def start_scheduler():
-    scheduler.start()
-    await load_all_userbots()
+# ───── Full Start ─────
 async def full_start():
     Thread(target=flask_thread).start()
     scheduler.start()
     await load_all_userbots()
-    await bot.run_until_disconnected()
+    asyncio.create_task(bot.run_until_disconnected())
+    log.info("🤖 Bot is now running on Hugging Face")
 
 def main():
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     loop.run_until_complete(full_start())
-    
+    loop.run_forever()
+
+if __name__ == "__main__":
+    main()