from GPT4KG import KnowledgeGraph import gradio as gr from PIL import Image def generate_graph(input_text,api_key): try: kg = KnowledgeGraph(api_key,"temp.kg") except: kg = KnowledgeGraph(api_key) kg.learn(str(input_text)) kg.save_graph("temp.kg") kg.display_graph("temp.png") return Image.open("temp.png") def answer_question(question,api_key): try: kg = KnowledgeGraph(api_key,"temp.kg") except: kg = KnowledgeGraph(api_key) return kg.chat_qa(question) title = "GPT-4 Knowledge Graph Generator" description = "Enter text to generate a knowledge graph using GPT4KG:" with gr.Blocks() as demo: with open("temp.kg","w") as f: f.write("") gr.Markdown("""