Update main.py

main.py

@@ -4,6 +4,7 @@ import os
 import asyncio
 from flask import Flask
 from threading import Thread
+import time
 
 # Bot setup
 intents = discord.Intents.default()
@@ -13,40 +14,104 @@ intents.presences = True
 
 bot = commands.Bot(command_prefix="!", intents=intents, help_command=None)
 
+# Global variables for web status
+bot_status = {
+    "ready": False,
+    "name": "Loading...",
+    "id": "...",
+    "latency": 0,
+    "servers": 0,
+    "error": None,
+    "start_time": time.time()
+}
+
 # Keep-alive server for Hugging Face
 app = Flask(__name__)
 
 @app.route('/')
 def home():
+    status_text = "Online ✅" if bot_status["ready"] else ("Error ❌" if bot_status["error"] else "Connecting...")
+    status_color = "#00ff00" if bot_status["ready"] else ("#ff0000" if bot_status["error"] else "#ffa500")
+
+    error_html = f"""
+    <div style="background: #ff4444; color: white; padding: 15px; border-radius: 8px; margin: 20px 0;">
+        <strong>⚠️ Error:</strong> {bot_status["error"]}
+    </div>
+    """ if bot_status["error"] else ""
+
     return f"""
     <html>
-        <head><title>{bot.user.name if bot.user else 'Discord Bot'} - Status</title></head>
-        <body style="font-family: Arial; padding: 40px; background: #1a1a2e; color: white;">
-            <h1>🤖 Discord Bot Status</h1>
-            <p><strong>Status:</strong> {'Online ✅' if bot.is_ready() else 'Connecting...'}</p>
-            <p><strong>Bot Name:</strong> {bot.user.name if bot.user else 'Loading...'}</p>
-            <p><strong>Latency:</strong> {round(bot.latency * 1000, 2) if bot.is_ready() else '...'} ms</p>
-            <p><strong>Servers:</strong> {len(bot.guilds) if bot.is_ready() else '...'}</p>
-            <hr>
-            <h3>Commands:</h3>
-            <ul>
-                <li><code>!ping</code> - Check bot latency</li>
-                <li><code>!server</code> - Server info</li>
-                <li><code>!user @mention</code> - User info</li>
-                <li><code>!clear [amount]</code> - Clear messages</li>
-                <li><code>!say [message]</code> - Bot says something</li>
-                <li><code>!embed [title] | [description]</code> - Create embed</li>
-                <li><code>!poll [question]</code> - Create poll</li>
-                <li><code>!avatar @mention</code> - Get user avatar</li>
-                <li><code>!help</code> - Show all commands</li>
-            </ul>
+        <head>
+            <title>{bot_status["name"]} - Discord Bot Status</title>
+            <meta http-equiv="refresh" content="10">
+            <style>
+                body {{ font-family: 'Segoe UI', Arial, sans-serif; padding: 40px; background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%); color: white; min-height: 100vh; }}
+                .container {{ max-width: 800px; margin: 0 auto; }}
+                .status-card {{ background: rgba(255,255,255,0.1); padding: 30px; border-radius: 15px; backdrop-filter: blur(10px); margin-bottom: 20px; }}
+                .status-indicator {{ display: inline-block; width: 15px; height: 15px; border-radius: 50%; background: {status_color}; margin-right: 10px; animation: pulse 2s infinite; }}
+                @keyframes pulse {{ 0% {{ opacity: 1; }} 50% {{ opacity: 0.5; }} 100% {{ opacity: 1; }} }}
+                .info-grid {{ display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-top: 20px; }}
+                .info-item {{ background: rgba(255,255,255,0.05); padding: 15px; border-radius: 8px; }}
+                .info-label {{ color: #aaa; font-size: 12px; text-transform: uppercase; }}
+                .info-value {{ font-size: 18px; font-weight: bold; margin-top: 5px; }}
+                .commands {{ background: rgba(255,255,255,0.05); padding: 20px; border-radius: 10px; }}
+                .command {{ background: rgba(0,0,0,0.3); padding: 8px 12px; border-radius: 5px; margin: 5px 0; font-family: monospace; }}
+                .token-error {{ background: #ff4444; color: white; padding: 20px; border-radius: 10px; margin: 20px 0; }}
+            </style>
+        </head>
+        <body>
+            <div class="container">
+                <div class="status-card">
+                    <h1>🤖 Discord Bot Status</h1>
+                    <div style="font-size: 24px; margin: 20px 0;">
+                        <span class="status-indicator"></span>
+                        <strong>{status_text}</strong>
+                    </div>
+                    {error_html}
+                    <div class="info-grid">
+                        <div class="info-item">
+                            <div class="info-label">Bot Name</div>
+                            <div class="info-value">{bot_status["name"]}</div>
+                        </div>
+                        <div class="info-item">
+                            <div class="info-label">Bot ID</div>
+                            <div class="info-value">{bot_status["id"]}</div>
+                        </div>
+                        <div class="info-item">
+                            <div class="info-label">Latency</div>
+                            <div class="info-value">{round(bot_status["latency"] * 1000, 2) if bot_status["ready"] else "..."} ms</div>
+                        </div>
+                        <div class="info-item">
+                            <div class="info-label">Servers</div>
+                            <div class="info-value">{bot_status["servers"]}</div>
+                        </div>
+                    </div>
+                </div>
+
+                <div class="commands">
+                    <h3>📋 Available Commands:</h3>
+                    <div class="command">!ping</div>
+                    <div class="command">!server</div>
+                    <div class="command">!user @mention</div>
+                    <div class="command">!clear [amount]</div>
+                    <div class="command">!say [message]</div>
+                    <div class="command">!embed [title] | [description]</div>
+                    <div class="command">!poll [question]</div>
+                    <div class="command">!avatar @mention</div>
+                    <div class="command">!help</div>
+                </div>
+
+                <div style="margin-top: 30px; text-align: center; color: #888; font-size: 12px;">
+                    Last updated: {time.strftime("%Y-%m-%d %H:%M:%S")} | Auto-refresh: 10s
+                </div>
+            </div>
         </body>
     </html>
     """
 
 @app.route('/health')
 def health():
