souleyez 2.43.29__py3-none-any.whl → 3.0.0__py3-none-any.whl

souleyez/ui/setup_wizard.py

@@ -14,7 +14,7 @@ from souleyez.ui.design_system import DesignSystem
 
 
 # Wizard state file
-WIZARD_STATE_FILE = Path.home() / '.souleyez' / '.wizard_completed'
+WIZARD_STATE_FILE = Path.home() / ".souleyez" / ".wizard_completed"
 
 
 def is_wizard_completed() -> bool:
@@ -31,7 +31,7 @@ def mark_wizard_completed():
 def _check_disk_space(min_mb: int = 500) -> bool:
     """Check if there's enough disk space for installs."""
     try:
-        usage = shutil.disk_usage('/')
+        usage = shutil.disk_usage("/")
         free_mb = usage.free // (1024 * 1024)
         return free_mb >= min_mb
     except Exception:
@@ -60,32 +60,36 @@ def _configure_sudoers(tool_name: str, tool_path: str) -> bool:
 
     try:
         # Write to temp file first
-        with open(tmp_file, 'w') as f:
-            f.write(sudoers_line + '\n')
+        with open(tmp_file, "w") as f:
+            f.write(sudoers_line + "\n")
 
         # Validate syntax with visudo before applying
         result = subprocess.run(
-            ['sudo', 'visudo', '-c', '-f', tmp_file],
-            capture_output=True,
-            timeout=30
+            ["sudo", "visudo", "-c", "-f", tmp_file], capture_output=True, timeout=30
         )
 
         if result.returncode != 0:
-            click.echo(f"    {click.style('!', fg='yellow')} Invalid sudoers syntax for {tool_name}, skipping")
+            click.echo(
+                f"    {click.style('!', fg='yellow')} Invalid sudoers syntax for {tool_name}, skipping"
+            )
             os.unlink(tmp_file)
             return False
 
         # Safe to move to sudoers.d
-        subprocess.run(['sudo', 'mv', tmp_file, sudoers_file], check=True, timeout=30)
-        subprocess.run(['sudo', 'chmod', '0440', sudoers_file], check=True, timeout=30)
+        subprocess.run(["sudo", "mv", tmp_file, sudoers_file], check=True, timeout=30)
+        subprocess.run(["sudo", "chmod", "0440", sudoers_file], check=True, timeout=30)
 
         return True
 
     except subprocess.TimeoutExpired:
-        click.echo(f"    {click.style('!', fg='yellow')} Timeout configuring sudoers for {tool_name}")
+        click.echo(
+            f"    {click.style('!', fg='yellow')} Timeout configuring sudoers for {tool_name}"
+        )
         return False
     except Exception as e:
-        click.echo(f"    {click.style('!', fg='yellow')} Error configuring sudoers for {tool_name}: {e}")
+        click.echo(
+            f"    {click.style('!', fg='yellow')} Error configuring sudoers for {tool_name}: {e}"
+        )
         # Clean up temp file if it exists
         if os.path.exists(tmp_file):
             try:
@@ -103,10 +107,10 @@ def _install_desktop_shortcut():
     disrupt the setup flow.
     """
     try:
-        applications_dir = Path.home() / '.local' / 'share' / 'applications'
-        icons_dir = Path.home() / '.local' / 'share' / 'icons'
-        desktop_file = applications_dir / 'souleyez.desktop'
-        icon_dest = icons_dir / 'souleyez.png'
+        applications_dir = Path.home() / ".local" / "share" / "applications"
+        icons_dir = Path.home() / ".local" / "share" / "icons"
+        desktop_file = applications_dir / "souleyez.desktop"
+        icon_dest = icons_dir / "souleyez.png"
 
         # Skip if already installed
         if desktop_file.exists():
@@ -121,16 +125,19 @@ def _install_desktop_shortcut():
             # Try importlib.resources first (Python 3.9+)
             try:
                 from importlib.resources import files
-                icon_source = files('souleyez.assets').joinpath('souleyez-icon.png')
-                with open(icon_source, 'rb') as src:
+
+                icon_source = files("souleyez.assets").joinpath("souleyez-icon.png")
+                with open(icon_source, "rb") as src:
                     icon_data = src.read()
             except (ImportError, TypeError, FileNotFoundError):
                 # Fallback: find icon relative to this file
-                icon_source = Path(__file__).parent.parent / 'assets' / 'souleyez-icon.png'
-                with open(icon_source, 'rb') as src:
+                icon_source = (
+                    Path(__file__).parent.parent / "assets" / "souleyez-icon.png"
+                )
+                with open(icon_source, "rb") as src:
                     icon_data = src.read()
 
-            with open(icon_dest, 'wb') as dst:
+            with open(icon_dest, "wb") as dst:
                 dst.write(icon_data)
         except Exception:
             icon_dest = "utilities-terminal"  # Fallback to system icon
@@ -151,12 +158,18 @@ Keywords=pentest;security;hacking;nmap;metasploit;
 
         # Update desktop database (optional)
         try:
-            subprocess.run(['update-desktop-database', str(applications_dir)],
-                          capture_output=True, check=False, timeout=5)
+            subprocess.run(
+                ["update-desktop-database", str(applications_dir)],
+                capture_output=True,
+                check=False,
+                timeout=5,
+            )
         except Exception:
             pass
 
-        click.echo(f"  {click.style('✓', fg='green')} Desktop shortcut: Added to Applications menu")
+        click.echo(
+            f"  {click.style('✓', fg='green')} Desktop shortcut: Added to Applications menu"
+        )
 
     except Exception:
         # Silently ignore errors - desktop shortcut is nice-to-have
@@ -164,9 +177,7 @@ Keywords=pentest;security;hacking;nmap;metasploit;
 
 
 def _run_tool_installs(
-    missing_tools: List[Dict],
-    wrong_version_tools: List[Dict],
-    distro: str
+    missing_tools: List[Dict], wrong_version_tools: List[Dict], distro: str
 ) -> bool:
     """
     Prompt user and run install/upgrade commands.
