cdupland commited on
Commit
24052d0
·
verified ·
1 Parent(s): fc8b475

Test user ID

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -12,11 +12,18 @@ load_dotenv()
12
 
13
  GROUP_NAME = os.environ.get("APP_NAME")
14
  LOGO = "assets/logo.png"
 
15
 
16
  def init_app():
17
 
18
  config = getYamlConfig()
19
 
 
 
 
 
 
 
20
  if len(st.session_state) == 0:
21
  # Define Vectore store strategy
22
  pinecone_connector = PineconeConnector()
@@ -48,6 +55,8 @@ def main():
48
  st.logo(LOGO)
49
  st.title(GROUP_NAME)
50
 
 
 
51
  saved_documents = st.Page("pages/persistent_documents.py", title="Communs", icon="🗃️")
52
  documents = st.Page("pages/documents.py", title="Vos documents", icon="📂")
53
  prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="🖊️", default=True)
 
12
 
13
  GROUP_NAME = os.environ.get("APP_NAME")
14
  LOGO = "assets/logo.png"
15
+ userId = None
16
 
17
  def init_app():
18
 
19
  config = getYamlConfig()
20
 
21
+ query_params = st.query_params
22
+ if "user" in query_params:
23
+ global userId
24
+ userId = query_params["user"]
25
+
26
+
27
  if len(st.session_state) == 0:
28
  # Define Vectore store strategy
29
  pinecone_connector = PineconeConnector()
 
55
  st.logo(LOGO)
56
  st.title(GROUP_NAME)
57
 
58
+ st.write(f"**User ID:** {userId}")
59
+
60
  saved_documents = st.Page("pages/persistent_documents.py", title="Communs", icon="🗃️")
61
  documents = st.Page("pages/documents.py", title="Vos documents", icon="📂")
62
  prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="🖊️", default=True)