import os

# List of files to extract code from, relative to the `src` folder
files = [
    "src/client/views/nodes/chatbot/agentsystem/Agent.ts",
    "src/client/views/nodes/chatbot/agentsystem/prompts.ts",
    "src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx",
    "src/client/views/nodes/chatbot/chatboxcomponents/MessageComponent.tsx",
    "src/client/views/nodes/chatbot/response_parsers/AnswerParser.ts",
    "src/client/views/nodes/chatbot/response_parsers/StreamedAnswerParser.ts",
    "src/client/views/nodes/chatbot/tools/BaseTool.ts",
    "src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts",
    "src/client/views/nodes/chatbot/tools/RAGTool.ts",
    "src/client/views/nodes/chatbot/tools/SearchTool.ts",
    "src/client/views/nodes/chatbot/tools/WebsiteInfoScraperTool.ts",
    "src/client/views/nodes/chatbot/types/tool_types.ts",
    "src/client/views/nodes/chatbot/types/types.ts",
    "src/client/views/nodes/chatbot/vectorstore/Vectorstore.ts",
]

# Output file name
output_file = "extracted_code.txt"

def extract_and_format_code(file_list, output_path):
    with open(output_path, "w") as outfile:
        for file in file_list:
            # Since the script runs from the chatbot folder, prepend the relative path from chatbot to src
            if os.path.exists(file):
                with open(file, "r") as infile:
                    code = infile.read()
                # Write formatted code to the output file
                outfile.write(f"--- {file} ---\n\n```\n{code}\n```\n\n")
            else:
                print(f"File not found: {file}")

# Run the extraction and formatting
extract_and_format_code(files, output_file)

print(f"Code extracted and saved to {output_file}")