Update jarvis.py

jarvis.py

@@ -222,26 +222,35 @@ async def jarvis_trigger(client, message):
             await progress_msg.edit(f"`Attempt {attempt}/5:` Thinking and generating code...\n- Executing prerequisite shell commands...\n- Testing generated code...\n- Tests passed. Writing files...\n- Files written. Initiating restart...")
 
             # === NEW: Show module details before restart ===
-            commands = extract_commands(code)
-            commands_str = ', '.join(f'/{cmd}' for cmd in commands) if commands else 'No commands found.'
-            explain_prompt = (
-                f"Explain in 2-3 sentences what this Pyrogram Telegram bot module does, given the following code and the user's request.\n"
-                f"User request: {description}\n"
-                f"Module code:\n{code}\n"
-                f"Be concise and clear."
-            )
             try:
+                commands = extract_commands(code)
+                commands_str = ', '.join(f'/{cmd}' for cmd in commands) if commands else 'No commands found.'
+            except Exception as cmd_error:
+                commands_str = f'Error extracting commands: {cmd_error}'
+            
+            try:
+                explain_prompt = (
+                    f"Explain in 2-3 sentences what this Pyrogram Telegram bot module does, given the following code and the user's request.\n"
+                    f"User request: {description}\n"
+                    f"Module code:\n{code}\n"
+                    f"Be concise and clear."
+                )
                 explain_response = model.generate_content(explain_prompt)
                 explanation = explain_response.text.strip()
             except Exception as exp_explain:
-                explanation = f"(Could not generate explanation: {exp_explain})"
-            details_msg = (
-                f" Module created successfully!\n\n"
-                f"Commands: {commands_str}\n\n"
-                f"What it does:\n{explanation}\n\n"
-                f"♻️ Restarting bot to load the new module..."
-            )
-            await progress_msg.edit(details_msg, parse_mode=ParseMode.MARKDOWN)
+                explanation = f"Could not generate explanation: {exp_explain}"
+            
+            try:
+                details_msg = (
+                    f"✅ Module created successfully!\n\n"
+                    f"**Commands:** {commands_str}\n\n"
+                    f"**What it does:**\n{explanation}\n\n"
+                    f"♻️ Restarting bot to load the new module..."
+                )
+                await progress_msg.edit(details_msg, parse_mode=ParseMode.MARKDOWN)
+            except Exception as msg_error:
+                # Fallback message if formatting fails
+                await progress_msg.edit(f"✅ Module created successfully! Commands: {commands_str}\n\nWhat it does: {explanation}\n\n♻️ Restarting bot...")
             # === END NEW ===
             await asyncio.sleep(2)
             asyncio.create_task(restart_bot(message.chat.id))
@@ -250,7 +259,18 @@ async def jarvis_trigger(client, message):
 
         except Exception as e:
             last_error = e
-            await progress_msg.edit(f"❌ Attempt {attempt} Error: `{str(e)[:100]}...`")
+            error_msg = f"❌ Attempt {attempt} Error: {str(e)}"
+            print(f"DEBUG: Module generation error on attempt {attempt}: {e}")
+            print(f"DEBUG: Error type: {type(e).__name__}")
+            try:
+                await progress_msg.edit(error_msg[:100] + "..." if len(error_msg) > 100 else error_msg)
+            except Exception as edit_error:
+                print(f"DEBUG: Failed to edit progress message: {edit_error}")
+                # Try to send a new message if editing fails
+                try:
+                    await message.reply(f"❌ Attempt {attempt} failed. Check logs for details.")
+                except:
+                    pass
 
     if not success:
         await progress_msg.edit("❌ All 5 attempts failed. Please try again with a simpler instruction.")