Update app.py

app.py

@@ -81,11 +81,7 @@ def predict(input, conversation_history): #history=[]):
     else:
     '''
 
-    global conversation_history
-    
-    conversation_history.append(user_input)
-    
-    bot_input = dialogpt_tokenizer.encode(input + " ".join(conversation_history)+ tokenizer.eos_token, return_tensors="pt")
+    bot_input = dialogpt_tokenizer.encode(input + tokenizer.eos_token, return_tensors="pt")
     chat_history_ids = model.generate(bot_input, max_length=1000, pad_token_id=tokenizer.eos_token_id)
     response = tokenizer.decode(chat_history_ids[:, bot_input.shape[-1]:][0], skip_special_tokens=True)
     #return response
@@ -107,7 +103,16 @@ def predict(input, conversation_history): #history=[]):
     
     return response #, history
 
+def chatbot_interface(user_input, table=gr.inputs.Textbox()):
+    global conversation_history
+    
+    conversation_history.append(user_input)
 
+    dialog_prompt = "User: " + " ".join(conversation_history) + "\nBot:"
+    response = predict(dialog_prompt, conversation_history)
+    conversation_history.append(response)
+    return "Bot (DialoGPT): " + response
+    
 def sqlquery(input):
 
     sql_encoding = sql_tokenizer(table=table, query=input + sql_tokenizer.eos_token, return_tensors="pt")
@@ -118,6 +123,8 @@ def sqlquery(input):
 
 
 chat_interface = gr.Interface(
+    fn=chatbot_interface, inputs=["text", "text"], outputs="text", live=True
+    '''
     fn=predict,
     theme="default",
     css=".footer {display:none !important}",
@@ -125,6 +132,7 @@ chat_interface = gr.Interface(
     outputs="text",   #["chatbot", "state"],
     title="ST Chatbot",
     description="Type your message in the box above, and the chatbot will respond.",
+    '''
 )
 
 sql_interface = gr.Interface(