Update app.py

app.py

@@ -21,11 +21,12 @@ def answer_question(question,api_key,graph):
     kg = graph[0]
   return kg.chat_qa(question)
 
-def clear_graph(api_key,graph):
+def clear_graph(api_key,graph,output_image):
   graph = []
   kg = KnowledgeGraph(api_key)
   graph.append(kg)
-  return graph,Image.new('RGB', (300, 100),(255, 255, 255))
+  w, h = output_image.size
+  return graph,Image.new('RGB', (w, h),(255, 255, 255))
 
 title = "GPT-4 Knowledge Graph Generator"
 description = "Enter text to generate a knowledge graph using GPT4KG:"
@@ -45,6 +46,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(fn=clear_graph, inputs=[api_key,graph], outputs=[graph,output_image],api_name="clear")
+  clear_btn.click(fn=clear_graph, inputs=[api_key,graph,output_image], outputs=[graph,output_image],api_name="clear")
 
 demo.launch()