uuid

app.py

@@ -44,7 +44,8 @@ def load_config(request: gr.Request):
     survey_template = config['template']
     initial_message = config['initial_message']
     model_args = {'model': config['model'], 'temperature': config['temperature']}
-    return survey_question, survey_template, initial_message, model_args
+    userid = config['userid']
+    return survey_question, survey_template, initial_message, model_args, userid
 
 # Post-loading
 def update_system_message(question: str, template: PromptTemplate | str) -> str:
@@ -87,10 +88,10 @@ def initialize_interview(
 
 
 def initialize_tracker(
-    model_args: dict[str, str | float], question: str, template: PromptTemplate
+    model_args: dict[str, str | float], question: str, template: PromptTemplate, userid=str
 ):
     "Initializes wandb run for interview"
-    run_config = model_args | {"question": question, "template": str(template)}
+    run_config = model_args | {"question": question, "template": str(template), "userid": userid}
     wandb.init(project="cognitive-debrief", config=run_config, tags=["dev"])
 
 
@@ -183,6 +184,7 @@ with gr.Blocks() as demo:
     initialMessage = gr.Textbox(visible=False)
     systemMessage = gr.Textbox(visible=False)
     modelArgs = gr.State(value={"model": "", "temperature": ""})
+    userid = gr.Textbox(visible=False)
 
     # Debugging
     # with gr.Accordion("Debugging Panel", open=False):
@@ -226,7 +228,7 @@ with gr.Blocks() as demo:
             startInterview,
             resetButton,
         ],
-    ).then(initialize_tracker, inputs=[modelArgs, surveyQuestion, surveyTemplate])
+    ).then(initialize_tracker, inputs=[modelArgs, surveyQuestion, surveyTemplate, userid])
 
     chatInput.submit(
         user_message,