Update app.py

app.py

@@ -62,17 +62,12 @@ def add_to_graph(
     fig: go.Figure = create_plotly_graph(lesson_graph)
     
     # Convert Plotly figure to HTML
-    plot_html: str = pio.to_html(
-        fig,
-        full_html=False,
-        include_plotlyjs=True,
-        config={'displayModeBar': False}
-    )
+    fig  # Directly use the Plotly figure object for Gradio's Plot component
     
     # Wrap the plot HTML in a div with a fixed size
     plot_html = f'<div style="width:100%;height:500px">{plot_html}</div>'
     
-    return search_string, plot_html
+    return search_string, fig
 
 def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
     pos: Dict[Any, Tuple[float, float]] = nx.spring_layout(graph)
@@ -187,7 +182,7 @@ with demo:
         clear_btn = gr.Button("Clear Landscape")
     
     search_output = gr.Textbox(label="Content Discovery Search String")
-    graph_output = gr.HTML(label="Your Educational Landscape")
+    graph_output = gr.Plot(label="Your Educational Landscape")
     message_output = gr.Textbox(label="Landscape Status")
     
     generate_btn.click(