zsh fun

Signed-off-by: Balazs Horvath <acsipont@gmail.com>

.zshrc

@@ -1,5 +1,28 @@
 alias ezc="nvim ~/.zshrc && source ~/.zshrc"
 
+# This function, rejiggle_captions, takes two arguments: a tag and a directory.
+# It searches for all .txt files in the specified directory and its subdirectories.
+# If a file contains the specified tag, the function removes the tag from its original position
+# and prepends it to the beginning of the file.
+# Usage: rejiggle_captions <tag> <directory>
+rejiggle_captions() {
+  local tag="$1"
+  local dir="$2"
+
+  if [[ -z "$tag" || -z "$dir" ]]; then
+    echo "Usage: rejiggle_captions <tag> <directory>"
+    return 1
+  fi
+
+  find "$dir" -type f -name "*.txt" | while read -r file; do
+    if grep -q "$tag" "$file"; then
+      sed -i "s/$tag//g" "$file"
+      sed -i "1s/^/$tag, /" "$file"
+    fi
+  done
+}
+
+
 # This function, `update_conda`, automates the process of upgrading all packages in every conda environment.
 # It performs the following steps:
 # 1. Retrieves the list of all conda environments using `conda env list` and extracts their names.
@@ -131,7 +154,8 @@ display_custom_help() {
     echo "----------------------------------------------------------------------------------------------------------------------"
     echo "LLMs"
     echo "---"
-    echo "conda activate openwebui && open-webui serve --port 6969"
+    echo "conda activate openwebui && PORT=6969 $HOME/source/repos/open-webui/backend/start.sh"
+    #echo "conda activate openwebui && open-webui serve --port 6969"
     echo "ollama serve"
     echo "----------------------------------------------------------------------------------------------------------------------"
     echo "Taggers + Captioners"
@@ -342,6 +366,8 @@ copy_matching_caption_files() {
 # The script uses a for loop to iterate through all .txt files in the target directory.
 # It utilizes the 'sed' command to perform an in-place replacement of the search_text with the replace_text in each file.
 # After processing all files, it prints a message indicating the completion of the text replacement operation.
+# Example usage:
+# replace_text_in_files "/path/to/directory" "squishy (artist)" "by squishy (artist)"
 replace_text_in_files() {
   local target_dir=$1
   local search_text=$2
@@ -356,11 +382,8 @@ replace_text_in_files() {
   echo "Text replacement complete in $target_dir!"
 }
 
-# Example usage:
-# replace_text_in_files "/path/to/directory" "squishy (artist)" "by squishy (artist)"
-
-
 # This script adds a specified prefix to the beginning of each text file in a given directory.
+# If the prefix already exists in the text file, it moves the prefix to the front of the text file without leaving extra commas or spaces.
 # Usage: inject_to_captions <directory> <prefix>
 # Arguments:
 #   <directory> - The directory containing the text files to be modified.
@@ -374,14 +397,18 @@ inject_to_captions() {
     if [[ -d "$dir" ]]; then
         for file in "$dir"/*.txt; do
             if [[ -f "$file" ]]; then
-                if ! grep -q "$prefix" "$file"; then
+                if grep -q "$prefix" "$file"; then
+                    # Move the existing prefix to the front of the text file without leaving extra commas or spaces
+                    local temp_file=$(mktemp)
+                    sed "s/$prefix//" "$file" | sed "1s/^/${prefix}, /" | sed 's/^, //' | sed 's/,,/,/g' | sed 's/, ,/,/g' | sed 's/ ,/,/g' > "$temp_file"
+                    mv "$temp_file" "$file"
+                    echo "Moved '${prefix}' to the front of $file"
+                else
                     # Use a temporary file to store the modified content
                     local temp_file=$(mktemp)
-                    echo "${prefix}, $(cat "$file")" > "$temp_file"
+                    echo "${prefix}, $(cat "$file")" | sed 's/,,/,/g' | sed 's/, ,/,/g' | sed 's/ ,/,/g' > "$temp_file"
                     mv "$temp_file" "$file"
                     echo "Added '${prefix}, ' to the front of $file"
-                else
-                    echo "The tag '${prefix}' already exists in $file"
                 fi
             fi
         done
@@ -563,7 +590,7 @@ c() {
     python main.py --listen 0.0.0.0 --preview-method taesd --use-pytorch-cross-attention --disable-xformers --front-end-version Comfy-Org/ComfyUI_frontend@latest --fast
 }
 
-alias cp='cp --reflinks=auto'
+alias cp='cp --reflink=auto'
 alias t="tensorboard --logdir=$HOME/output_dir/logs"
 alias rt="vim ~/.tmux.conf && echo \"Reloading tmux config\" && tmux source ~/.tmux.conf"
 alias zr="vim ~/.zshrc && echo \"Reloading zsh config\" && source ~/.zshrc"