Update app.py

app.py

@@ -83,25 +83,59 @@ def get_chat_response(name, input_txt = "Hello , what is your name?",history=[])
 | Brother | Allnut | Rose | Qui-Gon | Jar Jar
 '''
 
-def greet(character,input_txt):
-  ##add a bootstrap history (item 1 is speaker, item 2 is character)
-  history = ["Who are you?","I am " + character]
-  return  get_chat_response(character,history=history,input_txt=input_txt)
 
 
-#some selected ones are in for demo use
-personality_choices = ["Gandalf", "Riddick", "Macleod", "Morpheus", "Neo","Spock","Vader","Indy", "Ig-11","Threepio","Tony Stark","Batman","Vizzini"]
+
+
+def greet(character,message):
+  print(gr.get_state())
+  history = gr.get_state() or  {"character": character, "message_history" : [] }
+  """
+  if history["character"] != character:
+    #switching character
+    history = {"character": character, "message_history" : [] }
+  """
+
+  response = get_chat_response(character,history=history["message_history"],input_txt=message)
+  print(response)
+
+  history["message_history"].append((message, response))
+
+  gr.set_state(history)
+
+  print(history)
+
+  html = "<div class='chatbot'>"
+  for user_msg, resp_msg in history["message_history"]:
+      html += f"<div class='user_msg'>You: {user_msg}</div>"
+      html += f"<div class='resp_msg'>{character}: {resp_msg}</div>"
+  html += "</div>"
+
+  return html 
+
+personality_choices = ["Gandalf", "Riddick", "Macleod", "Morpheus", "Neo","Spock","Vader","Indy"]
 
 examples= ["Gandalf", "What is your name?"]
 
+css="""
+    .chatbox {display:flex;flex-direction:column}
+    .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
+    .user_msg {background-color:cornflowerblue;color:white;align-self:start}
+    .resp_msg {background-color:lightgray;align-self:self-end}
+"""
 
 
+#some selected ones are in for demo use
+personality_choices = ["Gandalf", "Riddick", "Macleod", "Morpheus", "Neo","Spock","Vader","Indy", "Ig-11","Threepio","Tony Stark","Batman","Vizzini"]
 title = "Metayazar - Movie Chatbot"
 description = "Chat with your favorite movie characters. This space demo has simple interface and no history, test it out in metayazar.com/chatbot for more movie/character options and history memorized."
 
 #History not implemented in this demo, use metayazar.com/chatbot for a movie and character dropdown chat interface
-interface = gr.Interface(fn=greet, inputs=[gr.inputs.Dropdown(personality_choices) ,"text"], title=title, description=description, outputs="text")
+##interface = gr.Interface(fn=greet, inputs=[gr.inputs.Dropdown(personality_choices) ,"text"], title=title, description=description, outputs="text")
 
 
+interface= gr.Interface(fn=greet, inputs=[gr.inputs.Dropdown(personality_choices) ,"text"], outputs="html",css=css, title=title, description=description)
+      
+      
 if __name__ == "__main__":
     interface.launch()