Update app.py

app.py

@@ -35,15 +35,14 @@ from typing import Dict, Optional
 import chainlit as cl
 
 
-@cl.oauth_callback
-def oauth_callback(
-  provider_id: str,
-  token: str,
-  raw_user_data: Dict[str, str],
-  default_user: cl.User,
-) -> Optional[cl.User]:
-  return default_user
-
+@cl.header_auth_callback
+def header_auth_callback(headers: Dict) -> Optional[cl.User]:
+  # Verify the signature of a token in the header (ex: jwt token)
+  # or check that the value is matching a row from your database
+  if headers.get("test-header") == "test-value":
+    return cl.User(identifier="admin", metadata={"role": "admin", "provider": "header"})
+  else:
+    return None
 
 @cl.on_chat_start
 async def on_chat_start():