Update app.py

app.py

@@ -19,11 +19,11 @@ def query(payload):
 context = "To extract information from documents, use sentence similarity task. To do sentiment analysis from tweets, use text classification task. To detect masks from images, use object detection task. To extract information from invoices, use named entity recognition from token classification task."
 
 
-message_history = ["Let's find out the best task for your use case! Tell me about your use case :)"]
+message_history = [{"text":"Let's find out the best task for your use case! Tell me about your use case :)", is_user = False}]
 
 
 for msg in message_history:
-    message(msg)   # display all the previous message
+    message(msg["text"], is_user = msg["is_user"])   # display all the previous message
 
 placeholder = st.empty()  # placeholder for latest message
 input = st.text_input("Ask me 🤗")
@@ -36,6 +36,7 @@ with placeholder.container():
         
 
 message(input, is_user=True)  # align's the message to the right
+message_history.append({"text":input, "is_user" = True})
 
 data, resp = query(
     {
@@ -46,10 +47,13 @@ data, resp = query(
     }
 )
 if resp.status_code == 200:
+
     model_answer = data["answer"]
     response_templates = [f"{model_answer} is the best task for this 🤩", f"I think you should use {model_answer} 🪄", f"I think {model_answer} should work for you 🤓"]
+    
     bot_answer = random.choice(response_templates)
-    message_history.append(bot_answer)
+    
+    message_history.append({"text":bot_answer, "is_user" : False})