Update utils.py

utils.py

@@ -1,3 +1,35 @@
+import os
+import openai
+import random
+import streamlit as st
+from datetime import datetime
+
+#decorator
+def enable_chat_history(func):
+    if os.environ.get("OPENAI_API_KEY"):
+
+        # to clear chat history after swtching chatbot
+        current_page = func.__qualname__
+        if "current_page" not in st.session_state:
+            st.session_state["current_page"] = current_page
+        if st.session_state["current_page"] != current_page:
+            try:
+                st.cache_resource.clear()
+                del st.session_state["current_page"]
+                del st.session_state["messages"]
+            except:
+                pass
+
+        # to show chat history on ui
+        if "messages" not in st.session_state:
+            st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
+        for msg in st.session_state["messages"]:
+            st.chat_message(msg["role"]).write(msg["content"])
+
+    def execute(*args, **kwargs):
+        func(*args, **kwargs)
+    return execute
+
 def display_msg(msg, author):
     """Method to display message on the UI
     Args:
@@ -42,4 +74,4 @@ def configure_openai():
         print(e)
         st.error("Something went wrong. Please try again later.")
         st.stop()
-    return model
+    return model