Update app.py

app.py

@@ -27,15 +27,25 @@ def predict(input, history=[]):
                             
     # convert the tokens to text, and then split the responses into lines
     response = tokenizer.decode(history[0]).split("<|endoftext|>")
-    response = [(response[i], response[i+1]) for i in range(0, len(response)-1, 2)]  # convert to tuples of list
-    return response, history
+    #response = [(response[i], response[i+1]) for i in range(0, len(response)-1, 2)]  # convert to tuples of list
+    response.remove("")
     
-
+    # write some HTML
+    html = "<div class='chatbot'>"
+    for m, msg in enumerate(response):
+        cls = "user" if m%2 == 0 else "bot"
+        html += "<div class='msg {}'> {}</div>".format(cls, msg)
+    html += "</div>"
+    
+    #return response, history
+    return html, history
+    
+chatbot = gr.Chatbot(color_map=("green", "gray"))
 
 gr.Interface(fn=predict,
     title="DialoGPT-large",
     inputs=["text", "state"],
-    outputs=["chatbot", "state"],
-    allow_screenshot=False,
-    allow_flagging='never').launch()
+    outputs=["html", "state"],
+    css=css
+    ).launch()