Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import discord
 from discord.ext import commands
 import json
@@ -11,6 +12,7 @@ from gradio_client import Client
 from concurrent.futures import ThreadPoolExecutor
 from gtts import gTTS
 import io
+import aiohttp
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -56,13 +58,20 @@ You are GaurdineX, a futuristic next-generation AI guardian model built to act a
 # Helper functions
 def load_data(filename, default_data):
     if os.path.exists(filename):
-        with open(filename, 'r', encoding='utf-8') as f:
-            return json.load(f)
+        try:
+            with open(filename, 'r', encoding='utf-8') as f:
+                return json.load(f)
+        except Exception as e:
+            logger.error(f"Error loading {filename}: {e}")
+            return default_data
     return default_data
 
 def save_data(data, filename):
-    with open(filename, 'w', encoding='utf-8') as f:
-        json.dump(data, f, indent=4, ensure_ascii=False)
+    try:
+        with open(filename, 'w', encoding='utf-8') as f:
+            json.dump(data, f, indent=4, ensure_ascii=False)
+    except Exception as e:
+        logger.error(f"Error saving {filename}: {e}")
 
 def log_action(guild_id, action, user_id, moderator_id, reason):
     log_entry = {
@@ -121,7 +130,12 @@ async def timeout_check():
 @bot.event
 async def on_guild_join(guild):
     try:
-        role = await guild.create_role(name="🛡 Bot Moderation", permissions=discord.Permissions(moderate_members=True, kick_members=True, ban_members=True))
+        role = await guild.create_role(
+            name="🛡 Bot Moderation",
+            permissions=discord.Permissions(moderate_members=True, kick_members=True, ban_members=True, manage_messages=True),
+            hoist=True,
+            mentionable=True
+        )
         guild_id = str(guild.id)
         config_data[guild_id] = {
             'admin_role_id': role.id,
@@ -130,7 +144,7 @@ async def on_guild_join(guild):
             'bad_words': ['fuck', 'shit', 'asshole']  # Add more bad words as needed
         }
         save_data(config_data, CONFIG_FILE)
-        logger.info(f"Created moderation role and set config for guild {guild_id}")
+        logger.info(f"Created '🛡 Bot Moderation' role and set config for guild {guild_id}")
     except discord.Forbidden:
         logger.error(f"Failed to create role in guild {guild.id}: Missing permissions")
     except Exception as e:
@@ -195,27 +209,32 @@ async def process_ai_command(ctx, prompt: str):
             return
     
     # Default AI response
-    try:
-        client = Client("amd/gpt-oss-120b-chatbot")
-        loop = asyncio.get_running_loop()
-        ai_output = await loop.run_in_executor(
-            executor,
-            lambda: client.predict(
-                message=prompt,
-                system_prompt=SYSTEM_PROMPT,
-                temperature=0.7,
-                reasoning_effort="medium",
-                enable_browsing=False,
-                api_name="/chat"
+    for attempt in range(3):  # Retry up to 3 times
+        try:
+            client = Client("amd/gpt-oss-120b-chatbot")
+            loop = asyncio.get_running_loop()
+            ai_output = await loop.run_in_executor(
+                executor,
+                lambda: client.predict(
+                    message=prompt,
+                    system_prompt=SYSTEM_PROMPT,
+                    temperature=0.7,
+                    reasoning_effort="medium",
+                    enable_browsing=False,
+                    api_name="/chat"
+                )
             )
-        )
-        match = re.search(r'💬 Response:\s*(.+)', ai_output, re.DOTALL)
-        response = match.group(1).strip() if match else ai_output.strip()
-        await ctx.send(response)
-    except Exception as e:
-        logger.error(f"AI error: {e}")
-        msg = f"Error communicating with AI: {e}" if lang == 'en' else f"حصل خطأ أثناء التواصل مع الذكاء الاصطناعي: {e}"
-        await ctx.send(msg)
+            match = re.search(r'💬 Response:\s*(.+)', ai_output, re.DOTALL)
+            response = match.group(1).strip() if match else ai_output.strip()
+            await ctx.send(response)
+            return
+        except Exception as e:
+            logger.error(f"AI attempt {attempt + 1} failed: {e}")
+            if attempt < 2:
+                await asyncio.sleep(2)  # Wait before retrying
+            else:
+                msg = f"Error communicating with AI after 3 attempts: {e}" if lang == 'en' else f"حصل خطأ أثناء التواصل مع الذكاء الاصطناعي بعد 3 محاولات: {e}"
+                await ctx.send(msg)
 
 async def chat_with_ai(message: discord.Message):
     guild_id = str(message.guild.id)
@@ -299,6 +318,8 @@ async def on_message(message):
         admin_role = discord.utils.get(message.guild.roles, id=admin_role_id)
         if admin_role and admin_role in message.author.roles:
             is_admin = True
+    if user_id in config.get('allowed_users', []):
+        is_admin = True
     if is_admin and bot.user.mentioned_in(message):
         clean_content = message.content.replace(f'<@!{bot.user.id}>', '').strip()
         await process_ai_command(message.channel, clean_content)
@@ -352,7 +373,7 @@ async def addaccess(ctx, member: discord.Member):
     if user_id not in config_data[guild_id]['allowed_users']:
         config_data[guild_id]['allowed_users'].append(user_id)
         save_data(config_data, CONFIG_FILE)
-    lang = config_data.get(guild_id, {}).get('language', 'en')
+    lang = config_data[guild_id].get('language', 'en')
     msg = f"✅ Added moderation access to {member.mention}." if lang == 'en' else f"✅ تم إضافة صلاحية الإدارة لـ {member.mention}."
     await ctx.send(msg)
     log_action(guild_id, "add_access", member.id, ctx.author.id, "Added individual mod access")
@@ -366,7 +387,7 @@ async def removeaccess(ctx, member: discord.Member):
         if user_id in config_data[guild_id]['allowed_users']:
             config_data[guild_id]['allowed_users'].remove(user_id)
             save_data(config_data, CONFIG_FILE)
-    lang = config_data.get(guild_id, {}).get('language', 'en')
+    lang = config_data[guild_id].get('language', 'en')
     msg = f"✅ Removed moderation access from {member.mention}." if lang == 'en' else f"✅ تم إزالة صلاحية الإدارة من {member.mention}."
     await ctx.send(msg)
     log_action(guild_id, "remove_access", member.id, ctx.author.id, "Removed individual mod access")
@@ -578,30 +599,35 @@ async def chat(ctx, *, user_message: str):
         memory.pop(0)
     thinking_msg = await ctx.send("💭 Thinking..." if lang == 'en' else "💭 OS Bot is thinking...")
     
-    try:
-        client = Client("amd/gpt-oss-120b-chatbot")
-        loop = asyncio.get_running_loop()
-        ai_output = await loop.run_in_executor(
-            executor,
-            lambda: client.predict(
-                message=user_message,
-                system_prompt=SYSTEM_PROMPT,
-                temperature=0.7,
-                reasoning_effort="medium",
-                enable_browsing=False,
-                api_name="/chat"
+    for attempt in range(3):
+        try:
+            client = Client("amd/gpt-oss-120b-chatbot")
+            loop = asyncio.get_running_loop()
+            ai_output = await loop.run_in_executor(
+                executor,
+                lambda: client.predict(
+                    message=user_message,
+                    system_prompt=SYSTEM_PROMPT,
+                    temperature=0.7,
+                    reasoning_effort="medium",
+                    enable_browsing=False,
+                    api_name="/chat"
+                )
             )
-        )
-        match = re.search(r'💬 Response:\s*(.+)', ai_output, re.DOTALL)
-        final_response = match.group(1).strip() if match else ai_output.strip()
-        memory.append({"role": "assistant", "content": final_response})
-        await thinking_msg.edit(content=final_response)
-    except Exception as e:
-        logger.error(f"AI error: {e}")
-        msg = f"Error communicating with AI: {e}" if lang == 'en' else f"حصل خطأ أثناء التواصل مع الذكاء الاصطناعي: {e}"
-        await thinking_msg.edit(content=msg)
-        if user_id in chat_history:
-            del chat_history[user_id]
+            match = re.search(r'💬 Response:\s*(.+)', ai_output, re.DOTALL)
+            final_response = match.group(1).strip() if match else ai_output.strip()
+            memory.append({"role": "assistant", "content": final_response})
+            await thinking_msg.edit(content=final_response)
+            return
+        except Exception as e:
+            logger.error(f"AI attempt {attempt + 1} failed: {e}")
+            if attempt < 2:
+                await asyncio.sleep(2)
+            else:
+                msg = f"Error communicating with AI after 3 attempts: {e}" if lang == 'en' else f"حصل خطأ أثناء التواصل مع الذكاء الاصطناعي بعد 3 محاولات: {e}"
+                await thinking_msg.edit(content=msg)
+                if user_id in chat_history:
+                    del chat_history[user_id]
 
 @bot.command(name='join')
 async def join(ctx):
@@ -646,72 +672,97 @@ async def ask(ctx, *, text_query: str):
     
     thinking_msg = await ctx.send("💭 Thinking..." if lang == 'en' else "💭 OS Bot is thinking...")
     
-    try:
-        client = Client("amd/gpt-oss-120b-chatbot")
-        loop = asyncio.get_running_loop()
-        ai_output = await loop.run_in_executor(
-            executor,
-            lambda: client.predict(
-                message=text_query,
-                system_prompt=SYSTEM_PROMPT,
-                temperature=0.7,
-                reasoning_effort="medium",
-                enable_browsing=False,
-                api_name="/chat"
-            )
-        )
-        match = re.search(r'💬 Response:\s*(.+)', ai_output, re.DOTALL)
-        final_response = match.group(1).strip() if match else ai_output.strip()
-        
-        # Save to chat history
-        user_id = str(ctx.author.id)
-        if user_id not in chat_history:
-            chat_history[user_id] = []
-        chat_history[user_id].append({"role": "user", "content": text_query})
-        chat_history[user_id].append({"role": "assistant", "content": final_response})
-        if len(chat_history[user_id]) > MAX_CHAT_HISTORY:
-            chat_history[user_id].pop(0)
-        
-        # Convert AI response to voice
+    for attempt in range(3):
         try:
-            logger.info(f"Generating TTS for response: {final_response}")
-            tts = gTTS(text=final_response, lang='ar' if lang == 'ar' else 'en', slow=False)
-            voice_buffer = io.BytesIO()
-            tts.write_to_fp(voice_buffer)
-            voice_buffer.seek(0)
-            logger.info(f"TTS audio buffer created, size: {len(voice_buffer.getvalue())} bytes")
-            
-            ffmpeg_options = {
-                'options': '-f s16le -ar 48000 -ac 2'
-            }
-            source = discord.FFmpegPCMAudio(
-                source=voice_buffer,
-                pipe=True,
-                executable=FFMPEG_PATH,
-                **ffmpeg_options
+            client = Client("amd/gpt-oss-120b-chatbot")
+            loop = asyncio.get_running_loop()
+            ai_output = await loop.run_in_executor(
+                executor,
+                lambda: client.predict(
+                    message=text_query,
+                    system_prompt=SYSTEM_PROMPT,
+                    temperature=0.7,
+                    reasoning_effort="medium",
+                    enable_browsing=False,
+                    api_name="/chat"
+                )
             )
+            match = re.search(r'💬 Response:\s*(.+)', ai_output, re.DOTALL)
+            final_response = match.group(1).strip() if match else ai_output.strip()
             
-            def after_playback(error):
-                if error:
-                    logger.error(f"Playback error: {error}")
-                else:
-                    logger.info("Audio playback completed successfully")
+            # Save to chat history
+            user_id = str(ctx.author.id)
+            if user_id not in chat_history:
+                chat_history[user_id] = []
+            chat_history[user_id].append({"role": "user", "content": text_query})
+            chat_history[user_id].append({"role": "assistant", "content": final_response})
+            if len(chat_history[user_id]) > MAX_CHAT_HISTORY:
+                chat_history[user_id].pop(0)
             
-            ctx.voice_client.play(source, after=after_playback)
-            while ctx.voice_client.is_playing():
-                await asyncio.sleep(1)
+            # Convert AI response to voice
+            try:
+                logger.info(f"Generating TTS for response: {final_response}")
+                tts = gTTS(text=final_response, lang='ar' if lang == 'ar' else 'en', slow=False)
+                voice_buffer = io.BytesIO()
+                tts.write_to_fp(voice_buffer)
+                voice_buffer.seek(0)
+                logger.info(f"TTS audio buffer created, size: {len(voice_buffer.getvalue())} bytes")
+                
+                ffmpeg_options = {
+                    'options': '-f s16le -ar 48000 -ac 2'
+                }
+                source = discord.FFmpegPCMAudio(
+                    source=voice_buffer,
+                    pipe=True,
+                    executable=FFMPEG_PATH,
+                    **ffmpeg_options
+                )
+                
+                def after_playback(error):
+                    if error:
+                        logger.error(f"Playback error: {error}")
+                    else:
+                        logger.info("Audio playback completed successfully")
+                
+                ctx.voice_client.play(source, after=after_playback)
+                while ctx.voice_client.is_playing():
+                    await asyncio.sleep(1)
+                
+                await thinking_msg.edit(content=f"Response: `{final_response}`" if lang == 'en' else f"الرد: `{final_response}`")
+                return
             
-            await thinking_msg.edit(content=f"Response: `{final_response}`" if lang == 'en' else f"الرد: `{final_response}`")
+            except Exception as e:
+                logger.error(f"TTS error: {e}")
+                msg = f"Error converting text to speech: {e}" if lang == 'en' else f"حصل خطأ أثناء تحويل النص إلى صوت: {e}"
+                await thinking_msg.edit(content=msg)
+                return
         
         except Exception as e:
-            logger.error(f"TTS error: {e}")
-            msg = f"Error converting text to speech: {e}" if lang == 'en' else f"حصل خطأ أثناء تحويل النص إلى صوت: {e}"
-            await thinking_msg.edit(content=msg)
-    
-    except Exception as e:
-        logger.error(f"AI error: {e}")
-        msg = f"Error communicating with AI: {e}" if lang == 'en' else f"حصل خطأ أثناء التواصل مع الذكاء الاصطناعي: {e}"
-        await thinking_msg.edit(content=msg)
+            logger.error(f"AI attempt {attempt + 1} failed: {e}")
+            if attempt < 2:
+                await asyncio.sleep(2)
+            else:
+                msg = f"Error communicating with AI after 3 attempts: {e}" if lang == 'en' else f"حصل خطأ أثناء التواصل مع الذكاء الاصطناعي بعد 3 محاولات: {e}"
+                await thinking_msg.edit(content=msg)
+                return
 
-# Run the bot
-bot.run(TOKEN)
+# Retry-enabled bot start
+async def start_bot():
+    for attempt in range(3):
+        try:
+            await bot.start(TOKEN)
+            return
+        except aiohttp.client_exceptions.ClientConnectorDNSError as e:
+            logger.error(f"Connection attempt {attempt + 1} failed: {e}")
+            if attempt < 2:
+                await asyncio.sleep(5)
+            else:
+                logger.error("Failed to connect to Discord after 3 attempts. Exiting.")
+                raise
+
+# Run the bot with retry
+if __name__ == "__main__":
+    try:
+        asyncio.run(start_bot())
+    except Exception as e:
+        logger.error(f"Fatal error: {e}")