Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from PIL import Image
 
 def generate_graph(input_text,api_key,graph):
-  if graph[0] =="":
+  if graph == []:
     kg = KnowledgeGraph(api_key)
     graph[0] = kg
   else:
@@ -28,7 +28,7 @@ with gr.Blocks() as demo:
   gr.Markdown("""<h1><center>GPT-4 Knowledge Graph Generator</center></h1>""")
   output_image = gr.Image(label="Knowledge Graph", type="pil")
   api_key = gr.Textbox(lines=1, label="OpenAI API Key")
-  graph = gr.State([""])
+  graph = gr.State([])
   input_text = gr.Textbox(lines=5, label="Information to be added to graph")
   submit_btn = gr.Button("Add info to graph")
   submit_btn.click(fn=generate_graph, inputs=[input_text,api_key,graph], outputs=[output_image,graph])
@@ -38,4 +38,7 @@ with gr.Blocks() as demo:
   qa_btn = gr.Button("Ask question")
   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")
+
 demo.launch()