Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import igraph as ig
 import plotly.graph_objects as go
 
+PLOTLY_TREE = None
+
+def get_plotly_tree():
+    return PLOTLY_TREE
+
 def create_plotly_tree():
     # Create an igraph tree
     g = ig.Graph.Tree(7, 2)  # Example tree
@@ -42,9 +47,11 @@ def create_plotly_tree():
 
     return fig
 
+PLOTLY_TREE = create_plotly_tree()
+
 def setup_interface():
     iface = gr.Interface(
-        fn=create_plotly_tree,
+        fn=get_plotly_tree,
         inputs=[],
         outputs=gr.Plot(),
         title="Interactive Tree Visualization"