Update app.py

app.py

@@ -17,8 +17,36 @@ import google.generativeai as genai
 
 
 
+
+
 load_dotenv()
 
+def check_password():
+    def password_entered():
+        if hmac.compare_digest(st.session_state["username"], os.environ.get("STREAMLIT_USERNAME", "")) and \
+           hmac.compare_digest(st.session_state["password"], os.environ.get("STREAMLIT_PASSWORD", "")):
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]
+            del st.session_state["username"]
+        else:
+            st.session_state["password_correct"] = False
+            
+    if "password_correct" not in st.session_state:
+        st.text_input("Username", key="username")
+        st.text_input("Password", type="password", key="password")
+        st.button("Login", on_click=password_entered)
+        return False
+    
+    # Password correct
+    elif not st.session_state["password_correct"]:
+        st.text_input("Username", key="username")
+        st.text_input("Password", type="password", key="password")
+        st.button("Login", on_click=password_entered)
+        st.error("User not known or password incorrect")
+        return False
+    
+    return True
+
 togetherai_client = OpenAI(
     api_key=os.getenv('TOGETHERAI_API_KEY'),
     base_url="https://api.together.xyz/v1"