Update app.py

app.py

@@ -176,7 +176,7 @@ def display_sample_query_boxes(key_prefix=""):
 def display_chat_history(task: str):
     for entry in st.session_state.get(f"all_chat_history_{task}", []):
         if entry.get("query"):
-            st.chat_message("assistant").write(entry["query"])
+            st.chat_message("user").write(entry["query"])
         if entry.get("fig"):
             st.plotly_chart(entry["fig"], use_container_width=True)
         if entry.get("df") is not None: