Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 def generate_graph(input_text,api_key,graph):
   if graph == []:
     kg = KnowledgeGraph(api_key)
-    graph[0] = kg
+    graph.append(kg)
   else:
     kg = graph[0]
   kg.learn(str(input_text))
@@ -14,13 +14,19 @@ def generate_graph(input_text,api_key,graph):
   return img,graph
 
 def answer_question(question,api_key,graph):
-  if graph[0] =="":
+  if graph == []:
     kg = KnowledgeGraph(api_key)
-    graph[0] = kg
+    graph.append(kg)
   else:
     kg = graph[0]
   return kg.chat_qa(question)
 
+def clear_graph(api_key,graph):
+  graph = []
+  kg = KnowledgeGraph(api_key)
+  graph.append(kg)
+  return graph,Image.new('RGB', (256, 256))
+
 title = "GPT-4 Knowledge Graph Generator"
 description = "Enter text to generate a knowledge graph using GPT4KG:"
 
@@ -39,6 +45,6 @@ with gr.Blocks() as demo:
   qa_btn.click(fn=answer_question, inputs=[question,api_key,graph], outputs=[answer])
 
   clear_btn = gr.Button("Clear graph")
-  clear_btn.click(lambda: None, None, graph, queue=False, api_name="clear")
+  clear_btn.click(fn=clear_graph, inputs=[api_key,graph], outputs=[graph,output_image],api_name="clear")
 
 demo.launch()