Update app.py

app.py

@@ -35,21 +35,23 @@ from typing import Dict, Optional
 import chainlit as cl
 
 
-@cl.password_auth_callback
-def auth_callback(username: str, password: str):
-    # Fetch the user matching username from your database
-    # and compare the hashed password with the value stored in the database
-    if (username, password) == ("admin", "admin"):
-        return cl.User(
-            identifier="admin", metadata={"role": "admin", "provider": "credentials"}
-        )
-    else:
-        return None
-
-
 print(os.getenv("OPENID_PROVIDER_URL"))
 print(os.getenv("OAUTH_CLIENT_ID"))
 
+
+@cl.oauth_callback
+def oauth_callback(
+  provider_id: str,
+  token: str,
+  raw_user_data: Dict[str, str],
+  default_user: cl.User,
+) -> Optional[cl.User]:
+  if provider_id == "google":
+    if raw_user_data["hd"] == "example.org":
+      return default_user
+  return None
+
+
 @cl.on_chat_start
 async def on_chat_start():
     app_user = cl.user_session.get("user")