trying to get graphviz working

app.py

@@ -53,7 +53,7 @@ class Config:
         self.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
         self.DTYPE = torch.float32
         self.TOPK = 5
-        self.CUTOFF = 0.00001  # Cumulative probability cutoff for tree branches
+        self.CUTOFF = 0.0001  # Cumulative probability cutoff for tree branches
         self.OUTPUT_LENGTH = 20
         self.SUB_TOKEN_ID = 23070  # Arbitrary token ID to overwrite with embedding (token = "OSS")
         self.LOG_BASE = 10
@@ -321,14 +321,23 @@ def add_nodes_edges(dot, node, config, max_weight, min_weight, parent=None, is_r
             add_nodes_edges(dot, child, config, max_weight, min_weight, parent=node_id, is_root=False, depth=depth+1, trim_cutoff=trim_cutoff)
 
 def create_tree_diagram(data, config, max_weight, min_weight, trim_cutoff=0):
-    dot = Digraph(comment='Definition Tree', format='png')
+    import os
+    from graphviz import Digraph
+
+    # Ensure the system can find the dot command
+    os.environ["PATH"] += os.pathsep + "/usr/bin"
+
+    # Initialize the Digraph object and explicitly set the dot command path
+    dot = Digraph(comment='Definition Tree', format='png', engine='dot')
+    dot.command = '/usr/bin/dot'
+    
     dot.attr(rankdir='LR', size='5040,5000', margin='0.06', nodesep='0.06', ranksep='1', dpi='120', bgcolor='white')
 
     add_nodes_edges(dot, data, config, max_weight, min_weight, trim_cutoff=trim_cutoff)
 
     # Save to a temporary file first
     temp_filename = "temp_tree_diagram"
-    dot.render(temp_filename, format='png', cleanup=True, engine='/usr/bin/dot')
+    dot.render(temp_filename, format='png', cleanup=True)
     
     # Read the file back into a BytesIO object
     with open(f"{temp_filename}.png", "rb") as f:
@@ -348,6 +357,7 @@ def create_tree_diagram(data, config, max_weight, min_weight, trim_cutoff=0):
 
 
 
+
 # Global variables to store loaded resources
 tokenizer = None
 model = None