Update app.py

app.py

@@ -71,11 +71,9 @@ def generate_text(
         temp += out["choices"][0]["text"]
         yield temp
 
-# Define the Gradio interface without retry_btn
-demo = gr.Interface(
-    fn=generate_text,
-    inputs="text",
-    outputs="text",
+# Define the Gradio interface
+demo = gr.ChatInterface(
+    generate_text,
     title="llama-cpp-python on GPU with ChromaDB",
     description="Running LLM with context retrieval from ChromaDB",
     examples=[
@@ -83,7 +81,8 @@ demo = gr.Interface(
         ["Can I make lunch for two people with this?"],
     ],
     cache_examples=False,
-    interpretation="clear",
+    retry_btn=None,
+    undo_btn="Delete Previous",
     clear_btn="Clear",
     additional_inputs=[
         gr.Textbox(value="You are a friendly Chatbot.", label="System message"),