Update app.py

app.py

@@ -5,6 +5,7 @@ import re
 import gradio as gr
 from neo4j import GraphDatabase
 import os
+from langchain.memory import ConversationBufferMemory
 
 api_key = os.getenv("GEMINI_API_KEY")
 neo4j_uri = os.getenv("NEO4J_URI")
@@ -115,16 +116,14 @@ def chatbot(input, history=[]):
         # we could log these situations in the backend to revisit later in development
         return "I'm sorry, there was an error retrieving the information you requested."
 
-gr.ChatInterface(fn = chatbot,
-                 title = "Mahabharata Chatbot",
-                 description = "powered by Neo4j",
-                 theme = "soft",
-                 chatbot = gr.Chatbot(height=500),
-                 undo_btn = None,
-                 clear_btn = "\U0001F5D1 Clear chat",
-                 examples = ["Who killed Ghatotakach?",
-                             "Who are the parents of Karna?",
-                             "Who are the kids of Kunti?",
-                             "Who are the siblings of Karna?"],
-                 inputs = ["text",'state'],
-                 outputs = ["chatbot",'state']).launch(debug = True, share=True)
+gr.Interface(fn = chatbot,
+             title = "Mahabharata Chatbot",
+             description = "powered by Neo4j",
+             theme = "soft",
+             clear_btn = "\U0001F5D1 Clear chat",
+             examples = ["Who killed Ghatotakach?",
+                         "Who are the parents of Karna?",
+                         "Who are the kids of Kunti?",
+                         "Who are the siblings of Karna?"],
+             inputs = ["text",'state'],
+             outputs = ["chatbot",'state']).launch(debug = True, share=True)