Write out dot for diagnostics. Tidy output

Author: mrseanryan

README.md

@@ -194,6 +194,7 @@ I'm sorry, but I don't have access to personal information.
 ## Example Execution - Describing a given workflow
 
 ```
+>> Descibe this workflow:
 digraph G {
 
   // start
@@ -238,6 +239,7 @@ Overall, this flow chart helps guide the decision-making process for loan approv
 ```
 
 ```
+>> Descibe this workflow:
 digraph G {
 
   // start

core.py

@@ -3,6 +3,7 @@
 
 import config
 import service_chat
+import util_file
 
 # DEV note: NOT using langchain.
 # Tried using langchain but it's validation gets in the way of more complex prompts.
@@ -16,16 +17,32 @@ def create_command_messages(expert_commands):
     return messages
 
 DOT_GRAPH_START = "digraph G"
+DOT_SECTION_DELIMITER = "```"
+
+def generate_output_file_path(prompt_id, extension):
+    return os.path.join(config.PATH_TO_PNG_OUTDIR, f"dot_graph_{prompt_id}.{extension}")
+
+def generate_png_from_dot(dot_string, prompt_id):
+    graphs = pydot.graph_from_dot_data(dot_string)
+    path_to_png = generate_output_file_path(prompt_id, "png")
+    print(f"Writing png to '{path_to_png}'")
+    graphs[0].write_png(path_to_png)
+
+def write_dot_to_file(dot_string, prompt_id):
+    path_to_dot = generate_output_file_path(prompt_id, "dot")
+    print(f"Writing dot to '{path_to_dot}'")
+    util_file.write_text_to_file(dot_string, path_to_dot)
 
 def process_response(rsp, prompt_id):
     if DOT_GRAPH_START not in rsp:
         return rsp
     parts = rsp.split(DOT_GRAPH_START)
     dot_string = DOT_GRAPH_START + parts[1]
-    graphs = pydot.graph_from_dot_data(dot_string)
-    path_to_png = os.path.join(config.PATH_TO_PNG_OUTDIR, f"dot_graph_{prompt_id}.png")
-    print(f"Writing png to '{path_to_png}'")
-    graphs[0].write_png(path_to_png)
+    if DOT_SECTION_DELIMITER in dot_string:
+        dot_string = dot_string.split(DOT_SECTION_DELIMITER)[0]
+    write_dot_to_file(dot_string, prompt_id)
+    generate_png_from_dot(dot_string, prompt_id)
+
     return dot_string
 
 def execute_prompt(user_prompt, previous_messages, command_messages, prompt_id):

util_file.py

@@ -0,0 +1,3 @@
+def write_text_to_file(text, filepath):
+    with open(filepath, "w", encoding='utf-8') as f:
+        f.write(text)