use chatbot instead of chatinterface

app.py

@@ -13,6 +13,7 @@ from autorag.evaluator import Evaluator
 from src.data.chunk import chunk
 from src.data.parse import parse_pdf
 from src.runner import GradioStreamRunner
+from gradio import ChatMessage
 
 root_dir = os.path.dirname(os.path.realpath(__file__))
 
@@ -93,7 +94,7 @@ def set_runner(project_dir):
     gradio_runner = runner
 
 
-def get_response(message, history):
+def get_response(history):
     global gradio_runner
     if gradio_runner is None:
         gradio.Warning("Please set the AutoRAG server first.")
@@ -102,9 +103,14 @@ def get_response(message, history):
         gradio.Warning("Please submit your OpenAI API key first.")
         return
 
-    for output in gradio_runner.stream_run(message):
-        yield output[0]
+    history.append({"role": "assistant", "content": ""})
+    for output in gradio_runner.stream_run(history[-2]["content"]):
+        stream_delta = output[0]
+        history[-1]["content"] = stream_delta
+        yield history
 
+def user(user_message, history: list):
+    return "", history + [{"role": "user", "content": user_message}]
 
 # interface one
 with gr.Blocks(theme="earneleh/paris") as demo:
@@ -129,9 +135,15 @@ with gr.Blocks(theme="earneleh/paris") as demo:
                 button.click(file_ingest, inputs=[file_input, gr.State(project_dir)], outputs=[text_output])
 
             with gr.Column(scale=7):
-                gr.ChatInterface(
-                    get_response, title="This is your Naive RAG Chatbot 🚀",
+                gr.Markdown("## This is your Naive RAG Chatbot 🚀")
+                chatbot = gr.Chatbot(type="messages")
+                chat_input = gr.Textbox()
+                clear = gr.Button(value="Clear Chat🗑️")
+
+                chat_input.submit(user, [chat_input, chatbot], outputs=[chat_input, chatbot], queue=False).then(
+                    get_response, inputs=chatbot, outputs=[chatbot]
                 )
+                clear.click(lambda: None, None, chatbot, queue=False)
 
         gr.Markdown("## Do you like the result?\n\nIf you don't like it, try to optimize it with AutoRAG. Press below button and go to make evaluation data and optimize it. Both on the Huggingface space so you don't need to install anything.")
         with gr.Row():
@@ -139,5 +151,7 @@ with gr.Blocks(theme="earneleh/paris") as demo:
                                            link="https://huggingface.co/spaces/AutoRAG/AutoRAG-data-creation")
             open_optimize = gr.Button(value="2️⃣ : Optimize", link="https://www.auto-rag.com/")
 
+if __name__ == "__main__":
+    demo.launch(share=False, debug=True)
 
-demo.launch(share=False, debug=True)
+# demo.launch(share=False, debug=True)