-    return {"status": "healthy", "bot_ready": bot.is_ready()}, 200
+    return {"status": "healthy", "bot_ready": bot_status["ready"], "bot_name": bot_status["name"]}, 200
 
 def run_flask():
     app.run(host='0.0.0.0', port=7860)
@@ -54,6 +119,10 @@ def run_flask():
 # Event: Bot ready
 @bot.event
 async def on_ready():
+    bot_status["ready"] = True
+    bot_status["name"] = bot.user.name
+    bot_status["id"] = str(bot.user.id)
+    bot_status["error"] = None
     print(f'✅ Logged in as {bot.user.name} ({bot.user.id})')
     print(f'📊 Connected to {len(bot.guilds)} servers')
     await bot.change_presence(
@@ -81,18 +150,14 @@ async def on_member_join(member):
 async def on_message(message):
     if message.author.bot:
         return
-
-    # Auto-react to mentions
     if bot.user in message.mentions:
         await message.add_reaction('👋')
-
     await bot.process_commands(message)
 
 # ============ COMMANDS ============
 
 @bot.command()
 async def ping(ctx):
-    """Check bot latency"""
     latency = round(bot.latency * 1000, 2)
     embed = discord.Embed(
         title="🏓 Pong!",
@@ -103,7 +168,6 @@ async def ping(ctx):
 
 @bot.command()
 async def server(ctx):
-    """Show server information"""
     guild = ctx.guild
     embed = discord.Embed(
         title=f"📊 {guild.name} Info",
@@ -121,7 +185,6 @@ async def server(ctx):
 
 @bot.command()
 async def user(ctx, member: discord.Member = None):
-    """Show user information"""
     member = member or ctx.author
     embed = discord.Embed(
         title=f"👤 {member.name} Info",
@@ -139,11 +202,9 @@ async def user(ctx, member: discord.Member = None):
 @bot.command()
 @commands.has_permissions(manage_messages=True)
 async def clear(ctx, amount: int = 5):
-    """Clear messages (default: 5)"""
     if amount < 1 or amount > 100:
         await ctx.send("❌ Please specify a number between 1 and 100!")
         return
-
     deleted = await ctx.channel.purge(limit=amount + 1)
     embed = discord.Embed(
         description=f"🗑️ Deleted **{len(deleted)-1}** messages!",
@@ -155,17 +216,14 @@ async def clear(ctx, amount: int = 5):
 
 @bot.command()
 async def say(ctx, *, message):
-    """Make the bot say something"""
     await ctx.message.delete()
     await ctx.send(message)
 
 @bot.command()
 async def embed(ctx, *, content):
-    """Create a custom embed. Usage: !embed Title | Description"""
     parts = content.split("|", 1)
     title = parts[0].strip()
     description = parts[1].strip() if len(parts) > 1 else ""
-
     embed = discord.Embed(
         title=title,
         description=description,
@@ -176,7 +234,6 @@ async def embed(ctx, *, content):
 
 @bot.command()
 async def poll(ctx, *, question):
-    """Create a poll"""
     embed = discord.Embed(
         title="📊 Poll",
         description=f"**{question}**",
@@ -190,7 +247,6 @@ async def poll(ctx, *, question):
 
 @bot.command()
 async def avatar(ctx, member: discord.Member = None):
-    """Get user avatar"""
     member = member or ctx.author
     embed = discord.Embed(
         title=f"🖼️ {member.name}'s Avatar",
@@ -203,13 +259,11 @@ async def avatar(ctx, member: discord.Member = None):
 
 @bot.command()
 async def help(ctx):
-    """Show all commands"""
     embed = discord.Embed(
         title="🤖 Bot Commands",
         description="Here are all available commands:",
         color=discord.Color.blue()
     )
-
     commands_list = [
         ("!ping", "Check bot latency"),
         ("!server", "Server information"),
@@ -221,10 +275,8 @@ async def help(ctx):
         ("!avatar @mention", "Get user avatar"),
         ("!help", "Show this message")
     ]
-
     for cmd, desc in commands_list:
         embed.add_field(name=cmd, value=desc, inline=False)
-
     embed.set_footer(text=f"Requested by {ctx.author.name}")
     await ctx.send(embed=embed)
 
@@ -242,16 +294,46 @@ async def on_command_error(ctx, error):
     else:
         print(f"Error: {error}")
 
+async def start_bot():
+    TOKEN = os.getenv("DISCORD_TOKEN")
+    if not TOKEN:
+        error_msg = "DISCORD_TOKEN not found! Please set it in Hugging Face secrets."
+        bot_status["error"] = error_msg
+        bot_status["name"] = "Token Missing"
+        print(f"❌ {error_msg}")
+        return
+
+    max_retries = 5
+    retry_count = 0
+
+    while retry_count < max_retries:
+        try:
+            print(f"🔄 Attempting to connect to Discord... (Attempt {retry_count + 1}/{max_retries})")
+            await bot.start(TOKEN)
+            break
+        except discord.LoginFailure as e:
+            error_msg = f"Invalid Discord token! Please check your token. Error: {str(e)}"
+            bot_status["error"] = error_msg
+            bot_status["name"] = "Invalid Token"
+            print(f"❌ {error_msg}")
+            break
+        except Exception as e:
+            retry_count += 1
+            error_msg = f"Connection error: {str(e)}. Retrying in 10 seconds..."
+            bot_status["error"] = error_msg
+            bot_status["name"] = "Connection Error"
+            print(f"⚠️ {error_msg}")
+            if retry_count < max_retries:
+                await asyncio.sleep(10)
+            else:
+                bot_status["error"] = f"Failed to connect after {max_retries} attempts. Discord API might be down or network blocked."
+                bot_status["name"] = "Connection Failed"
+                print(f"❌ {bot_status['error']}")
+
 # Run bot
 if __name__ == "__main__":
-    # Start Flask server in background
     flask_thread = Thread(target=run_flask)
     flask_thread.daemon = True
     flask_thread.start()
 
-    # Run Discord bot
-    TOKEN = os.getenv("DISCORD_TOKEN")
-    if not TOKEN:
-        print("❌ DISCORD_TOKEN not found! Please set it in Hugging Face secrets.")
-    else:
-        bot.run(TOKEN)
+    asyncio.run(start_bot())