Update app.py

app.py

@@ -9,6 +9,7 @@ from langchain_experimental.sql.base import SQLDatabaseChain
 import streamlit as st
 import pyodbc                                                                                           
 import openai
+import hmac
 from langchain_openai import AzureChatOpenAI
 
 from tabulate import tabulate
@@ -22,6 +23,8 @@ openai.api_key = os.environ['OPENAI_API_KEY']
 openai.api_type = 'azure'
 openai.api_base = os.environ['AZURE_OPENAI_ENDPOINT']
 openai.api_version = os.environ['OPENAI_API_VERSION']
+password = os.environ['app_password']
+
 deployment_name = "gpt-4o"
 
 print(pyodbc.drivers()) 
@@ -96,6 +99,29 @@ template = """
             Question: {input}
             """
 
+def check_password():
+    """Returns `True` if the user had the correct password."""
+
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        if hmac.compare_digest(st.session_state["password"], password):
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # Don't store the password.
+        else:
+            st.session_state["password_correct"] = False
+
+    # Return True if the password is validated.
+    if st.session_state.get("password_correct", False):
+        return True
+
+    # Show input for password.
+    st.text_input(
+        "Password", type="password", on_change=password_entered, key="password"
+    )
+    if "password_correct" in st.session_state:
+        st.error("😕 Password incorrect")
+    return False
+    
 if __name__ == '__main__':
     connection_string = ("Driver=FreeTDS;Server=crawlersdb.c3pzpntwjvdf.us-east-1.rds.amazonaws.com;Database=SmartCleverST;PORT=1433;UID=CleverData;PWD={};TrustServerCertificate=yes;".format(os.environ['DB_PWD'])
     )
@@ -120,7 +146,7 @@ if __name__ == '__main__':
         })
 
     #question = st.text_input("Ask a question in natural language and press enter")
-
+    
     if "messages" not in st.session_state:
         st.session_state.messages = []
 
@@ -135,6 +161,9 @@ if __name__ == '__main__':
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    if not check_password():
+        st.stop()  # Do not continue if check_password is not True.
+        
     question = st.chat_input("Ask a question in natural language and press enter")
         
     #if 'questions' not in st.session_state: