finetool dialogs tool finaly working!

finetune_dialogs_tool/main.py

@@ -1,11 +1,14 @@
 
 # I know this code sucks and there are many repeating parts but it's just a simple temporary tool and I didn't bother to make it better
+# This is not how something serious should be done
 
 import os
 import numpy as np
 from order_items import possible_order_items
+import json
 
 root= os.path.dirname(os.path.abspath(__file__))
+vscode_available = any(os.access(os.path.join(path, "code"), os.X_OK) for path in os.environ["PATH"].split(os.pathsep))
 
 def mkdir_if_not_exists_rel(path):
     if not os.path.exists(root + "/" + path):
@@ -19,6 +22,8 @@ mkdir_if_not_exists_rel("output_dialogs")
 mkdir_if_not_exists_rel("temp")
 
 
+# if temp/dialog_input.py does not exist, do the processing instead of just opening it
+if not os.path.exists(root + "/temp/dialog_input.py"):
     # read templates/dialog_input.txt
     with open(root + "/templates/dialog_input.txt", "r", encoding="utf-8") as f:
         dialog_input_template = f.read()
@@ -88,8 +93,6 @@ dialog_input_template = dialog_input_template.replace("{{order_items}}", order_i
     with open(root + "/temp/dialog_input.py", "w", encoding="utf-8") as f:
         f.write(dialog_input_template)
 
-vscode_available = any(os.access(os.path.join(path, "code"), os.X_OK) for path in os.environ["PATH"].split(os.pathsep))
-
 while True:
     # open it in notepad and wait for user to close it
     # (or vscode if installed)
@@ -111,14 +114,14 @@ while True:
 
     # execute part from the beginning to "## PYTHON END ##" as a python code here
     python_end_idx = dialog_input.find("## PYTHON END ##")
-    phone_number, payment_method, order_items = "", "", ""
+    phone_number, delivery_address, payment_method, order_items = None, None, None, None
     exec(dialog_input[:python_end_idx])
     # it will set variables: phone_number, payment_method and order_items
 
     # check if theres any empty variables
     not_ok = False
     for val in [phone_number, delivery_address, payment_method, order_items]:
-        if val == "":
+        if val == "" or val is None:
             print("Some values are empty. Please fill them.")
             not_ok = True
             break
@@ -167,21 +170,74 @@ while True:
 dialog[-1] += " CALLEND"
 
 now_pizzeria = True
+print("=============================================")
 for line in dialog:
-    print("piz -" if now_pizzeria else "bot -", line)
+    print("🍕 -" if now_pizzeria else "🤖 -", line)
+    now_pizzeria = not now_pizzeria
+print("=============================================")
+
+
+
+input("Everything seems okay. Press enter to save it as a new JSON file.")
+
+
+
+
+# read templates/system_instructions.txt
+with open(root + "/templates/system_instructions.txt", "r", encoding="utf-8") as f:
+    system_instructions_template = f.read()
+
+system_instructions_template = system_instructions_template.replace("{{phone_number}}", phone_number)
+system_instructions_template = system_instructions_template.replace("{{delivery_location}}", delivery_address)
+system_instructions_template = system_instructions_template.replace("{{payment_method}}", payment_method)
+
+order_items_str = ""
+for line in order_items:
+    order_items_str += f"{line}\n"
+order_items_str = order_items_str[:-1]
+
+system_instructions_template = system_instructions_template.replace("{{order_items}}", order_items_str)
+
+
+# use all of these to create chatgpt finetuning JSON file
+
+training_dict = {
+    "messages": []
+}
+
+# add system instructions
+training_dict["messages"].append({
+    "role": "system",
+    "content": system_instructions_template
+})
+
+# add dialog
+now_pizzeria = True
+for line in dialog:
+    training_dict["messages"].append({
+        "role": "user" if now_pizzeria else "assistant",
+        "content": line
+    })
     now_pizzeria = not now_pizzeria
 
 
+# find all json files consisting of number (0.json, 1.json .... 123.json)
+files = [f for f in os.listdir(root + "/output_dialogs") if f.endswith(".json") and f[:-5].isdigit()]
+
+# get name for next file (number higher than the highest number in the folder)
+if len(files) == 0:
+    next_file = "0.json"
+else:
+    next_file = str(max([int(f[:-5]) for f in files]) + 1) + ".json"
 
 
+# save to output_dialogs/next_file
+with open(root + "/output_dialogs/" + next_file, "w", encoding="utf-8") as f:
+    json.dump(training_dict, f, indent=4, ensure_ascii=False)
 
 
+# remove temp/dialog_input.py
+os.remove(root + "/temp/dialog_input.py")
 
-## find all json files consisting of number (0.json, 1.json .... 123.json)
-#files = [f for f in os.listdir("output_dialogs") if f.endswith(".json") and f[:-5].isdigit()]
-#
-## get name for next file (number higher than the highest number in the folder)
-#if len(files) == 0:
-#    next_file = "0.json"
-#else:
-#    next_file = str(max([int(f[:-5]) for f in files]) + 1) + ".json"
+
+print(f"Saved as {next_file}")

finetune_dialogs_tool/templates/system_instructions.txt

@@ -4,9 +4,10 @@ DATA:
 phone: {{phone_number}}
 delivery location: {{delivery_location}}
 paymentMethod: {{payment_method}}
+order:
 {{order_items}}
 
-REMEMBER DONT USE NUMBERS USE WORDS for example dont say 1x say one time also REMEMBER to use replacement words to a word so it is appropriate to the whole sentence example:
+REMEMBER DONT USE NUMBERS, USE WORDS for example dont say 1x, say one time also REMEMBER to use replacement words to a word so it is appropriate to the whole sentence example:
 WRONG:
 Chciałbym zamówić jedną Margherittę, dwie Colę, pięć Fant i jedną Sprite.
 RIGHT: