plotly instead of matplotlib

psychohistory.py

@@ -124,21 +124,31 @@ def draw_graph_plotly(G, title="3D Event Tree", highlight_color='gray'):
     html_str = fig.to_html(full_html=False, include_plotlyjs='cdn')
     return html_str
 
+
 def main(json_data):
     G = nx.DiGraph()
     build_graph_from_json(json_data, G)
 
+    # Find the best, worst, longest, and shortest paths FIRST
+    best_path, best_mean_prob, worst_path, worst_mean_prob, longest_path, shortest_path = find_paths(G)
+
     # Generate the HTML string for the Plotly graph
     html_graph = draw_graph_plotly(G)
 
-    # ... (Rest of your code for finding paths)
-
+    # Now you can use the path variables
     if best_path:
         best_path_graph = draw_graph_plotly(G.subgraph(best_path), title="Best Path", highlight_color='blue')
         html_graph += best_path_graph
     if worst_path:
         worst_path_graph = draw_graph_plotly(G.subgraph(worst_path), title="Worst Path", highlight_color='red')
         html_graph += worst_path_graph
-    # ... (Similar for longest_path and shortest_path)
+    if longest_path:
+        longest_path_graph = draw_graph_plotly(G.subgraph(longest_path), title="Longest Path", highlight_color='green')
+        html_graph += longest_path_graph
+    if shortest_path:
+        shortest_path_graph = draw_graph_plotly(G.subgraph(shortest_path), title="Shortest Path", highlight_color='purple')
+        html_graph += shortest_path_graph
 
     return html_graph  # Return the HTML string
+
+