#!/bin/bash

# Function to create directories
create_directories() {
    mkdir -p ./{data/{raw,processed},notebooks,src/{agent,llm,tools,utils},tests/{agent,llm,tools,utils}}
}

# Function to create files
create_files() {
    touch ./{.gitignore,requirements.txt,README.md}
    touch ./src/{agent,llm,tools,utils}/__init__.py
    touch ./tests/{agent,llm,tools,utils}/test_{agent,llm,tool1,tool2,utils}.py
}

# Function to initialize Git repository
initialize_git() {
    echo "Do you want to initialize a Git repository? (y/n)"
    read answer
    if [ "$answer" == "y" ]; then
        git init
        echo "Git repository initialized."
        cd ..
    fi
}

# Main script execution
create_directories
create_files
#initialize_git

echo "Project setup complete."