Upload script_v1_0.R

old_versions/script_v1_0.R

@@ -0,0 +1,196 @@
+library(tidyverse)
+
+# ----- #
+read_captions_from_directory <- function(directory_path) {
+  # ディレクトリ内の.txtファイルのリストを取得
+  txt_files <- list.files(directory_path, pattern = "\\.txt$", full.names = TRUE)
+  
+  # 各.txtファイルからキャプションを読み込み
+  data <- lapply(txt_files, function(file) {
+    captions <- readLines(file, warn = FALSE)
+    captions_list <- strsplit(captions, ",")[[1]]
+    captions_list <- trimws(captions_list) # 余分な空白を取り除く
+    tibble(
+      image_path = gsub(".txt$", ".png", file),
+      caption_order = 1:length(captions_list),
+      caption = captions_list
+    )
+  })
+  
+  # データフレームに変換
+  bind_rows(data)
+}
+
+get_caption_frequency <- function(data) {
+  data %>%
+    group_by(caption) %>%
+    summarise(frequency = n()) %>%
+    arrange(-frequency)
+}
+
+search_by_caption <- function(data, target_caption) {
+  data %>%
+    filter(caption == target_caption) %>%
+    group_by(image_path) %>%
+    distinct()
+}
+
+remove_caption_and_adjust_order <- function(data, target_image_path, target_caption) {
+  # キャプションが存在するか確認
+  if (!any(data$image_path == target_image_path & data$caption == target_caption)) {
+    cat(sprintf("The caption '%s' does not exist for image '%s'.\n", target_caption, target_image_path))
+    return(data)
+  }
+
+  # 削除するキャプションのcaption_orderを取得
+  removed_order <- data$caption_order[data$image_path == target_image_path & data$caption == target_caption]
+  # キャプションを削除
+  data <- data %>% filter(!(image_path == target_image_path & caption == target_caption))
+  # caption_orderを調整
+  data$caption_order[data$image_path == target_image_path & data$caption_order > removed_order] <- data$caption_order[data$image_path == target_image_path & data$caption_order > removed_order] - 1
+  
+  return(data)
+}
+
+remove_low_frequency_captions <- function(data, threshold) {
+  # キャプションの頻度を取得
+  caption_freq <- get_caption_frequency(data)
+  
+  # 指定された頻度以下のキャプションのリストを作成
+  low_freq_captions <- caption_freq %>%
+    filter(frequency <= threshold) %>%
+    pull(caption)
+  
+  # 低頻度のキャプションを削除し、caption_orderを調整
+  for (caption in low_freq_captions) {
+    unique_images <- unique(data$image_path[data$caption == caption])
+    for (image in unique_images) {
+      data <- remove_caption_and_adjust_order(data, image, caption)
+    }
+  }
+  
+  return(data)
+}
+
+edit_captions_interactively <- function(data, target_caption) {
+  # キャプションで画像を検索
+  image_paths <- search_by_caption(data, target_caption)$image_path
+  
+  for (path in image_paths) {
+    # OSに応じて画像を開く
+    if (Sys.info()["sysname"] == "Windows") {
+      cmd <- sprintf('start "" "%s"', path)
+      shell(cmd, intern = TRUE) 
+    } else if (Sys.info()["sysname"] == "Darwin") { # macOS
+      cmd <- sprintf('open "%s"', path)
+      system(cmd)
+    } else { # Linux
+      cmd <- sprintf('xdg-open "%s"', path)
+      system(cmd)
+    }
+    
+    # ユーザーにキャプションの削除を選択させる
+    cat(sprintf("Do you want to remove the caption '%s' from image '%s'? (yes/no/end): ", target_caption, path))
+    response <- readline()
+    
+    if (tolower(response) == "end") {
+      break
+    } else if (tolower(response) == "yes") {
+      data <- remove_caption_and_adjust_order(data, path, target_caption)
+    }
+  }
+  
+  return(data)
+}
+
+add_caption_at_order <- function(data, target_image_path, target_caption, target_order = NULL) {
+  # 指定された画像の最大のcaption_orderを取得
+  max_order <- max(data$caption_order[data$image_path == target_image_path], na.rm = TRUE)
+  
+  # キャプションの重複チェック
+  if (target_caption %in% data$caption[data$image_path == target_image_path]) {
+    return(data)  # 重複がある場合、データをそのまま返す
+  }
+  
+  # target_orderが指定されていない場合、キャプションを表示してユーザーに選ばせる
+  if (is.null(target_order)) {
+    print_image_captions_as_csv(data, target_image_path)
+    cat("Enter the position (order) to insert the new caption (1 to", max_order + 1, "): ")
+    target_order <- as.numeric(readline())
+    
+    # 不適切な値が入力された場合、最大のorder + 1で追加
+    if (target_order <= 0 || target_order > max_order + 1) {
+      target_order <- max_order + 1
+    }
+  }
+  
+  # 指定されたorder以降のcaption_orderを増加
+  data <- data %>%
+    mutate(caption_order = ifelse(image_path == target_image_path & caption_order >= target_order, caption_order + 1, caption_order))
+  
+  # 新しいキャプションを追加
+  new_caption <- tibble(
+    image_path = target_image_path,
+    caption_order = target_order,
+    caption = target_caption
+  )
+  data <- bind_rows(data, new_caption)
+  
+  return(data)
+}
+
+move_caption_order <- function(data, target_image_path, target_caption, new_order) {
+
+  # キャプションが存在するか確認
+  if (!any(data$image_path == target_image_path & data$caption == target_caption)) {
+    cat(sprintf("The caption '%s' does not exist for image '%s'.\n", target_caption, target_image_path))
+    return(data)
+  }
+
+  # キャプションを削除
+  data_after_removal <- remove_caption_and_adjust_order(data, target_image_path, target_caption)
+  
+  # 新しい位置にキャプションを追加
+  data_after_addition <- add_caption_at_order(data_after_removal, target_image_path, target_caption, new_order)
+   return(data_after_addition)
+}
+
+# キャプションが存在するか確認 (使わないかも)
+is_caption_present <- function(data, target_image_path, target_caption) {
+  return(any(data$image_path == target_image_path & data$caption == target_caption))
+}
+
+# すべてのキャプションを表示
+print_all_unique_captions_as_csv <- function(data) {
+  # 重複なく全てのキャプションを取得
+  unique_captions <- unique(data$caption) 
+  # CSV形式で表示
+  cat(paste(unique_captions, collapse = ", "), "\n")
+}
+
+print_image_captions_as_csv <- function(data, target_image_path) {  
+  captions <- filter(data, image_path == target_image_path) %>%
+    arrange(caption_order) %>%
+    pull(caption)
+  
+  cat(paste(captions, collapse = ", "), "\n")
+}
+
+# 代表するキャプションに集約
+remove_related_captions_except_representative <- function(data, related_captions, representative_caption, target_image_path) {
+  
+  # representative_captionがtarget_image_pathに紐づいているか確認
+  if (!any(data$image_path == target_image_path & data$caption == representative_caption)) {
+    cat(sprintf("The representative caption '%s' is not associated with image '%s'.\n", representative_caption, target_image_path))
+    return(data)
+  }
+  
+  # target_image_pathに関連するキャプションを削除
+  for (caption in related_captions) {
+    if (caption != representative_caption) {
+      data <- remove_caption_and_adjust_order(data, target_image_path, caption)
+    }
+  }
+  
+  return(data)
+}