added authentication

utils/login.py

@@ -1,5 +1,10 @@
 import streamlit as st
 import os
+import requests
+import os
+import json
+
+# Fetch the service account key JSON file contents
 
 
 def initialize_login():
@@ -7,9 +12,9 @@ def initialize_login():
         st.columns(3)[1].image("assets/logo.png")
         username = st.text_input("Username")
         password = st.text_input("Password", type="password")
+
         if st.button("Login"):
-            # TODO: replace with actual authorization check
-            authorized = {"status": True, "Name": "John Doe", "username": "johndoe"}
+            authorized = authenticate(username, password)
             if authorized["status"]:
                 st.session_state["login"] = authorized
                 os.makedirs(
@@ -25,5 +30,31 @@ def initialize_login():
         st.sidebar.image("assets/logo.png", use_column_width=True)
 
 
+def authenticate(username, password):
+    FIREBASE_WEB_API_KEY = os.environ.get("FIREBASE_WEB_API_KEY")
+    payload = json.dumps(
+        {
+            "email": f"{username}@aieye.com",
+            "password": password,
+            "returnSecureToken": False,
+        }
+    )
+
+    rest_api_url = (
+        f"https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword"
+    )
+    r = requests.post(rest_api_url, params={"key": FIREBASE_WEB_API_KEY}, data=payload)
+    print(r.json())
+
+    if r.status_code == 200:
+        return {
+            "status": True,
+            "Name": " ".join(username.split("_")),
+            "username": username,
+        }
+    else:
+        return {"status": False}
+
+
 def get_login():
     return st.session_state.get("login", {"status": False})