app

app.py

@@ -52,16 +52,18 @@ def predict(input, history=[]):
     bot_text = bot_text_temp
     print("Bot: {}".format(bot_text))    
 
+    history.append((input, bot_text))
+
     # write some HTML
-    # html = "<div class='chatbot'>"
-    # for chat in chat_history_all:
-    #     html += "<div class='msg {}'> {}</div>".format(chat[0], chat[1])
-    # html += "</div>"
+    html = "<div class='chatbot'>"
+    for chat in history:
+        html += "<div class='msg {}'> {}</div>".format("User", chat[0])
+        html += "<div class='msg {}'> {}</div>".format("Bot", chat[1])
+    html += "</div>"
     
-    #return html, chat_history_all
+    return html, history
 
-    history.append((input, bot_text))
-    return history, history
+    # return history, history
 
 import gradio as gr
 
@@ -73,17 +75,17 @@ css = """
 .footer {display:none !important}
 """
 
-# gr.Interface(fn=predict,
-#              theme="default",
-#              inputs=[gr.inputs.Textbox(placeholder="밥은 먹었냐?"), "state"],
-#              outputs=["html", "state"],
-#              css=css).launch()
+gr.Interface(fn=predict,
+             theme="default",
+             inputs=[gr.inputs.Textbox(placeholder="밥은 먹었냐?"), "state"],
+             outputs=["html", "state"],
+             css=css).launch()
 
-chatbot = gr.Chatbot().style(color_map=("green", "pink"))
-demo = gr.Interface(
-    predict,
-    ["text", "state"],
-    [chatbot, "state"],
-    allow_flagging="never",
-)
-demo.launch()                     
+# chatbot = gr.Chatbot().style(color_map=("green", "pink"))
+# demo = gr.Interface(
+#     predict,
+#     ["text", "state"],
+#     [chatbot, "state"],
+#     allow_flagging="never",
+# )
+# demo.launch()                     

requirements.txt

@@ -1,4 +1,3 @@
 transformers
 torch
-nltk
-gradio==3.4.0
+nltk