main.py

#!/usr/bin/env python3
"""
AI Code Agent - Command Line Interface
"""
import sys
import logging
import json
from pathlib import Path
from src.agent import CodeAgent

# Configure logging - suppress verbose Gemini logs
logging.basicConfig(level=logging.WARNING)

# Only show specific loggers if needed
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)

# Session file path
SESSION_FILE = Path.home() / ".ai_code_agent_session.json"


def load_session(working_dir: str):
    """Load previous session if exists for this directory."""
    try:
        if SESSION_FILE.exists():
            with open(SESSION_FILE, 'r', encoding='utf-8') as f:
                session_data = json.load(f)
            
            # Check if session is for the same working directory
            if session_data.get('working_dir') == working_dir:
                return session_data.get('messages', [])
    except (json.JSONDecodeError, KeyError, OSError):
        pass
    
    return []


def save_session(working_dir: str, messages):
    """Save current session to file."""
    try:
        session_data = {
            'working_dir': working_dir,
            'messages': messages
        }
        
        with open(SESSION_FILE, 'w', encoding='utf-8') as f:
            json.dump(session_data, f, ensure_ascii=False, indent=2)
    except OSError:
        pass  # Ignore save errors


def clear_session():
    """Clear the current session."""
    try:
        if SESSION_FILE.exists():
            SESSION_FILE.unlink()
            print("🗑️  Session cleared!")
    except OSError:
        print("❌ Could not clear session")


def interactive_chat():
    """
    Start an interactive chat session.
    """
    try:
        # Initialize the agent
        agent = CodeAgent()
        
        # Load previous conversation
        previous_messages = load_session(agent.working_dir)
        if previous_messages:
            agent.load_conversation_history(previous_messages)
            print(f"💾 Loaded previous conversation ({len(previous_messages)} messages)")
        
        print("🤖 AI Code Agent - Interactive Mode")
        print(f"📁 Working directory: {agent.working_dir}")
        print("=" * 50)
        print("Type 'q' to quit, 'clear' to clear history, 'help' for commands")
        print("=" * 50)
        
        while True:
            try:
                # Get user input
                user_input = input("\n💬 You: ").strip()
                
                if not user_input:
                    continue
                    
                # Handle special commands
                if user_input.lower() == 'q' or user_input.lower() == 'quit':
                    print("👋 Goodbye!")
                    break
                elif user_input.lower() == 'clear':
                    clear_session()
                    agent.messages = []
                    print("🗑️ Conversation history cleared!")
                    continue
                elif user_input.lower() == 'help':
                    print("\n📋 Commands:")
                    print("  q, quit - Exit the chat")
                    print("  clear - Clear conversation history")
                    print("  help - Show this help message")
                    continue
                
                # Process the request
                print("\n🤖 Assistant:")
                response = agent.chat(user_input, verbose=False)
                print(response)
                
                # Save conversation after each exchange
                save_session(agent.working_dir, agent.export_conversation_history())
                
            except KeyboardInterrupt:
                print("\n👋 Interrupted by user")
                break
                
    except ValueError as e:
        print(f"❌ Configuration Error: {e}")
        print("Make sure you have set the GEMINI_API_KEY environment variable.")
    except Exception as e:
        print(f"❌ Unexpected Error: {e}")


def main():
    """
    Main entry point for the AI Code Agent CLI.
    """
    # Check for command line arguments
    if len(sys.argv) > 1:
        # Handle clear command
        if sys.argv[1] == "--clear":
            clear_session()
            return
            
        # Single command mode (legacy support)
        request = sys.argv[1]
        verbose = "--verbose" in sys.argv[2:]

        try:
            agent = CodeAgent()
            
            if verbose:
                print("🤖 AI Code Agent starting...")
                print(f"📁 Working directory: {agent.working_dir}")
                print(f"💬 Request: {request}")
                print("=" * 50)

            response = agent.chat(request, verbose=verbose)

            if verbose:
                print("=" * 50)
                print("🎯 Final Response:")

            print(response)
            return
            
        except ValueError as e:
            print(f"❌ Configuration Error: {e}")
            print("Make sure you have set the GEMINI_API_KEY environment variable.")
            return
        except Exception as e:
            print(f"❌ Unexpected Error: {e}")
            return
    
    # Default: Interactive mode
    interactive_chat()


if __name__ == "__main__":
    main()