Update app.py

app.py

@@ -20,21 +20,29 @@ def chatbot_initialize():
     Chatbot = core.chatbot.RetrievalChatbot(retriever=retriever)
     return Chatbot
 
-def respond(query, additional_inputs, image):
+def respond(query, chat_history, img_path, chat_history_string):
     global Chatbot
-    response = Chatbot.response(query, image)
-    for i in range(len(response)):
-        time.sleep(0.01)
-        yield response[: i+1]
+    response, logs = Chatbot.response(query, image_path=img_path, return_logs=True)
+    chat_history.append((query, response))
+    if img_path is None:
+        chat_history_string += "Query: " + query + "\nImage: None" + "\nRepsonse: " + response + "\n\n\n"
+    else:
+        chat_history_string += "Query: " + query + "\nImage: " + img_path + "\nRepsonse: " + response + "\n\n\n"
+    return "", chat_history, logs, chat_history_string
 
 if __name__ == "__main__":
     global Chatbot
     Chatbot=chatbot_initialize()
 
-    demo = gr.ChatInterface(
-        fn=respond,
-        additional_inputs=[
-            gr.Image(type="filepath"),
-        ]
-    )
+    with gr.Blocks() as demo:
+        with gr.Row():
+            with gr.Column(scale=2):
+                chatbot = gr.Chatbot()
+                msg = gr.Textbox(label="Query", show_label=True)
+                img = gr.Image(type="filepath")
+                clear = gr.ClearButton([msg, chatbot])
+            with gr.Column(scale=1):
+                sidebar = gr.Textbox(label="Subquestions", show_label=True, show_copy_button=True, interactive=False, max_lines=30)
+                history = gr.Textbox(label="Copy Chat History", show_label=True, show_copy_button=True, interactive=False, max_lines=5)
+            msg.submit(respond, inputs=[msg, chatbot, img, history], outputs=[msg, chatbot, sidebar, history])
     demo.queue().launch()