@@ -186,25 +197,29 @@ def _run_tool_installs(
     # Add wrong version tools (upgrades)
     for tool in wrong_version_tools:
         # Try to get upgrade command, fall back to install command
-        tool_info = tool.get('tool_info', {})
+        tool_info = tool.get("tool_info", {})
         upgrade_cmd = get_upgrade_command(tool_info, distro) if tool_info else None
-        install_cmd = upgrade_cmd or tool.get('install', '')
-
-        all_installs.append({
-            'name': tool['name'],
-            'cmd': install_cmd,
-            'action': 'upgrade',
-            'tool_info': tool_info
-        })
+        install_cmd = upgrade_cmd or tool.get("install", "")
+
+        all_installs.append(
+            {
+                "name": tool["name"],
+                "cmd": install_cmd,
+                "action": "upgrade",
+                "tool_info": tool_info,
+            }
+        )
 
     # Add missing tools (installs)
     for tool in missing_tools:
-        all_installs.append({
-            'name': tool['name'],
-            'cmd': tool['install'],
-            'action': 'install',
-            'tool_info': tool.get('tool_info', {})
-        })
+        all_installs.append(
+            {
+                "name": tool["name"],
+                "cmd": tool["install"],
+                "action": "install",
+                "tool_info": tool.get("tool_info", {}),
+            }
+        )
 
     if not all_installs:
         return False
@@ -212,7 +227,9 @@ def _run_tool_installs(
     # Check disk space
     if not _check_disk_space(500):
         click.echo()
-        click.echo(f"  {click.style('!', fg='yellow')} Low disk space (<500MB free). Installs may fail.")
+        click.echo(
+            f"  {click.style('!', fg='yellow')} Low disk space (<500MB free). Installs may fail."
+        )
         if not click.confirm("  Continue anyway?", default=False):
             return False
 
@@ -220,8 +237,10 @@ def _run_tool_installs(
     click.echo()
     click.echo(f"  {len(all_installs)} tool(s) to install/upgrade:")
     for item in all_installs:
-        action_color = 'cyan' if item['action'] == 'upgrade' else 'green'
-        click.echo(f"    - {item['name']} ({click.style(item['action'], fg=action_color)})")
+        action_color = "cyan" if item["action"] == "upgrade" else "green"
+        click.echo(
+            f"    - {item['name']} ({click.style(item['action'], fg=action_color)})"
+        )
 
     click.echo()
     if not click.confirm("  Install/upgrade now?", default=False):
@@ -231,19 +250,23 @@ def _run_tool_installs(
     # Request sudo upfront
     click.echo()
     click.echo("  Requesting sudo access...")
-    result = subprocess.run(['sudo', '-v'], check=False)
+    result = subprocess.run(["sudo", "-v"], check=False)
     if result.returncode != 0:
-        click.echo(f"  {click.style('x', fg='red')} Could not obtain sudo. Aborting installs.")
+        click.echo(
+            f"  {click.style('x', fg='red')} Could not obtain sudo. Aborting installs."
+        )
         return False
 
     # Update apt cache first (for apt-based installs)
-    has_apt_installs = any('apt' in item['cmd'] for item in all_installs)
+    has_apt_installs = any("apt" in item["cmd"] for item in all_installs)
     if has_apt_installs:
         click.echo()
         click.echo("  Updating package lists...")
-        result = subprocess.run(['sudo', 'apt', 'update'], capture_output=True)
+        result = subprocess.run(["sudo", "apt", "update"], capture_output=True)
         if result.returncode != 0:
-            click.echo(f"  {click.style('!', fg='yellow')} apt update had issues, continuing anyway...")
+            click.echo(
+                f"  {click.style('!', fg='yellow')} apt update had issues, continuing anyway..."
+            )
 
     # Track results for summary
     results = []
@@ -251,29 +274,51 @@ def _run_tool_installs(
     # Run each install command
     for item in all_installs:
         click.echo()
-        click.echo(f"  {click.style(item['action'].capitalize() + 'ing', fg='cyan')} {item['name']}...")
+        click.echo(
+            f"  {click.style(item['action'].capitalize() + 'ing', fg='cyan')} {item['name']}..."
+        )
 
         try:
             # Run install command
             proc = subprocess.run(
-                item['cmd'],
+                item["cmd"],
                 shell=True,  # nosec B602 - commands from trusted EXTERNAL_TOOLS
-                timeout=600  # 10 minute timeout per tool
+                timeout=600,  # 10 minute timeout per tool
             )
 
             if proc.returncode == 0:
-                click.echo(f"  {click.style('✓', fg='green')} {item['name']} {item['action']} complete")
-                results.append({'name': item['name'], 'success': True, 'tool_info': item['tool_info']})
+                click.echo(
+                    f"  {click.style('✓', fg='green')} {item['name']} {item['action']} complete"
+                )
+                results.append(
+                    {
+                        "name": item["name"],
+                        "success": True,
+                        "tool_info": item["tool_info"],
+                    }
+                )
             else:
-                click.echo(f"  {click.style('✗', fg='red')} {item['name']} failed (exit {proc.returncode})")
-                results.append({'name': item['name'], 'success': False, 'tool_info': item['tool_info']})
+                click.echo(
+                    f"  {click.style('✗', fg='red')} {item['name']} failed (exit {proc.returncode})"
+                )
+                results.append(
+                    {
+                        "name": item["name"],
+                        "success": False,
+                        "tool_info": item["tool_info"],
+                    }
+                )
 
         except subprocess.TimeoutExpired:
             click.echo(f"  {click.style('✗', fg='red')} {item['name']} timed out")
-            results.append({'name': item['name'], 'success': False, 'tool_info': item['tool_info']})
+            results.append(
+                {"name": item["name"], "success": False, "tool_info": item["tool_info"]}
+            )
         except Exception as e:
             click.echo(f"  {click.style('✗', fg='red')} {item['name']} error: {e}")
-            results.append({'name': item['name'], 'success': False, 'tool_info': item['tool_info']})
+            results.append(
+                {"name": item["name"], "success": False, "tool_info": item["tool_info"]}
+            )
 
     # Configure sudoers for privileged tools that installed successfully
     click.echo()
@@ -281,32 +326,38 @@ def _run_tool_installs(
 
     sudoers_configured = False
     for res in results:
-        if not res['success']:
+        if not res["success"]:
             continue
 
-        tool_info = res['tool_info']
-        if not tool_info.get('needs_sudo'):
+        tool_info = res["tool_info"]
+        if not tool_info.get("needs_sudo"):
             continue
 
         sudoers_configured = True
         # Find the actual binary path
-        command = tool_info.get('command', res['name'])
+        command = tool_info.get("command", res["name"])
         tool_path = shutil.which(command)
 
         # Check alt_commands if primary not found
-        if not tool_path and tool_info.get('alt_commands'):
-            for alt in tool_info['alt_commands']:
+        if not tool_path and tool_info.get("alt_commands"):
+            for alt in tool_info["alt_commands"]:
                 tool_path = shutil.which(alt)
                 if tool_path:
                     break
 
         if tool_path:
-            if _configure_sudoers(res['name'], tool_path):
-                click.echo(f"    {click.style('✓', fg='green')} {res['name']} configured for passwordless sudo")
+            if _configure_sudoers(res["name"], tool_path):
+                click.echo(
+                    f"    {click.style('✓', fg='green')} {res['name']} configured for passwordless sudo"
+                )
             else:
-                click.echo(f"    {click.style('!', fg='yellow')} {res['name']} sudoers config failed")
+                click.echo(
+                    f"    {click.style('!', fg='yellow')} {res['name']} sudoers config failed"
+                )
         else:
-            click.echo(f"    {click.style('!', fg='yellow')} {res['name']} binary not found, skipping sudoers")
+            click.echo(
+                f"    {click.style('!', fg='yellow')} {res['name']} binary not found, skipping sudoers"
+            )
 
     if not sudoers_configured:
         click.echo("    No privileged tools needed configuration.")
@@ -319,9 +370,9 @@ def _run_tool_installs(
     fail_count = 0
 
     for res in results:
-        tool_info = res['tool_info']
-        command = tool_info.get('command', res['name'])
-        alt_commands = tool_info.get('alt_commands')
+        tool_info = res["tool_info"]
+        command = tool_info.get("command", res["name"])
+        alt_commands = tool_info.get("alt_commands")
 
         # Check if tool is now available
         found = shutil.which(command) is not None
@@ -335,15 +386,21 @@ def _run_tool_installs(
             click.echo(f"    {click.style('✓', fg='green')} {res['name']} verified")
             success_count += 1
         else:
-            click.echo(f"    {click.style('✗', fg='red')} {res['name']} not found after install")
+            click.echo(
+                f"    {click.style('✗', fg='red')} {res['name']} not found after install"
+            )
             fail_count += 1
 
     # Summary
     click.echo()
     if fail_count == 0:
-        click.echo(f"  {click.style('✓', fg='green')} All {success_count} tool(s) installed successfully!")
+        click.echo(
+            f"  {click.style('✓', fg='green')} All {success_count} tool(s) installed successfully!"
+        )
     else:
-        click.echo(f"  {click.style('!', fg='yellow')} {success_count} succeeded, {fail_count} failed")
+        click.echo(
+            f"  {click.style('!', fg='yellow')} {success_count} succeeded, {fail_count} failed"
+        )
         click.echo("  Failed tools may need manual installation.")
 
     return True
@@ -352,14 +409,56 @@ def _run_tool_installs(
 def _show_wizard_banner():
     """Display the SoulEyez ASCII banner for wizard steps."""
     click.echo()
-    click.echo(click.style("   ███████╗ ██████╗ ██╗   ██╗██╗     ███████╗██╗   ██╗███████╗███████╗", fg='bright_cyan', bold=True) + click.style("        ▄██▄", fg='bright_blue', bold=True))
-    click.echo(click.style("   ██╔════╝██╔═══██╗██║   ██║██║     ██╔════╝╚██╗ ██╔╝██╔════╝╚══███╔╝", fg='bright_cyan', bold=True) + click.style("      ▄█▀  ▀█▄", fg='bright_blue', bold=True))
-    click.echo(click.style("   ███████╗██║   ██║██║   ██║██║     █████╗   ╚████╔╝ █████╗    ███╔╝ ", fg='bright_cyan', bold=True) + click.style("     █   ◉   █", fg='bright_blue', bold=True))
-    click.echo(click.style("   ╚════██║██║   ██║██║   ██║██║     ██╔══╝    ╚██╔╝  ██╔══╝   ███╔╝  ", fg='bright_cyan', bold=True) + click.style("     █  ═══  █", fg='bright_blue', bold=True))
-    click.echo(click.style("   ███████║╚██████╔╝╚██████╔╝███████╗███████╗   ██║   ███████╗███████╗", fg='bright_cyan', bold=True) + click.style("      ▀█▄  ▄█▀", fg='bright_blue', bold=True))
-    click.echo(click.style("   ╚══════╝ ╚═════╝  ╚═════╝ ╚══════╝╚══════╝   ╚═╝   ╚══════╝╚══════╝", fg='bright_cyan', bold=True) + click.style("        ▀██▀", fg='bright_blue', bold=True))
+    click.echo(
+        click.style(
+            "   ███████╗ ██████╗ ██╗   ██╗██╗     ███████╗██╗   ██╗███████╗███████╗",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("        ▄██▄", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ██╔════╝██╔═══██╗██║   ██║██║     ██╔════╝╚██╗ ██╔╝██╔════╝╚══███╔╝",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("      ▄█▀  ▀█▄", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ███████╗██║   ██║██║   ██║██║     █████╗   ╚████╔╝ █████╗    ███╔╝ ",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("     █   ◉   █", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ╚════██║██║   ██║██║   ██║██║     ██╔══╝    ╚██╔╝  ██╔══╝   ███╔╝  ",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("     █  ═══  █", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ███████║╚██████╔╝╚██████╔╝███████╗███████╗   ██║   ███████╗███████╗",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("      ▀█▄  ▄█▀", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ╚══════╝ ╚═════╝  ╚═════╝ ╚══════╝╚══════╝   ╚═╝   ╚══════╝╚══════╝",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("        ▀██▀", fg="bright_blue", bold=True)
+    )
     click.echo()
-    click.echo(click.style("  Created by CyberSoul SecurITy", fg='bright_blue'))
+    click.echo(click.style("  Created by CyberSoul SecurITy", fg="bright_blue"))
     click.echo()
 
 
@@ -373,6 +472,7 @@ def run_setup_wizard() -> bool:
     try:
         # Check if user has Pro tier
         from souleyez.auth import get_current_user, Tier
+
         user = get_current_user()
         is_pro = user and user.tier == Tier.PRO
 
@@ -399,11 +499,11 @@ def run_setup_wizard() -> bool:
         if is_pro:
             automation_prefs = _wizard_automation_prefs()
         else:
-            automation_prefs = {'enabled': False, 'mode': None}
+            automation_prefs = {"enabled": False, "mode": None}
 
         # Step 7: Deliverable Templates (Pro only)
         if is_pro:
-            templates = _wizard_deliverables(engagement_info.get('type'))
+            templates = _wizard_deliverables(engagement_info.get("type"))
         else:
             templates = []
 
@@ -415,14 +515,14 @@ def run_setup_wizard() -> bool:
             ai_enabled,
             automation_prefs,
             templates,
-            is_pro
+            is_pro,
         )
 
         mark_wizard_completed()
         return True
 
     except (KeyboardInterrupt, click.Abort):
-        click.echo(click.style("\n\n  Setup wizard cancelled.", fg='yellow'))
+        click.echo(click.style("\n\n  Setup wizard cancelled.", fg="yellow"))
         mark_wizard_completed()  # Don't show wizard again
         click.pause()
         return False
@@ -435,40 +535,100 @@ def _wizard_welcome(is_pro: bool = False) -> bool:
 
     # Header
     click.echo("\n┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" WELCOME TO SOULEYEZ - SETUP WIZARD ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " WELCOME TO SOULEYEZ - SETUP WIZARD ".center(width - 2),
+            bold=True,
+            fg="cyan",
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
     # ASCII Art Banner - SOULEYEZ with all-seeing eye on the right
-    click.echo(click.style("   ███████╗ ██████╗ ██╗   ██╗██╗     ███████╗██╗   ██╗███████╗███████╗", fg='bright_cyan', bold=True) + click.style("        ▄██▄", fg='bright_blue', bold=True))
-    click.echo(click.style("   ██╔════╝██╔═══██╗██║   ██║██║     ██╔════╝╚██╗ ██╔╝██╔════╝╚══███╔╝", fg='bright_cyan', bold=True) + click.style("      ▄█▀  ▀█▄", fg='bright_blue', bold=True))
-    click.echo(click.style("   ███████╗██║   ██║██║   ██║██║     █████╗   ╚████╔╝ █████╗    ███╔╝ ", fg='bright_cyan', bold=True) + click.style("     █   ◉   █", fg='bright_blue', bold=True))
-    click.echo(click.style("   ╚════██║██║   ██║██║   ██║██║     ██╔══╝    ╚██╔╝  ██╔══╝   ███╔╝  ", fg='bright_cyan', bold=True) + click.style("     █  ═══  █", fg='bright_blue', bold=True))
-    click.echo(click.style("   ███████║╚██████╔╝╚██████╔╝███████╗███████╗   ██║   ███████╗███████╗", fg='bright_cyan', bold=True) + click.style("      ▀█▄  ▄█▀", fg='bright_blue', bold=True))
-    click.echo(click.style("   ╚══════╝ ╚═════╝  ╚═════╝ ╚══════╝╚══════╝   ╚═╝   ╚══════╝╚══════╝", fg='bright_cyan', bold=True) + click.style("        ▀██▀", fg='bright_blue', bold=True))
+    click.echo(
+        click.style(
+            "   ███████╗ ██████╗ ██╗   ██╗██╗     ███████╗██╗   ██╗███████╗███████╗",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("        ▄██▄", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ██╔════╝██╔═══██╗██║   ██║██║     ██╔════╝╚██╗ ██╔╝██╔════╝╚══███╔╝",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("      ▄█▀  ▀█▄", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ███████╗██║   ██║██║   ██║██║     █████╗   ╚████╔╝ █████╗    ███╔╝ ",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("     █   ◉   █", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ╚════██║██║   ██║██║   ██║██║     ██╔══╝    ╚██╔╝  ██╔══╝   ███╔╝  ",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("     █  ═══  █", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ███████║╚██████╔╝╚██████╔╝███████╗███████╗   ██║   ███████╗███████╗",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("      ▀█▄  ▄█▀", fg="bright_blue", bold=True)
+    )
+    click.echo(
+        click.style(
+            "   ╚══════╝ ╚═════╝  ╚═════╝ ╚══════╝╚══════╝   ╚═╝   ╚══════╝╚══════╝",
+            fg="bright_cyan",
+            bold=True,
+        )
+        + click.style("        ▀██▀", fg="bright_blue", bold=True)
+    )
     click.echo()
 
     # Tagline and description
-    click.echo(click.style("  Created by CyberSoul SecurITy", fg='bright_blue'))
+    click.echo(click.style("  Created by CyberSoul SecurITy", fg="bright_blue"))
     click.echo()
-    click.echo("  SoulEyez brings your hacking tools together so you can spend less time switching windows")
-    click.echo("  and more time breaking things (ethically, of course). Launch scans with Nmap, Metasploit,")
-    click.echo("  Gobuster, theHarvester, and many more. Manage engagements, review findings, generate reports,")
+    click.echo(
+        "  SoulEyez brings your hacking tools together so you can spend less time switching windows"
+    )
+    click.echo(
+        "  and more time breaking things (ethically, of course). Launch scans with Nmap, Metasploit,"
+    )
+    click.echo(
+        "  Gobuster, theHarvester, and many more. Manage engagements, review findings, generate reports,"
+    )
     click.echo("  and let AI recommend your next moves — all in one place.")
     click.echo()
-    click.echo(click.style("  SETUP WIZARD", bold=True, fg='cyan'))
+    click.echo(click.style("  SETUP WIZARD", bold=True, fg="cyan"))
     click.echo("  ─────────────")
     click.echo()
     click.echo("  This wizard will help you get started:")
-    click.echo("  " + click.style("✓", fg='green') + " Set up encryption for credentials")
-    click.echo("  " + click.style("✓", fg='green') + " Create your first engagement")
-    click.echo("  " + click.style("✓", fg='green') + " Check installed tools")
-    click.echo("  " + click.style("✓", fg='green') + " Set up AI features (optional)")
+    click.echo(
+        "  " + click.style("✓", fg="green") + " Set up encryption for credentials"
+    )
+    click.echo("  " + click.style("✓", fg="green") + " Create your first engagement")
+    click.echo("  " + click.style("✓", fg="green") + " Check installed tools")
+    click.echo("  " + click.style("✓", fg="green") + " Set up AI features (optional)")
 
     # Only show Pro steps if user has Pro tier
     if is_pro:
-        click.echo("  " + click.style("✓", fg='green') + " Configure automation preferences")
-        click.echo("  " + click.style("✓", fg='green') + " Select report templates")
+        click.echo(
+            "  " + click.style("✓", fg="green") + " Configure automation preferences"
+        )
+        click.echo("  " + click.style("✓", fg="green") + " Select report templates")
 
     click.echo()
     click.pause("  Press ENTER to continue...")
@@ -485,38 +645,54 @@ def _wizard_encryption_setup() -> bool:
     width = 60
 
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" STEP 2: ENCRYPTION SETUP ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " STEP 2: ENCRYPTION SETUP ".center(width - 2), bold=True, fg="cyan"
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
     crypto = CryptoManager()
 
     if crypto.is_encryption_enabled():
-        click.echo("  " + click.style("✓ Encryption already configured", fg='green'))
+        click.echo("  " + click.style("✓ Encryption already configured", fg="green"))
         click.echo()
         click.pause("  Press any key to continue...")
         return True
 
     click.echo("  SoulEyez encrypts all credentials (passwords, API keys, tokens)")
-    click.echo("  with a master password. This is " + click.style("required", fg='green', bold=True) + " for security.")
+    click.echo(
+        "  with a master password. This is "
+        + click.style("required", fg="green", bold=True)
+        + " for security."
+    )
     click.echo()
-    click.echo("  " + click.style("[y]", fg='green') + " Enable encryption (recommended)")
-    click.echo("  " + click.style("[n]", fg='bright_black') + " Skip encryption")
+    click.echo(
+        "  " + click.style("[y]", fg="green") + " Enable encryption (recommended)"
+    )
+    click.echo("  " + click.style("[n]", fg="bright_black") + " Skip encryption")
     click.echo()
 
-    choice = click.prompt("  Enable encryption? [y/n]", default='y').strip().lower()
+    choice = click.prompt("  Enable encryption? [y/n]", default="y").strip().lower()
 
-    if choice in ('n', 's', 'no', 'skip'):
+    if choice in ("n", "s", "no", "skip"):
         click.echo()
         time.sleep(0.5)
-        click.echo("  " + click.style("SIKE! 😏", fg='magenta', bold=True))
+        click.echo("  " + click.style("SIKE! 😏", fg="magenta", bold=True))
         time.sleep(0.8)
         click.echo()
-        click.echo("  " + click.style("Security is not optional.", fg='cyan', bold=True))
+        click.echo(
+            "  " + click.style("Security is not optional.", fg="cyan", bold=True)
+        )
         click.echo("  " + "We're a security company. Encryption is mandatory.")
         click.echo()
         time.sleep(1)
-        click.echo("  " + click.style("Let's set up your vault password...", fg='green'))
+        click.echo(
+            "  " + click.style("Let's set up your vault password...", fg="green")
+        )
         click.pause("  Press any key to continue...")
 
     click.echo()
@@ -526,39 +702,63 @@ def _wizard_encryption_setup() -> bool:
     click.echo("    • At least one number")
     click.echo("    • At least one special character (!@#$%^&*)")
     click.echo()
-    click.echo("  " + click.style("⚠️  If you lose this password, encrypted credentials cannot be recovered!",
-                                  fg='yellow', bold=True))
+    click.echo(
+        "  "
+        + click.style(
+            "⚠️  If you lose this password, encrypted credentials cannot be recovered!",
+            fg="yellow",
+            bold=True,
+        )
+    )
     click.echo()
 
     import re
+
     password_set = False
     while not password_set:
         password = getpass.getpass("  Enter master password: ")
 
         # Validate password strength
         if len(password) < 12:
-            click.echo(click.style("  ✗ Password must be at least 12 characters.", fg='red'))
+            click.echo(
+                click.style("  ✗ Password must be at least 12 characters.", fg="red")
+            )
             continue
 
-        if not re.search(r'[a-z]', password):
-            click.echo(click.style("  ✗ Password must contain at least one lowercase letter.", fg='red'))
+        if not re.search(r"[a-z]", password):
+            click.echo(
+                click.style(
+                    "  ✗ Password must contain at least one lowercase letter.", fg="red"
+                )
+            )
             continue
 
-        if not re.search(r'[A-Z]', password):
-            click.echo(click.style("  ✗ Password must contain at least one uppercase letter.", fg='red'))
+        if not re.search(r"[A-Z]", password):
+            click.echo(
+                click.style(
+                    "  ✗ Password must contain at least one uppercase letter.", fg="red"
+                )
+            )
             continue
 
-        if not re.search(r'\d', password):
-            click.echo(click.style("  ✗ Password must contain at least one number.", fg='red'))
+        if not re.search(r"\d", password):
+            click.echo(
+                click.style("  ✗ Password must contain at least one number.", fg="red")
+            )
             continue
 
         if not re.search(r'[!@#$%^&*()_+\-=\[\]{};:\'",.<>?/\\|`~]', password):
-            click.echo(click.style("  ✗ Password must contain at least one special character.", fg='red'))
+            click.echo(
+                click.style(
+                    "  ✗ Password must contain at least one special character.",
+                    fg="red",
+                )
+            )
             continue
 
         password_confirm = getpass.getpass("  Confirm master password: ")
         if password != password_confirm:
-            click.echo(click.style("  ✗ Passwords don't match!", fg='red'))
+            click.echo(click.style("  ✗ Passwords don't match!", fg="red"))
             continue
 
         password_set = True
@@ -568,13 +768,13 @@ def _wizard_encryption_setup() -> bool:
 
     try:
         if crypto.enable_encryption(password):
-            click.echo("  " + click.style("✓ Encryption enabled!", fg='green'))
+            click.echo("  " + click.style("✓ Encryption enabled!", fg="green"))
         else:
-            click.echo("  " + click.style("✗ Failed to enable encryption!", fg='red'))
+            click.echo("  " + click.style("✗ Failed to enable encryption!", fg="red"))
             click.pause("  Press any key to continue...")
             return False
     except Exception as e:
-        click.echo("  " + click.style(f"✗ Error: {e}", fg='red'))
+        click.echo("  " + click.style(f"✗ Error: {e}", fg="red"))
         click.pause("  Press any key to continue...")
         return False
 
@@ -591,39 +791,55 @@ def _wizard_create_engagement() -> dict:
     width = 60
 
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" STEP 3: CREATE YOUR ENGAGEMENT ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " STEP 3: CREATE YOUR ENGAGEMENT ".center(width - 2), bold=True, fg="cyan"
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
     click.echo("  An engagement organizes all your work for a specific target.")
-    click.echo("  Give it a descriptive name like 'Acme Corp Pentest' or 'HackTheBox Lab'.")
+    click.echo(
+        "  Give it a descriptive name like 'Acme Corp Pentest' or 'HackTheBox Lab'."
+    )
     click.echo()
 
     name = click.prompt("  Engagement Name").strip()
 
     # Require a name
     while not name:
-        click.echo(click.style("  ✗ Name is required!", fg='red'))
+        click.echo(click.style("  ✗ Name is required!", fg="red"))
         name = click.prompt("  Engagement Name").strip()
 
     click.echo()
     click.echo("  " + click.style("Engagement Type:", bold=True))
     click.echo("    [1] Penetration Test  - Full-scope security assessment")
     click.echo("    [2] Bug Bounty        - Vulnerability hunting with defined scope")
-    click.echo("    [3] CTF / Lab         - Practice environment, aggressive scanning OK")
+    click.echo(
+        "    [3] CTF / Lab         - Practice environment, aggressive scanning OK"
+    )
     click.echo("    [4] Red Team          - Adversary simulation, stealth preferred")
     click.echo("    [5] Custom            - Define your own")
     click.echo()
-    click.echo("  " + click.style("NOTE:", fg='yellow') + " Type affects default automation and scan aggressiveness")
+    click.echo(
+        "  "
+        + click.style("NOTE:", fg="yellow")
+        + " Type affects default automation and scan aggressiveness"
+    )
     click.echo()
 
-    type_choice = click.prompt("  Select option", type=click.IntRange(1, 5), default=1, show_default=False)
+    type_choice = click.prompt(
+        "  Select option", type=click.IntRange(1, 5), default=1, show_default=False
+    )
 
     engagement_types = {
-        1: 'penetration_test',
-        2: 'bug_bounty',
-        3: 'ctf',
-        4: 'red_team',
-        5: 'custom'
+        1: "penetration_test",
+        2: "bug_bounty",
+        3: "ctf",
+        4: "red_team",
+        5: "custom",
     }
 
     engagement_type = engagement_types[type_choice]
@@ -633,35 +849,44 @@ def _wizard_create_engagement() -> dict:
 
     # Handle duplicate engagement names
     try:
-        engagement_id = em.create_engagement(name, f"Created via Setup Wizard - Type: {engagement_type}")
+        engagement_id = em.create_engagement(
+            name, f"Created via Setup Wizard - Type: {engagement_type}"
+        )
         click.echo()
-        click.echo("  " + click.style(f"✓ Engagement '{name}' created and set as active!", fg='green'))
+        click.echo(
+            "  "
+            + click.style(
+                f"✓ Engagement '{name}' created and set as active!", fg="green"
+            )
+        )
     except ValueError as e:
         if "already exists" in str(e):
             click.echo()
-            click.echo("  " + click.style(f"Engagement '{name}' already exists.", fg='yellow'))
-            click.echo("  " + click.style("✓ Using existing engagement.", fg='green'))
+            click.echo(
+                "  " + click.style(f"Engagement '{name}' already exists.", fg="yellow")
+            )
+            click.echo("  " + click.style("✓ Using existing engagement.", fg="green"))
             # Get existing engagement ID
             existing = em.get_engagement(name)
-            engagement_id = existing['id'] if existing else None
+            engagement_id = existing["id"] if existing else None
         else:
             raise
 
     em.set_current(name)  # Set as active so user drops into this engagement
     click.pause("  Press any key to continue...")
 
-    return {
-        'id': engagement_id,
-        'name': name,
-        'type': engagement_type
-    }
+    return {"id": engagement_id, "name": name, "type": engagement_type}
 
 
 def _wizard_tool_check() -> dict:
     """Check installed tools using the centralized tool_checker module."""
     from souleyez.utils.tool_checker import (
-        check_tool_version, EXTERNAL_TOOLS, get_install_command, detect_distro,
-        get_tool_version, get_upgrade_command
+        check_tool_version,
+        EXTERNAL_TOOLS,
+        get_install_command,
+        detect_distro,
+        get_tool_version,
+        get_upgrade_command,
     )
 
     DesignSystem.clear_screen()
@@ -670,7 +895,13 @@ def _wizard_tool_check() -> dict:
     distro = detect_distro()
 
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" STEP 4: TOOL AVAILABILITY ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " STEP 4: TOOL AVAILABILITY ".center(width - 2), bold=True, fg="cyan"
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
     click.echo("  Checking installed tools...")
@@ -679,17 +910,17 @@ def _wizard_tool_check() -> dict:
     # Core tools to check in the wizard (subset of full tool list)
     # Maps display name -> tool_checker key (category, tool_name)
     wizard_tools = {
-        'nmap': ('reconnaissance', 'nmap'),
-        'gobuster': ('web_scanning', 'gobuster'),
-        'sqlmap': ('exploitation', 'sqlmap'),
-        'nuclei': ('web_scanning', 'nuclei'),
-        'hydra': ('credential_attacks', 'hydra'),
-        'theHarvester': ('reconnaissance', 'theharvester'),
-        'wpscan': ('web_scanning', 'wpscan'),
-        'netexec': ('windows_ad', 'netexec'),
-        'enum4linux': ('windows_ad', 'enum4linux'),
-        'dnsrecon': ('reconnaissance', 'dnsrecon'),
-        'whois': ('reconnaissance', 'whois'),
+        "nmap": ("reconnaissance", "nmap"),
+        "gobuster": ("web_scanning", "gobuster"),
+        "sqlmap": ("exploitation", "sqlmap"),
+        "nuclei": ("web_scanning", "nuclei"),
+        "hydra": ("credential_attacks", "hydra"),
+        "theHarvester": ("reconnaissance", "theharvester"),
+        "wpscan": ("web_scanning", "wpscan"),
+        "netexec": ("windows_ad", "netexec"),
+        "enum4linux": ("windows_ad", "enum4linux"),
+        "dnsrecon": ("reconnaissance", "dnsrecon"),
+        "whois": ("reconnaissance", "whois"),
     }
 
     found = 0
@@ -700,58 +931,71 @@ def _wizard_tool_check() -> dict:
 
     for display_name, (category, tool_key) in wizard_tools.items():
         tool_info = EXTERNAL_TOOLS[category][tool_key]
-        command = tool_info['command']
+        command = tool_info["command"]
 
         # Use version-aware check
         version_status = check_tool_version(tool_info)
 
-        if version_status['installed']:
+        if version_status["installed"]:
             # Use the actual command that was found (may be primary or alt)
-            actual_cmd = version_status.get('actual_command') or command
+            actual_cmd = version_status.get("actual_command") or command
             path = shutil.which(actual_cmd)
 
             # Try to get version for display (even if no min_version requirement)
-            version_str = version_status['version']
+            version_str = version_status["version"]
             if not version_str:
                 # Try to get version using the actual found command
                 version_str = get_tool_version(actual_cmd)
 
             # Check if version is OK
-            if version_status['needs_upgrade']:
+            if version_status["needs_upgrade"]:
                 # Tool installed but wrong version
-                ver_display = version_str or 'unknown'
-                min_ver = version_status['min_version']
-                click.echo(f"  {click.style('!', fg='yellow')} {display_name:<15} v{ver_display} " +
-                          click.style(f"(needs v{min_ver}+)", fg='yellow'))
+                ver_display = version_str or "unknown"
+                min_ver = version_status["min_version"]
+                click.echo(
+                    f"  {click.style('!', fg='yellow')} {display_name:<15} v{ver_display} "
+                    + click.style(f"(needs v{min_ver}+)", fg="yellow")
+                )
                 tool_status[display_name] = {
-                    'found': True,
-                    'path': path,
-                    'version': ver_display,
-                    'needs_upgrade': True
+                    "found": True,
+                    "path": path,
+                    "version": ver_display,
+                    "needs_upgrade": True,
                 }
-                wrong_version_tools.append({
-                    'name': display_name,
-                    'installed': ver_display,
-                    'required': min_ver,
-                    'install': get_upgrade_command(tool_info, distro) or get_install_command(tool_info, distro),
-                    'note': version_status.get('version_note'),
-                    'tool_info': tool_info
-                })
+                wrong_version_tools.append(
+                    {
+                        "name": display_name,
+                        "installed": ver_display,
+                        "required": min_ver,
+                        "install": get_upgrade_command(tool_info, distro)
+                        or get_install_command(tool_info, distro),
+                        "note": version_status.get("version_note"),
+                        "tool_info": tool_info,
+                    }
+                )
                 found += 1  # Still counts as found, just needs upgrade
             else:
                 # Tool installed with correct version
                 ver_display = f"v{version_str}" if version_str else ""
-                click.echo(f"  {click.style('✓', fg='green')} {display_name:<15} Found {ver_display}")
-                tool_status[display_name] = {'found': True, 'path': path, 'version': version_str}
+                click.echo(
+                    f"  {click.style('✓', fg='green')} {display_name:<15} Found {ver_display}"
+                )
+                tool_status[display_name] = {
+                    "found": True,
+                    "path": path,
+                    "version": version_str,
+                }
                 found += 1
         else:
             click.echo(f"  {click.style('✗', fg='red')} {display_name:<15} NOT FOUND")
-            tool_status[display_name] = {'found': False, 'path': None}
-            missing_tools.append({
-                'name': display_name,
-                'install': get_install_command(tool_info, distro),
-                'tool_info': tool_info
-            })
+            tool_status[display_name] = {"found": False, "path": None}
+            missing_tools.append(
+                {
+                    "name": display_name,
+                    "install": get_install_command(tool_info, distro),
+                    "tool_info": tool_info,
+                }
+            )
 
     click.echo()
     click.echo(f"  Found {found}/{total} recommended tools")
@@ -759,10 +1003,12 @@ def _wizard_tool_check() -> dict:
     # Show version warnings
     if wrong_version_tools:
         click.echo()
-        click.echo("  " + click.style("VERSION ISSUES:", fg='yellow', bold=True))
+        click.echo("  " + click.style("VERSION ISSUES:", fg="yellow", bold=True))
         for tool in wrong_version_tools:
-            click.echo(f"    - {tool['name']}: installed v{tool['installed']}, needs v{tool['required']}+")
-            if tool.get('note'):
+            click.echo(
+                f"    - {tool['name']}: installed v{tool['installed']}, needs v{tool['required']}+"
+            )
+            if tool.get("note"):
                 click.echo(f"      {click.style(tool['note'], fg='bright_black')}")
 
     # Offer to install/upgrade tools
@@ -781,22 +1027,30 @@ def _wizard_tool_check() -> dict:
                 tool_info = EXTERNAL_TOOLS[category][tool_key]
                 version_status = check_tool_version(tool_info)
 
-                if version_status['installed'] and not version_status['needs_upgrade']:
-                    tool_status[display_name] = {'found': True, 'path': shutil.which(tool_info['command'])}
+                if version_status["installed"] and not version_status["needs_upgrade"]:
+                    tool_status[display_name] = {
+                        "found": True,
+                        "path": shutil.which(tool_info["command"]),
+                    }
                     found += 1
-                elif version_status['installed'] and version_status['needs_upgrade']:
-                    tool_status[display_name] = {'found': True, 'needs_upgrade': True}
-                    wrong_version_tools.append({'name': display_name})
+                elif version_status["installed"] and version_status["needs_upgrade"]:
+                    tool_status[display_name] = {"found": True, "needs_upgrade": True}
+                    wrong_version_tools.append({"name": display_name})
                     found += 1
                 else:
-                    tool_status[display_name] = {'found': False, 'path': None}
+                    tool_status[display_name] = {"found": False, "path": None}
 
             click.echo(f"  Now have {found}/{total} tools available")
 
     click.echo()
     click.pause("  Press any key to continue...")
 
-    return {'found': found, 'total': total, 'tools': tool_status, 'wrong_version': wrong_version_tools}
+    return {
+        "found": found,
+        "total": total,
+        "tools": tool_status,
+        "wrong_version": wrong_version_tools,
+    }
 
 
 def _wizard_ai_setup() -> bool:
@@ -809,54 +1063,83 @@ def _wizard_ai_setup() -> bool:
     width = 60
 
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" STEP 5: AI FEATURES (OPTIONAL) ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " STEP 5: AI FEATURES (OPTIONAL) ".center(width - 2), bold=True, fg="cyan"
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
     click.echo("  SoulEyez can use AI to suggest scanning strategies and")
-    click.echo("  prioritize targets. This requires " + click.style("Ollama", bold=True) + " (local AI runtime).")
+    click.echo(
+        "  prioritize targets. This requires "
+        + click.style("Ollama", bold=True)
+        + " (local AI runtime)."
+    )
     click.echo()
 
     # Check if Ollama is installed
-    ollama_path = shutil.which('ollama')
+    ollama_path = shutil.which("ollama")
 
     if ollama_path:
-        click.echo("  " + click.style("✓", fg='green') + " Ollama is installed")
+        click.echo("  " + click.style("✓", fg="green") + " Ollama is installed")
 
         # Check if it's running
         try:
             result = subprocess.run(
-                ['curl', '-s', 'http://localhost:11434/api/tags'],
-                capture_output=True, timeout=5
+                ["curl", "-s", "http://localhost:11434/api/tags"],
+                capture_output=True,
+                timeout=5,
             )
             if result.returncode == 0:
-                click.echo("  " + click.style("✓", fg='green') + " Ollama is running")
+                click.echo("  " + click.style("✓", fg="green") + " Ollama is running")
 
                 # Check for models
                 import json
+
                 try:
                     data = json.loads(result.stdout)
-                    models = data.get('models', [])
+                    models = data.get("models", [])
                     if models:
-                        click.echo(f"  " + click.style("✓", fg='green') + f" {len(models)} model(s) available")
+                        click.echo(
+                            f"  "
+                            + click.style("✓", fg="green")
+                            + f" {len(models)} model(s) available"
+                        )
                         click.echo()
-                        click.echo("  " + click.style("AI features are ready!", fg='green', bold=True))
+                        click.echo(
+                            "  "
+                            + click.style(
+                                "AI features are ready!", fg="green", bold=True
+                            )
+                        )
                         click.echo()
                         click.pause("  Press any key to continue...")
                         return True
                     else:
-                        click.echo("  " + click.style("!", fg='yellow') + " No models installed")
+                        click.echo(
+                            "  "
+                            + click.style("!", fg="yellow")
+                            + " No models installed"
+                        )
                 except:
                     pass
 
                 # Offer to pull a model
                 click.echo()
                 click.echo("  Would you like to download a model now?")
-                click.echo("  Recommended: " + click.style("llama3.1:8b", fg='cyan') + " (~4.7GB)")
+                click.echo(
+                    "  Recommended: "
+                    + click.style("llama3.1:8b", fg="cyan")
+                    + " (~4.7GB)"
+                )
                 click.echo()
 
                 try:
                     response = input("  Download model? (y/n): ").strip().lower()
-                    if response == 'y':
+                    if response == "y":
                         return _pull_ollama_model()
                 except (KeyboardInterrupt, EOFError):
                     click.echo("\n  Skipped.")
@@ -865,11 +1148,15 @@ def _wizard_ai_setup() -> bool:
                 return False
             else:
                 # Ollama installed but not running
-                click.echo("  " + click.style("!", fg='yellow') + " Ollama is not running")
+                click.echo(
+                    "  " + click.style("!", fg="yellow") + " Ollama is not running"
+                )
                 return _start_ollama_and_setup()
 
         except subprocess.TimeoutExpired:
-            click.echo("  " + click.style("!", fg='yellow') + " Ollama is not responding")
+            click.echo(
+                "  " + click.style("!", fg="yellow") + " Ollama is not responding"
+            )
             return _start_ollama_and_setup()
         except FileNotFoundError:
             # curl not found, try another method
@@ -877,7 +1164,7 @@ def _wizard_ai_setup() -> bool:
 
     else:
         # Ollama not installed - offer to install
-        click.echo("  " + click.style("!", fg='yellow') + " Ollama is not installed")
+        click.echo("  " + click.style("!", fg="yellow") + " Ollama is not installed")
         click.echo()
         click.echo("  Would you like to install Ollama now?")
         click.echo("  This will download and install the Ollama runtime (~100MB).")
@@ -885,13 +1172,17 @@ def _wizard_ai_setup() -> bool:
 
         try:
             response = input("  Install Ollama? (y/n): ").strip().lower()
-            if response == 'y':
+            if response == "y":
                 return _install_ollama()
         except (KeyboardInterrupt, EOFError):
             click.echo("\n  Skipped.")
 
     click.echo()
-    click.echo("  " + click.style("Skipping AI setup.", fg='bright_black') + " You can enable it later in Settings.")
+    click.echo(
+        "  "
+        + click.style("Skipping AI setup.", fg="bright_black")
+        + " You can enable it later in Settings."
+    )
     click.pause("  Press any key to continue...")
     return False
 
@@ -902,28 +1193,45 @@ def _install_ollama() -> bool:
 
     click.echo()
     click.echo("  Installing Ollama...")
-    click.echo("  " + click.style("(This may take a minute - downloading ~100MB)", fg='bright_black'))
+    click.echo(
+        "  "
+        + click.style(
+            "(This may take a minute - downloading ~100MB)", fg="bright_black"
+        )
+    )
     click.echo()
 
     try:
         # Run the official Ollama install script with timeout
         # Add curl timeouts to fail faster on network issues
         result = subprocess.run(
-            ['bash', '-c', 'curl --connect-timeout 30 --max-time 300 -fsSL https://ollama.ai/install.sh | sh'],
+            [
+                "bash",
+                "-c",
+                "curl --connect-timeout 30 --max-time 300 -fsSL https://ollama.ai/install.sh | sh",
+            ],
             check=False,
-            timeout=360  # 6 minute total timeout
+            timeout=360,  # 6 minute total timeout
         )
 
         if result.returncode == 0:
             click.echo()
-            click.echo("  " + click.style("✓ Ollama installed successfully!", fg='green'))
+            click.echo(
+                "  " + click.style("✓ Ollama installed successfully!", fg="green")
+            )
 
             # Start Ollama and set up model
             return _start_ollama_and_setup()
         else:
             click.echo()
-            click.echo("  " + click.style("✗ Installation failed", fg='red'))
-            click.echo("  " + click.style("This is usually a network issue (slow connection or timeout).", fg='yellow'))
+            click.echo("  " + click.style("✗ Installation failed", fg="red"))
+            click.echo(
+                "  "
+                + click.style(
+                    "This is usually a network issue (slow connection or timeout).",
+                    fg="yellow",
+                )
+            )
             click.echo()
             click.echo("  To install manually:")
             click.echo("    curl -fsSL https://ollama.ai/install.sh | sh")
@@ -934,8 +1242,14 @@ def _install_ollama() -> bool:
 
     except subprocess.TimeoutExpired:
         click.echo()
-        click.echo("  " + click.style("✗ Installation timed out", fg='red'))
-        click.echo("  " + click.style("The download took too long. Check your internet connection.", fg='yellow'))
+        click.echo("  " + click.style("✗ Installation timed out", fg="red"))
+        click.echo(
+            "  "
+            + click.style(
+                "The download took too long. Check your internet connection.",
+                fg="yellow",
+            )
+        )
         click.echo()
         click.echo("  To install manually:")
         click.echo("    curl -fsSL https://ollama.ai/install.sh | sh")
@@ -944,7 +1258,7 @@ def _install_ollama() -> bool:
 
     except Exception as e:
         click.echo()
-        click.echo(click.style(f"  ✗ Error: {e}", fg='red'))
+        click.echo(click.style(f"  ✗ Error: {e}", fg="red"))
         click.echo("  You can install manually from https://ollama.ai")
         click.pause("  Press any key to continue...")
         return False
@@ -961,10 +1275,10 @@ def _start_ollama_and_setup() -> bool:
     try:
         # Start ollama serve in background
         subprocess.Popen(
-            ['ollama', 'serve'],
+            ["ollama", "serve"],
             stdout=subprocess.DEVNULL,
             stderr=subprocess.DEVNULL,
-            start_new_session=True
+            start_new_session=True,
         )
 
         # Wait for it to start
@@ -973,22 +1287,25 @@ def _start_ollama_and_setup() -> bool:
 
         # Verify it's running
         result = subprocess.run(
-            ['curl', '-s', 'http://localhost:11434/api/tags'],
-            capture_output=True, timeout=5
+            ["curl", "-s", "http://localhost:11434/api/tags"],
+            capture_output=True,
+            timeout=5,
         )
 
         if result.returncode == 0:
-            click.echo("  " + click.style("✓ Ollama started!", fg='green'))
+            click.echo("  " + click.style("✓ Ollama started!", fg="green"))
 
             # Offer to pull model
             click.echo()
             click.echo("  Would you like to download a model now?")
-            click.echo("  Recommended: " + click.style("llama3.1:8b", fg='cyan') + " (~4.7GB)")
+            click.echo(
+                "  Recommended: " + click.style("llama3.1:8b", fg="cyan") + " (~4.7GB)"
+            )
             click.echo()
 
             try:
                 response = input("  Download model? (y/n): ").strip().lower()
-                if response == 'y':
+                if response == "y":
                     return _pull_ollama_model()
             except (KeyboardInterrupt, EOFError):
                 click.echo("\n  Skipped.")
@@ -996,12 +1313,12 @@ def _start_ollama_and_setup() -> bool:
             click.pause("  Press any key to continue...")
             return True
         else:
-            click.echo("  " + click.style("✗ Failed to start Ollama", fg='red'))
+            click.echo("  " + click.style("✗ Failed to start Ollama", fg="red"))
             click.pause("  Press any key to continue...")
             return False
 
     except Exception as e:
-        click.echo(click.style(f"  ✗ Error: {e}", fg='red'))
+        click.echo(click.style(f"  ✗ Error: {e}", fg="red"))
         click.pause("  Press any key to continue...")
         return False
 
@@ -1014,32 +1331,38 @@ def _pull_ollama_model() -> bool:
 
     click.echo()
     click.echo(f"  Downloading {model}...")
-    click.echo("  " + click.style("(This may take several minutes depending on your connection)", fg='bright_black'))
+    click.echo(
+        "  "
+        + click.style(
+            "(This may take several minutes depending on your connection)",
+            fg="bright_black",
+        )
+    )
     click.echo()
 
     try:
         # Run ollama pull and let it output directly to terminal for proper progress bar
-        result = subprocess.run(
-            ['ollama', 'pull', model],
-            check=False
-        )
+        result = subprocess.run(["ollama", "pull", model], check=False)
 
         if result.returncode == 0:
             click.echo()
-            click.echo("  " + click.style(f"✓ Model {model} ready!", fg='green'))
+            click.echo("  " + click.style(f"✓ Model {model} ready!", fg="green"))
             click.echo()
-            click.echo("  " + click.style("AI features are now enabled!", fg='green', bold=True))
+            click.echo(
+                "  "
+                + click.style("AI features are now enabled!", fg="green", bold=True)
+            )
             click.pause("  Press any key to continue...")
             return True
         else:
             click.echo()
-            click.echo("  " + click.style("✗ Failed to download model", fg='red'))
+            click.echo("  " + click.style("✗ Failed to download model", fg="red"))
             click.echo("  You can try later with: ollama pull " + model)
             click.pause("  Press any key to continue...")
             return False
 
     except Exception as e:
-        click.echo(click.style(f"  ✗ Error: {e}", fg='red'))
+        click.echo(click.style(f"  ✗ Error: {e}", fg="red"))
         click.pause("  Press any key to continue...")
         return False
 
@@ -1051,7 +1374,13 @@ def _wizard_automation_prefs() -> dict:
     width = 60
 
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" STEP 6: AUTOMATION PREFERENCES ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " STEP 6: AUTOMATION PREFERENCES ".center(width - 2), bold=True, fg="cyan"
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
     click.echo("  SoulEyez can automatically chain tools based on discoveries.")
@@ -1069,15 +1398,17 @@ def _wizard_automation_prefs() -> dict:
         click.echo()
         click.echo(f"  {click.style('✓', fg='green')} Auto-chaining disabled")
         click.pause("  Press any key to continue...")
-        return {'enabled': False, 'mode': None}
+        return {"enabled": False, "mode": None}
 
-    mode = 'auto' if mode_choice == 1 else 'approval'
+    mode = "auto" if mode_choice == 1 else "approval"
 
     click.echo()
-    click.echo(f"  {click.style('✓', fg='green')} Auto-chaining enabled in {mode.upper()} mode")
+    click.echo(
+        f"  {click.style('✓', fg='green')} Auto-chaining enabled in {mode.upper()} mode"
+    )
     click.pause("  Press any key to continue...")
 
-    return {'enabled': True, 'mode': mode}
+    return {"enabled": True, "mode": mode}
 
 
 def _wizard_deliverables(engagement_type: str) -> list:
@@ -1087,25 +1418,39 @@ def _wizard_deliverables(engagement_type: str) -> list:
     width = 60
 
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" STEP 7: REPORT TEMPLATES ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            " STEP 7: REPORT TEMPLATES ".center(width - 2), bold=True, fg="cyan"
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
     click.echo("  Pre-select report templates that will be available for export.")
     click.echo("  You can change these later in Reports & Export menu.")
     click.echo()
-    click.echo("  " + click.style("NOTE:", fg='yellow') + " Templates help generate professional reports from your findings")
+    click.echo(
+        "  "
+        + click.style("NOTE:", fg="yellow")
+        + " Templates help generate professional reports from your findings"
+    )
     click.echo()
 
     # Default templates based on engagement type
     type_templates = {
-        'penetration_test': ['Executive Summary', 'Technical Findings Report', 'Vulnerability Details'],
-        'bug_bounty': ['Vulnerability Details', 'Proof of Concept'],
-        'ctf': ['Technical Findings Report', 'Attack Narrative'],
-        'red_team': ['Attack Narrative', 'Remediation Roadmap'],
-        'custom': ['Technical Findings Report']
+        "penetration_test": [
+            "Executive Summary",
+            "Technical Findings Report",
+            "Vulnerability Details",
+        ],
+        "bug_bounty": ["Vulnerability Details", "Proof of Concept"],
+        "ctf": ["Technical Findings Report", "Attack Narrative"],
+        "red_team": ["Attack Narrative", "Remediation Roadmap"],
+        "custom": ["Technical Findings Report"],
     }
 
-    recommended = type_templates.get(engagement_type, ['Technical Findings Report'])
+    recommended = type_templates.get(engagement_type, ["Technical Findings Report"])
 
     click.echo(f"  Recommended for {engagement_type.replace('_', ' ').title()}:")
     for template in recommended:
@@ -1118,7 +1463,15 @@ def _wizard_deliverables(engagement_type: str) -> list:
     return recommended
 
 
-def _wizard_summary(encryption_enabled, engagement_info, tool_status, ai_enabled, automation_prefs, templates, is_pro=False):
+def _wizard_summary(
+    encryption_enabled,
+    engagement_info,
+    tool_status,
+    ai_enabled,
+    automation_prefs,
+    templates,
+    is_pro=False,
+):
     """Show wizard summary."""
     DesignSystem.clear_screen()
     _show_wizard_banner()
@@ -1127,43 +1480,65 @@ def _wizard_summary(encryption_enabled, engagement_info, tool_status, ai_enabled
     # Adjust step number based on tier
     step_num = "8" if is_pro else "6"
     click.echo("┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(f" STEP {step_num}: SETUP COMPLETE! ".center(width - 2), bold=True, fg='green') + "│")
+    click.echo(
+        "│"
+        + click.style(
+            f" STEP {step_num}: SETUP COMPLETE! ".center(width - 2),
+            bold=True,
+            fg="green",
+        )
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
     # Summary
-    enc_status = click.style("Enabled", fg='green') if encryption_enabled else click.style("Disabled", fg='yellow')
+    enc_status = (
+        click.style("Enabled", fg="green")
+        if encryption_enabled
+        else click.style("Disabled", fg="yellow")
+    )
     click.echo(f"  {click.style('✓', fg='green')} Encryption: {enc_status}")
 
-    eng_name = engagement_info['name']
-    eng_type = engagement_info['type'].replace('_', ' ').title()
-    click.echo(f"  {click.style('✓', fg='green')} Engagement: \"{eng_name}\" ({eng_type})")
+    eng_name = engagement_info["name"]
+    eng_type = engagement_info["type"].replace("_", " ").title()
+    click.echo(
+        f"  {click.style('✓', fg='green')} Engagement: \"{eng_name}\" ({eng_type})"
+    )
 
-    tools_found = tool_status['found']
-    tools_total = tool_status['total']
-    click.echo(f"  {click.style('✓', fg='green')} Tools: {tools_found}/{tools_total} available")
+    tools_found = tool_status["found"]
+    tools_total = tool_status["total"]
+    click.echo(
+        f"  {click.style('✓', fg='green')} Tools: {tools_found}/{tools_total} available"
+    )
 
     # AI status
     if ai_enabled:
         click.echo(f"  {click.style('✓', fg='green')} AI Features: Enabled (Ollama)")
     else:
-        click.echo(f"  {click.style('○', fg='bright_black')} AI Features: Not configured")
+        click.echo(
+            f"  {click.style('○', fg='bright_black')} AI Features: Not configured"
+        )
 
     # Show tier status
     if is_pro:
         click.echo(f"  {click.style('💎', fg='magenta')} License: PRO")
-        if automation_prefs['enabled']:
-            auto_mode = automation_prefs['mode'].upper()
-            click.echo(f"  {click.style('✓', fg='green')} Auto-Chain: ON ({auto_mode} mode)")
+        if automation_prefs["enabled"]:
+            auto_mode = automation_prefs["mode"].upper()
+            click.echo(
+                f"  {click.style('✓', fg='green')} Auto-Chain: ON ({auto_mode} mode)"
+            )
         else:
             click.echo(f"  {click.style('✓', fg='green')} Auto-Chain: OFF")
 
         if templates:
-            click.echo(f"  {click.style('✓', fg='green')} Templates: {len(templates)} selected")
+            click.echo(
+                f"  {click.style('✓', fg='green')} Templates: {len(templates)} selected"
+            )
     else:
         click.echo(f"  {click.style('○', fg='bright_black')} License: FREE")
         click.echo()
-        click.echo(click.style("  Upgrade to Pro for:", fg='yellow'))
+        click.echo(click.style("  Upgrade to Pro for:", fg="yellow"))
         click.echo("    • AI Execute - Autonomous exploitation")
         click.echo("    • Automation - Smart tool chaining")
         click.echo("    • MSF Integration - Advanced attack chains")
@@ -1171,7 +1546,7 @@ def _wizard_summary(encryption_enabled, engagement_info, tool_status, ai_enabled
         click.echo(f"    {click.style('→ cybersoulsecurity.com/upgrade', fg='cyan')}")
 
     click.echo()
-    click.echo("  " + click.style("You're ready to start!", bold=True, fg='cyan'))
+    click.echo("  " + click.style("You're ready to start!", bold=True, fg="cyan"))
     click.echo()
 
     # Install desktop shortcut automatically
@@ -1179,22 +1554,41 @@ def _wizard_summary(encryption_enabled, engagement_info, tool_status, ai_enabled
 
     # Prompt for interactive tutorial
     click.echo("  ┌" + "─" * 56 + "┐")
-    click.echo("  │" + click.style(" Would you like to run the interactive tutorial?", fg='cyan').center(65) + "│")
-    click.echo("  │" + " Recommended for new users - takes about 5 minutes".center(56) + "│")
+    click.echo(
+        "  │"
+        + click.style(
+            " Would you like to run the interactive tutorial?", fg="cyan"
+        ).center(65)
+        + "│"
+    )
+    click.echo(
+        "  │" + " Recommended for new users - takes about 5 minutes".center(56) + "│"
+    )
     click.echo("  └" + "─" * 56 + "┘")
     click.echo()
-    click.echo("  " + click.style("[Y]", fg='green', bold=True) + " Yes, show me around")
-    click.echo("  " + click.style("[n]", fg='bright_black') + " No, go to main menu")
+    click.echo(
+        "  " + click.style("[Y]", fg="green", bold=True) + " Yes, show me around"
+    )
+    click.echo("  " + click.style("[n]", fg="bright_black") + " No, go to main menu")
     click.echo()
 
-    choice = click.prompt("  Run tutorial?", default='y', show_default=False).strip().lower()
+    choice = (
+        click.prompt("  Run tutorial?", default="y", show_default=False).strip().lower()
+    )
 
-    if choice in ('y', 'yes', ''):
+    if choice in ("y", "yes", ""):
         from souleyez.ui.tutorial import run_tutorial
+
         run_tutorial()
     else:
         click.echo()
-        click.echo("  " + click.style("No problem!", fg='cyan') + " You can run the tutorial anytime from:")
-        click.echo("  " + click.style("Settings & Security → [t] Tutorial", fg='yellow'))
+        click.echo(
+            "  "
+            + click.style("No problem!", fg="cyan")
+            + " You can run the tutorial anytime from:"
+        )
+        click.echo(
+            "  " + click.style("Settings & Security → [t] Tutorial", fg="yellow")
+        )
         click.echo()
         click.pause("  Press any key to continue to main menu...")