codeaois 0.1.0__py3-none-any.whl

codeaois/agents/3d_agent.py

@@ -0,0 +1,23 @@
+# codeaois/agents/3d_agent.py
+from codeaois.models.llm_interface import call_openrouter
+
+def generate_3d_code(user_prompt: str, file_context: str = "") -> str:
+    """Specialized 3D Agent worker."""
+    system_prompt = (
+        "You are the CodeAOIS 3D Agent. "
+        "Your job is to provide highly optimized, expert-level code strictly for your specific domain. "
+        "CRITICAL: Output ONLY the raw, complete code. Do not wrap the code in markdown blocks."
+    )
+    
+    full_prompt = user_prompt
+    if file_context:
+        full_prompt += f"\n\nHere are the existing files for context:\n{file_context}"
+        
+    response = call_openrouter(system_prompt, full_prompt)
+    
+    if response.startswith("```"):
+        lines = response.split("\n")
+        if len(lines) > 2:
+            response = "\n".join(lines[1:-1])
+            
+    return response.strip()

codeaois/agents/coder_agent.py

@@ -0,0 +1,25 @@
+# codeaois/agents/coder_agent.py
+from codeaois.models.llm_interface import call_openrouter
+
+def generate_code(user_prompt: str, file_context: str = "") -> str:
+    if file_context:
+        system_prompt = (
+            "You are the CodeAOIS Coder Agent. "
+            "1. You MUST output your code changes using the <<<<<<< SEARCH and >>>>>>> REPLACE format.\n"
+            "2. AFTER your code blocks, you MUST add the exact text '---SUMMARY---' on a new line.\n"
+            "3. AFTER the summary delimiter, write a brief, friendly explanation of how the code works and what you changed."
+        )
+    else:
+        system_prompt = (
+            "You are the CodeAOIS Coder Agent, an expert senior software engineer. "
+            "1. Output the raw, complete code first. Do not wrap the code in markdown blocks.\n"
+            "2. AFTER the code, you MUST add the exact text '---SUMMARY---' on a new line.\n"
+            "3. AFTER the summary delimiter, write a brief, friendly explanation of how the code works."
+        )
+    
+    full_prompt = user_prompt
+    if file_context:
+        full_prompt += f"\n\nHere are the existing files for context:\n{file_context}"
+        
+    response = call_openrouter(system_prompt, full_prompt, intent="code")
+    return response.strip()

codeaois/agents/data_science_agent.py

@@ -0,0 +1,24 @@
+# codeaois/agents/data_science_agent.py
+from codeaois.models.llm_interface import call_openrouter
+
+def generate_ds_code(user_prompt: str, file_context: str = "") -> str:
+    system_prompt = (
+        "You are the CodeAOIS Data Science Agent, an elite data scientist and Python developer. "
+        "Your job is to provide highly optimized, memory-efficient code for data manipulation, "
+        "analysis, and visualization. "
+        "CRITICAL: Output ONLY the raw, complete code. Do not wrap the code in markdown blocks."
+    )
+    
+    full_prompt = user_prompt
+    if file_context:
+        full_prompt += f"\n\nHere are the existing files for context. Modify as requested:\n{file_context}"
+        
+    # Pass the data_science intent to trigger the Qwen/Step model list
+    response = call_openrouter(system_prompt, full_prompt, intent="data_science")
+    
+    if response.startswith("```"):
+        lines = response.split("\n")
+        if len(lines) > 2:
+            response = "\n".join(lines[1:-1])
+            
+    return response.strip()

codeaois/agents/database_agent.py

@@ -0,0 +1,23 @@
+# codeaois/agents/database_agent.py
+from codeaois.models.llm_interface import call_openrouter
+
+def generate_database_code(user_prompt: str, file_context: str = "") -> str:
+    """Specialized Database Agent worker."""
+    system_prompt = (
+        "You are the CodeAOIS Database Agent. "
+        "Your job is to provide highly optimized, expert-level code strictly for your specific domain. "
+        "CRITICAL: Output ONLY the raw, complete code. Do not wrap the code in markdown blocks."
+    )
+    
+    full_prompt = user_prompt
+    if file_context:
+        full_prompt += f"\n\nHere are the existing files for context:\n{file_context}"
+        
+    response = call_openrouter(system_prompt, full_prompt)
+    
+    if response.startswith("```"):
+        lines = response.split("\n")
+        if len(lines) > 2:
+            response = "\n".join(lines[1:-1])
+            
+    return response.strip()

