fix add_to_graph

app.py

@@ -29,7 +29,7 @@ def add_to_graph(
     assessment: str,
     resource: str,
     school_board: str
-) -> Tuple[str, str]:
+) -> Tuple[str, go.Figure]:
     global lesson_graph
     
     # Clear previous graph
@@ -61,14 +61,9 @@ def add_to_graph(
     # Create Plotly graph
     fig: go.Figure = create_plotly_graph(lesson_graph)
     
-    # Convert Plotly figure to HTML
-    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, fig
 
+
 def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
     pos: Dict[Any, Tuple[float, float]] = nx.spring_layout(graph)