Update jarvis.py

jarvis.py

@@ -14,6 +14,7 @@ API_ID = os.environ.get("API_ID")                 # Replace with your API ID
 API_HASH = os.environ.get("API_HASH")   # Replace with your API HASH
 BOT_TOKEN = os.environ.get("BOT_TOKEN")
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+OWNER_ID = int(os.environ.get("OWNER_ID", "7361622601"))  # Owner user ID
 if not BOT_TOKEN:
     raise RuntimeError("BOT_TOKEN environment variable not set!")
 if not GEMINI_API_KEY:
@@ -123,7 +124,10 @@ async def restart_bot(chat_id):
 @bot.on_message(filters.private & filters.text)
 async def jarvis_trigger(client, message):
     # Owner-only access control
-    if message.from_user.id != 7361622601:
+    print(f"DEBUG: User ID: {message.from_user.id}, Expected: {OWNER_ID}")
+    print(f"DEBUG: User: {message.from_user.first_name} {message.from_user.last_name}")
+    if message.from_user.id != OWNER_ID:
+        print(f"DEBUG: Access denied for user {message.from_user.id}")
         return await message.reply("❌ Access denied. Only the owner can use this bot.")
 
     text = message.text.strip()
@@ -201,7 +205,9 @@ async def jarvis_trigger(client, message):
                 f"2. Include commands using @bot.on_message(filters.command(...))\n"
                 f"3. Import only what you need from pyrogram\n and use 'from ..t1 import bot'\n 'try: from ..t1 import bot except (ImportError, ValueError): \n from __main__ import bot' \n"
                 f"4. Don't create a new Client instance\n"
-                f"5. Make functions async when using bot methods\n\n"
+                f"5. Make functions async when using bot methods\n"
+                f"6. ONLY create the specific commands requested - do NOT add standard commands like /start, /help, /about unless explicitly asked for\n"
+                f"7. If the user asks for '/bash', only create the /bash command, not additional commands\n\n"
                 f"Output ONLY Python code, no explanations or markdown."
             )
             if attempt > 1 and last_error is not None:
@@ -214,6 +220,24 @@ async def jarvis_trigger(client, message):
             code = code.replace('asyncio.PIPE', 'subprocess.PIPE')
             # Add more auto-fixes as needed
 
+            # === REMOVE UNWANTED STANDARD COMMANDS ===
+            # Only keep commands that were explicitly requested
+            import re
+            requested_commands = []
+            # Extract commands mentioned in the description
+            cmd_matches = re.findall(r'/(\w+)', description.lower())
+            if cmd_matches:
+                requested_commands = [f'/{cmd}' for cmd in cmd_matches]
+            
+            # If specific commands were requested, remove unwanted standard commands
+            if requested_commands:
+                unwanted_commands = ['/start', '/help', '/about', '/info']
+                for unwanted in unwanted_commands:
+                    if unwanted not in requested_commands:
+                        # Remove the entire function/handler for unwanted commands
+                        pattern = rf'@bot\.on_message\(filters\.command\(["\']{unwanted[1:]}["\']\)\).*?(?=@bot\.on_message|$)'
+                        code = re.sub(pattern, '', code, flags=re.DOTALL)
+
             # === LINTER/STATIC CHECKER ===
             with open('tmp_lint_module.py', 'w', encoding='utf-8') as tmpf:
                 tmpf.write(code)
@@ -325,7 +349,7 @@ async def jarvis_trigger(client, message):
 @bot.on_message(filters.command("modules") & filters.private)
 async def list_modules(client, message):
     # Owner-only access control
-    if message.from_user.id != 7361622601:
+    if message.from_user.id != OWNER_ID:
         return await message.reply("❌ Access denied. Only the owner can use this bot.")
 
     modules = [f for f in os.listdir("modules") if f.endswith(".py")]
@@ -339,7 +363,7 @@ async def list_modules(client, message):
 @bot.on_message(filters.command("delete") & filters.private)
 async def delete_module(client, message):
     # Owner-only access control
-    if message.from_user.id != 7361622601:
+    if message.from_user.id != OWNER_ID:
         return await message.reply("❌ Access denied. Only the owner can use this bot.")
 
     if len(message.command) < 2:
@@ -358,7 +382,7 @@ async def delete_module(client, message):
 @bot.on_message(filters.regex(r"(?i)what( can)? i do\??") & filters.private)
 async def what_to_do(_, message):
     # Owner-only access control
-    if message.from_user.id != 7361622601:
+    if message.from_user.id != OWNER_ID:
         return await message.reply("❌ Access denied. Only the owner can use this bot.")
 
     await message.reply(