codeaois/agents/debugging_agent.py

@@ -0,0 +1,23 @@
+# codeaois/agents/debugging_agent.py
+from codeaois.models.llm_interface import call_openrouter
+
+def generate_debugging_code(user_prompt: str, file_context: str = "") -> str:
+    """Specialized Debugging Agent worker."""
+    system_prompt = (
+        "You are the CodeAOIS Debugging Agent. "
+        "Your job is to provide highly optimized, expert-level code strictly for your specific domain. "
+        "CRITICAL: Output ONLY the raw, complete code. Do not wrap the code in markdown blocks."
+    )
+    
+    full_prompt = user_prompt
+    if file_context:
+        full_prompt += f"\n\nHere are the existing files for context:\n{file_context}"
+        
+    response = call_openrouter(system_prompt, full_prompt)
+    
+    if response.startswith("```"):
+        lines = response.split("\n")
+        if len(lines) > 2:
+            response = "\n".join(lines[1:-1])
+            
+    return response.strip()

codeaois/agents/devops_agent.py

@@ -0,0 +1,23 @@
+# codeaois/agents/devops_agent.py
+from codeaois.models.llm_interface import call_openrouter
+
+def generate_devops_code(user_prompt: str, file_context: str = "") -> str:
+    """Specialized Devops Agent worker."""
+    system_prompt = (
+        "You are the CodeAOIS Devops Agent. "
+        "Your job is to provide highly optimized, expert-level code strictly for your specific domain. "
+        "CRITICAL: Output ONLY the raw, complete code. Do not wrap the code in markdown blocks."
+    )
+    
+    full_prompt = user_prompt
+    if file_context:
+        full_prompt += f"\n\nHere are the existing files for context:\n{file_context}"
+        
+    response = call_openrouter(system_prompt, full_prompt)
+    
+    if response.startswith("```"):
+        lines = response.split("\n")
+        if len(lines) > 2:
+            response = "\n".join(lines[1:-1])
+            
+    return response.strip()

codeaois/agents/git_agent.py

@@ -0,0 +1,45 @@
+# codeaois/agents/git_agent.py
+import subprocess
+import os
+
+C_YELLOW = '\033[93m'
+C_GREEN = '\033[92m'
+C_RESET = '\033[0m'
+
+def run_git_command(command: list) -> tuple[bool, str]:
+    """Quietly executes a terminal command and returns the result."""
+    try:
+        result = subprocess.run(command, capture_output=True, text=True, check=True)
+        return True, result.stdout.strip()
+    except subprocess.CalledProcessError as e:
+        return False, e.stderr.strip()
+
+def auto_commit(file_path: str, message: str):
+    """Automatically stages and commits a modified file."""
+    
+    # 1. Check if the user is inside a Git repository
+    is_git, _ = run_git_command(["git", "rev-parse", "--is-inside-work-tree"])
+    if not is_git:
+        print(f"   {C_YELLOW}⚠️ [Git Agent] No .git repository found in this folder. Skipping auto-commit.{C_RESET}")
+        return
+
+    print(f"   {C_GREEN}🌿 [Git Agent] Engaging version control...{C_RESET}")
+
+    # 2. Stage only the specific file the AI just modified
+    success, err = run_git_command(["git", "add", file_path])
+    if not success:
+        print(f"   {C_YELLOW}⚠️ [Git Agent] Failed to stage file: {err}{C_RESET}")
+        return
+
+    # 3. Double-check that there are actually changes to save
+    has_changes, diff = run_git_command(["git", "status", "--porcelain"])
+    if not has_changes or not diff:
+        print(f"   {C_YELLOW}⚠️ [Git Agent] No actual code changes detected for {file_path}.{C_RESET}")
+        return
+
+    # 4. Commit the changes to the timeline
+    commit_success, commit_err = run_git_command(["git", "commit", "-m", message])
+    if commit_success:
+        print(f"   {C_GREEN}✅ [Git Agent] Successfully committed changes to {file_path}!{C_RESET}")
+    else:
+        print(f"   {C_YELLOW}⚠️ [Git Agent] Commit failed: {commit_err}{C_RESET}")

