gokaygokay commited on
Commit
b9877bd
1 Parent(s): 7762285

print_debug

Browse files
Files changed (1) hide show
  1. prompt_generator.py +4 -0
prompt_generator.py CHANGED
@@ -113,6 +113,7 @@ class PromptGenerator:
113
  def load_next_data(self):
114
  next_data = {}
115
  next_path = os.path.join("data", "next")
 
116
  for category in os.listdir(next_path):
117
  category_path = os.path.join(next_path, category)
118
  if os.path.isdir(category_path):
@@ -120,12 +121,15 @@ class PromptGenerator:
120
  for file in os.listdir(category_path):
121
  if file.endswith(".json"):
122
  file_path = os.path.join(category_path, file)
 
123
  with open(file_path, "r", encoding="utf-8") as f:
124
  json_data = json.load(f)
125
  next_data[category][file[:-5]] = json_data
 
126
  return next_data
127
 
128
  def process_next_data(self, prompt, separator, category, field, value):
 
129
  if category in self.next_data and field in self.next_data[category]:
130
  field_data = self.next_data[category][field]
131
 
 
113
  def load_next_data(self):
114
  next_data = {}
115
  next_path = os.path.join("data", "next")
116
+ print(f"Loading next data from: {next_path}")
117
  for category in os.listdir(next_path):
118
  category_path = os.path.join(next_path, category)
119
  if os.path.isdir(category_path):
 
121
  for file in os.listdir(category_path):
122
  if file.endswith(".json"):
123
  file_path = os.path.join(category_path, file)
124
+ print(f"Loading file: {file_path}")
125
  with open(file_path, "r", encoding="utf-8") as f:
126
  json_data = json.load(f)
127
  next_data[category][file[:-5]] = json_data
128
+ print(f"Loaded next_data: {next_data}")
129
  return next_data
130
 
131
  def process_next_data(self, prompt, separator, category, field, value):
132
+ print(f"Processing next data: category={category}, field={field}, value={value}")
133
  if category in self.next_data and field in self.next_data[category]:
134
  field_data = self.next_data[category][field]
135