update app

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import networkx as nx
 import matplotlib.pyplot as plt
 from io import BytesIO
+from PIL import Image
 
 # Initialize the lesson plan graph
 lesson_graph = nx.DiGraph()
@@ -52,7 +53,10 @@ def add_to_graph(teacher_name, subject, grade_level, learning_objective, activit
     buf.seek(0)
     plt.close()
     
-    return search_string, buf
+    # Convert BytesIO to PIL Image
+    image = Image.open(buf)
+    
+    return search_string, image
 
 def clear_graph():
     global lesson_graph