codeaois/agents/tester_agent.py

@@ -0,0 +1,27 @@
+# codeaois/agents/tester_agent.py
+import subprocess
+import sys
+
+def run_test(file_path: str) -> tuple[bool, str]:
+    """Runs a Python script and catches any terminal errors."""
+    if not file_path.endswith('.py'):
+        return True, "Not a Python file. Skipping auto-test."
+
+    try:
+        # sys.executable ensures it uses your isolated 'venv' Python, not the global Ubuntu one!
+        result = subprocess.run(
+            [sys.executable, file_path],
+            capture_output=True,
+            text=True,
+            timeout=10 # 10-second kill switch so infinite loops don't crash your computer
+        )
+        
+        if result.returncode == 0:
+            return True, result.stdout.strip()
+        else:
+            return False, result.stderr.strip()
+            
+    except subprocess.TimeoutExpired:
+        return False, "Error: Script execution timed out (possible infinite loop)."
+    except Exception as e:
+        return False, str(e)

codeaois/app.py

@@ -0,0 +1 @@
+# AI-generated code for: Modify `app.py` with instruction: create Flask API, preserving style and existing code

codeaois/brain/embeddings.py

@@ -0,0 +1,27 @@
+import os
+import numpy as np
+from sentence_transformers import SentenceTransformer
+
+# Lightweight model for semantic search
+model = SentenceTransformer("all-MiniLM-L6-v2")
+
+class ProjectEmbeddings:
+    def __init__(self):
+        self.index = {}  # {filepath: embedding}
+
+    def build_index(self, project_path="."):
+        for root, dirs, files in os.walk(project_path):
+            dirs[:] = [d for d in dirs if d not in {"aois_env", ".git", "__pycache__", "node_modules"}]
+            for f in files:
+                if f.endswith(".py"):
+                    path = os.path.join(root, f)
+                    with open(path, "r", encoding="utf-8") as file:
+                        content = file.read()
+                    embedding = model.encode(content)
+                    self.index[path] = embedding
+
+    def search(self, query, top_k=3):
+        q_vec = model.encode(query)
+        distances = [(f, np.dot(q_vec, emb)) for f, emb in self.index.items()]
+        distances.sort(key=lambda x: x[1], reverse=True)
+        return distances[:top_k]

codeaois/brain/memory.py

@@ -0,0 +1,10 @@
+class ProjectMemory:
+
+    def __init__(self):
+        self.data = {}
+
+    def store(self, key, value):
+        self.data[key] = value
+
+    def retrieve(self, key):
+        return self.data.get(key)

codeaois/brain/scanner.py

@@ -0,0 +1,49 @@
+# codeaois/brain/scanner.py
+import os
+
+def scan_project_structure(root_dir: str = ".", ignore_dirs: list = None, max_depth: int = 3, max_files: int = 200) -> str:
+    """
+    Goal #9: Project Brain (with safety limits).
+    Scans the folder structure but strictly limits depth and file count 
+    to prevent network timeouts when run in massive directories like '~'.
+    """
+    if ignore_dirs is None:
+        # Added OS-level hidden folders to ignore list
+        ignore_dirs = ['.git', '__pycache__', 'venv', 'env', 'node_modules', '.venv', 'codeaois.egg-info', '.cache', '.config', '.local']
+
+    tree_str = f"📁 Project Root: {os.path.abspath(root_dir)}\n"
+    file_count = 0
+    
+    # Calculate starting depth
+    start_level = root_dir.rstrip(os.sep).count(os.sep)
+
+    for dirpath, dirnames, filenames in os.walk(root_dir):
+        # Calculate current depth relative to start
+        current_level = dirpath.rstrip(os.sep).count(os.sep)
+        level = current_level - start_level
+        
+        # Stop digging if we hit our depth limit
+        if level > max_depth:
+            dirnames[:] = [] # Clear the list to stop os.walk from going deeper
+            continue
+
+        dirnames[:] = [d for d in dirnames if d not in ignore_dirs and not d.startswith('.')]
+        
+        indent = ' ' * 4 * level
+        folder_name = os.path.basename(dirpath)
+        
+        if level > 0:
+            tree_str += f"{indent}📂 {folder_name}/\n"
+            
+        sub_indent = ' ' * 4 * (level + 1)
+        for f in filenames:
+            # Emergency Stop: Prevent massive payloads from timing out the API
+            if file_count >= max_files:
+                tree_str += f"{sub_indent}... [Max file limit reached to protect API connection]\n"
+                return tree_str 
+                
+            if f != '.env' and not f.startswith('.'):
+                tree_str += f"{sub_indent}📄 {f}\n"
+                file_count += 1
+                
+    return tree_str

codeaois/cli/main.py

@@ -0,0 +1,351 @@
+# codeaois/cli/main.py
+import argparse
+import sys
+import os
+import importlib 
+
+# --- CLI UI ENGINES ---
+from prompt_toolkit import PromptSession
+from prompt_toolkit.completion import WordCompleter
+from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
+from prompt_toolkit.formatted_text import HTML
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.table import Table # <-- ADD THIS!
+
+# --- CODEAOIS MODULES ---
+from codeaois.core.planner import analyze_intent, get_installed_agents
+from codeaois.core.context import extract_file_context
+from codeaois.brain.scanner import scan_project_structure
+from codeaois.models.llm_interface import call_openrouter
+from codeaois.agents.coder_agent import generate_code
+from codeaois.agents.data_science_agent import generate_ds_code
+from codeaois.utils.file_writer import extract_and_save_code
+from codeaois.agents.git_agent import auto_commit
+from codeaois.core.marketplace import install_agent
+from codeaois.core.memory import load_profile, create_profile, load_history, save_history, clear_history_data, clear_profile_data
+from codeaois.agents.tester_agent import run_test
+
+# Print Colors
+C_BLUE = '\033[94m'
+C_GREEN = '\033[92m'
+C_YELLOW = '\033[93m'
+C_PURPLE = '\033[95m'
+C_CYAN = '\033[96m'
+C_RESET = '\033[0m'
+C_BOLD = '\033[1m'
+
+# Globals
+console = Console()
+chat_history = load_history()
+active_file = None  # The OS remembers the file you are working on
+
+# --- CONFIGURE AUTOCOMPLETE OPTIONS ---
+AVAILABLE_COMMANDS = [
+    '/help', '/clear', '/clear_history', '/clear_user', 
+    '/status', '/marketplace', '/install', '/exit', '/quit'
+]
+command_completer = WordCompleter(AVAILABLE_COMMANDS, ignore_case=True)
+
+def print_logo():
+    logo = f"""{C_BLUE}{C_BOLD}
+   ____          _        _    ___ _____  _____ 
+  / ___|___   __| | ___  / \\  / _ \\_   _|/ ____|
+ | |   / _ \\ / _` |/ _ \\/ _ \\| | | || | | (___  
+ | |__| (_) | (_| |  __/ ___ \\ |_| || |_ \\___ \\ 
+  \\____\\___/ \\__,_|\\___/_/   \\_\\___/_____|____/ 
+{C_RESET}"""
+    print(logo)
+    print(f"{C_CYAN}  ▶ Advanced AI Developer OS v0.1.0{C_RESET}")
+    print(f"{C_CYAN}  ▶ Type /help to see available commands.{C_RESET}\n")
+
+def show_help():
+    print(f"\n{C_PURPLE}=== CodeAOIS Command Center ==={C_RESET}")
+    print(f"{C_YELLOW}/marketplace{C_RESET}   - View available specialized agents")
+    print(f"{C_YELLOW}/install <x>{C_RESET}   - Download an agent (e.g., /install database)")
+    print(f"{C_YELLOW}/clear{C_RESET}         - Clear the terminal screen")
+    print(f"{C_YELLOW}/clear_history{C_RESET} - Wipe your conversation memory")
+    print(f"{C_YELLOW}/clear_user{C_RESET}    - Wipe your profile and restart setup")
+    print(f"{C_YELLOW}/status{C_RESET}        - View current OS constraints and active models")
+    print(f"{C_YELLOW}/exit{C_RESET}          - Close the CodeAOIS environment")
+    print(f"{C_PURPLE}==============================={C_RESET}\n")
+
+def show_status():
+    print(f"\n{C_BLUE}[OS Status]{C_RESET}")
+    print(" 🟢 Core Engine: Online")
+    print(" 🟢 Model Router: Liquid/Arcee (Chat) | Step/Qwen (Code)")
+    print(" 🟢 Project Brain: Active")
+    print(" 🟢 Plugin Architecture: Dynamic Auto-Discovery")
+    print(" 🟢 UI Engine: Prompt Toolkit & Rich Markdown")
+    print(" 🟢 Security Dry-Run: Enabled\n")
+
+def process_command(user_input: str):
+    global chat_history, active_file
+    clean_input = user_input.lower().strip()
+    
+    if clean_input in ["/exit", "/quit", "exit", "quit"]:
+        print(f"\n{C_GREEN}👋 Terminating CodeAOIS session. Happy coding!{C_RESET}")
+        sys.exit(0)
+        
+    if clean_input == "/help":
+        show_help()
+        return
+        
+    if clean_input == "/clear":
+        os.system('clear' if os.name == 'posix' else 'cls')
+        print_logo()
+        return
+        
+    if clean_input == "/clear_history":
+        chat_history.clear() 
+        clear_history_data() 
+        print(f"\n{C_GREEN}🧹 Conversation history wiped clean. I have amnesia!{C_RESET}\n")
+        return
+        
+    if clean_input == "/clear_user":
+        clear_profile_data() 
+        print(f"\n{C_GREEN}👤 User profile deleted. Exiting so you can restart setup...{C_RESET}\n")
+        sys.exit(0) 
+        
+    if clean_input == "/status":
+        show_status()
+        return
+
+    if clean_input == "/marketplace":
+        installed = get_installed_agents()
+        
+        # --- 1. THE EXPANDED AGENT REGISTRY ---
+        available_agents = {
+            "database": {
+                "desc": "SQL, NoSQL, ORM",
+                "summary": "Writes optimized queries, designs schemas, and handles database migrations."
+            },
+            "3d": {
+                "desc": "Three.js, WebGL, Unity",
+                "summary": "Generates complex 3D math, shaders, and game engine scripts."
+            },
+            "debugging": {
+                "desc": "Deep Error Analysis",
+                "summary": "Reads crash logs, finds root causes, and suggests exact line-by-line fixes."
+            },
+            "devops": {
+                "desc": "Docker, CI/CD, AWS",
+                "summary": "Writes Dockerfiles, GitHub Actions, and infrastructure automation."
+            }
+        }
+        
+        # --- 2. BUILD THE STUNNING UI TABLE ---
+        print("\n")
+        table = Table(title="🛒 CodeAOIS Agent Marketplace", border_style="purple", header_style="bold cyan")
+        table.add_column("Agent Name", style="bold white", no_wrap=True)
+        table.add_column("Status", justify="center")
+        table.add_column("Core Focus", style="magenta")
+        table.add_column("Agent Capabilities", style="dim")
+        
+        for agent, info in available_agents.items():
+            if agent in installed:
+                status = "[bold green]✓ Installed[/bold green]"
+            else:
+                status = "[bold yellow]○ Available[/bold yellow]"
+                
+            table.add_row(agent.title(), status, info["desc"], info["summary"])
+            
+        console.print(table)
+        print("\n")
+        
+        # --- 3. THE INTERACTIVE AUTO-INSTALLER ---
+        choice = input(f"{C_BLUE}Enter an agent name to install (or press Enter to close): {C_RESET}").strip().lower()
+        
+        if choice in available_agents:
+            if choice in installed:
+                print(f"{C_YELLOW}⚠️ You already have the {choice.title()} Agent installed!{C_RESET}\n")
+            else:
+                print(f"{C_GREEN}⬇️ Downloading {choice.title()} Agent from registry...{C_RESET}")
+                install_agent(choice)
+                print(f"{C_GREEN}✅ {choice.title()} Agent successfully integrated into the OS!{C_RESET}\n")
+        elif choice:
+            print(f"{C_YELLOW}⚠️ Agent '{choice}' not found in the marketplace.{C_RESET}\n")
+            
+        return
+
+    if clean_input.startswith("/install"):
+        parts = clean_input.split()
+        if len(parts) > 1:
+            agent_target = parts[1].strip()
+            install_agent(agent_target)
+        else:
+            print(f"\n{C_YELLOW}⚠️  Missing agent name! Please specify an agent (e.g., /install database){C_RESET}\n")
+        return
+
+    print(f"\n{C_CYAN}[System]{C_RESET} Analyzing intent...")
+    intent = analyze_intent(user_input)
+    
+    if intent == "chat":
+        print(f"{C_BLUE}💬 [Chat Mode]{C_RESET} Asking AI...")
+        
+        profile = load_profile()
+        user_name = profile["name"] if profile else "Developer"
+        user_role = profile["role"] if profile else "Coding"
+        
+        project_tree = scan_project_structure()
+        
+        system_prompt = (
+            f"You are CodeAOIS, an advanced AI Developer OS running directly in the user's terminal. "
+            f"The user's name is {user_name} ({user_role}).\n"
+            f"CRITICAL: You have FULL access to the user's file system. Here is their current project structure:\n{project_tree}\n"
+            f"Currently focused file: {active_file if active_file else 'None'}\n"
+            f"If the user asks 'can you see my code' or asks about their files, say YES and reference the project structure. "
+            f"DO NOT EVER say you cannot view files."
+        )
+        
+        response = call_openrouter(system_prompt, user_input, intent="chat", history=chat_history)
+        print(f"\n{C_BOLD}🤖 CodeAOIS:{C_RESET} {response}\n")
+        
+        chat_history.append({"role": "user", "content": user_input})
+        chat_history.append({"role": "assistant", "content": response})
+        if len(chat_history) > 20: chat_history = chat_history[-20:]
+        save_history(chat_history)
+        
+    elif intent in ["code", "data_science"] or intent.endswith("_agent"):
+        
+        if intent == "data_science":
+            print(f"{C_PURPLE}📊 [Data Science Mode]{C_RESET} Engaging specialized agent...")
+        elif intent == "code":
+            print(f"{C_GREEN}⚙️ [Code Mode]{C_RESET} Engaging Coder agent...")
+        else:
+            plugin_name = intent.replace('_agent', '').title()
+            print(f"{C_PURPLE}🔌 [Plugin Mode]{C_RESET} Engaging specialized {plugin_name} Agent...")
+            
+        target_file, file_context = extract_file_context(user_input)
+        
+        # ACTIVE FILE MEMORY
+        if not target_file and active_file:
+            print(f"   🔗 Auto-locking to previous file: {active_file}")
+            target_file = active_file
+            if os.path.exists(active_file):
+                with open(active_file, "r", encoding="utf-8") as f:
+                    file_context = f.read()
+
+        if target_file:
+            active_file = target_file
+            print(f"   📂 Target file locked: {target_file}")
+            
+        print(f"   🧠 Scanning project context...")
+        project_tree = scan_project_structure()
+        
+        brain_context = f"\n--- Project Structure Overview ---\n{project_tree}\n----------------------------------\n"
+        full_context = brain_context + file_context if file_context else brain_context
+            
+        print(f"   {C_YELLOW}⚡ Generating code...{C_RESET}")
+        
+        # --- GENERATE THE CODE_RESULT VARIABLE ---
+        if intent == "data_science":
+            code_result = generate_ds_code(user_input, full_context)
+        elif intent == "code":
+            code_result = generate_code(user_input, full_context)
+        else:
+            try:
+                module = importlib.import_module(f"codeaois.agents.{intent}")
+                func_name = f"generate_{intent.replace('_agent', '')}_code"
+                agent_func = getattr(module, func_name)
+                code_result = agent_func(user_input, full_context)
+            except Exception as e:
+                print(f"\n{C_YELLOW}⚠️ Error executing plugin {intent}: {e}{C_RESET}")
+                return
+        
+        # --- EXTRACT SAVE PATH ---
+        if target_file:
+            save_path = target_file
+        else:
+            print("\n" + "-"*50)
+            custom_name = input(f"{C_BLUE}📁 Enter filename with extension (e.g., app.py): {C_RESET}").strip()
+            save_path = custom_name if custom_name else "untitled_generation.txt"
+            print("-"*50)
+        
+        # --- SAVE AND SHOW SUMMARY ---
+        success, ai_summary = extract_and_save_code(code_result, default_filename=save_path)
+        if success:
+            auto_commit(save_path, message=f"CodeAOIS auto-update: {save_path} via {intent}")
+            print("\n")
+            md = Markdown(ai_summary)
+            console.print(Panel(md, title="[bold cyan]Agent Summary[/bold cyan]", border_style="cyan", expand=False))
+            print("\n")
+            
+            # --- NEW: BRIDGE THE BRAINS ---
+            # Inject the coding action into the shared chat memory!
+            chat_history.append({"role": "user", "content": user_input})
+            chat_history.append({
+                "role": "assistant", 
+                "content": f"[System Log: I successfully generated code and saved it to '{save_path}'. Here is my summary of what I did: {ai_summary}]"
+            })
+            if len(chat_history) > 20: chat_history = chat_history[-20:]
+            save_history(chat_history)
+# --- NEW: THE MULTI-AGENT TESTER LOOP ---
+            if save_path.endswith('.py'):
+                confirm_test = input(f"{C_PURPLE}🧪 [Tester Agent] Shall I execute '{save_path}' to check for errors? [Y/n]: {C_RESET}").strip().lower()
+                
+                if confirm_test in ['y', '']:
+                    print(f"   {C_PURPLE}⚙️ Running tests...{C_RESET}")
+                    test_success, test_output = run_test(save_path)
+                    
+                    if test_success:
+                        print(f"{C_GREEN}✅ Test Passed! Terminal Output:\n{C_RESET}{test_output}\n")
+                    else:
+                        print(f"{C_YELLOW}❌ Test Failed! Crash Log:\n{C_RESET}{test_output}\n")
+                        
+                        # The Loop-Back!
+                        fix_confirm = input(f"{C_PURPLE}🔁 Pass crash log back to Coder Agent to fix? [Y/n]: {C_RESET}").strip().lower()
+                        if fix_confirm in ['y', '']:
+                            print(f"{C_GREEN}⚙️ Re-Engaging Coder agent for bug fix...{C_RESET}")
+                            
+                            # Automatically generate a new prompt with the error log
+                            fix_prompt = f"I ran {save_path} and got this error:\n{test_output}\nPlease fix the code."
+                            
+                            # Recursively call the command processor to handle the fix!
+                            process_command(fix_prompt)
+    else:
+        print(f"{C_YELLOW}❓ Unknown intent. Please try rephrasing.{C_RESET}")
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="CodeAOIS: Advanced AI Developer OS",
+        usage="codeaois [prompt] or codeaois [options]"
+    )
+    parser.add_argument("prompt", nargs="*", help="Chat or command for CodeAOIS")
+    parser.add_argument("-v", "--version", action="version", version="CodeAOIS Core Engine v0.1.0")
+    args = parser.parse_args()
+
+    profile = load_profile()
+    if not profile:
+        profile = create_profile()
+
+    if args.prompt:
+        user_input = " ".join(args.prompt)
+        process_command(user_input)
+        sys.exit(0)
+
+    os.system('clear' if os.name == 'posix' else 'cls')
+    print_logo()
+    
+    if profile:
+        print(f"{C_CYAN}  Welcome back, {profile.get('name', 'Developer')}!{C_RESET}\n")
+    
+    session = PromptSession(
+        completer=command_completer,
+        auto_suggest=AutoSuggestFromHistory(),
+        complete_while_typing=True
+    )
+    
+    while True:
+        try:
+            user_input = session.prompt(HTML('<b><ansigreen>CodeAOIS></ansigreen></b> ')).strip()
+            if not user_input:
+                continue
+            process_command(user_input)
+        except KeyboardInterrupt:
+            print(f"\n{C_GREEN}👋 Terminating CodeAOIS session. Happy coding!{C_RESET}")
+            sys.exit(0)
+
+if __name__ == "__main__":
+    main()