user dict for gallery output

app.py

@@ -54,8 +54,7 @@ model_bundle = DirectedDiffusion.AttnEditorUtils.load_all_models(
 #     model_path_diffusion="../DirectedDiffusion/assets/models/stable-diffusion-v1-4"
 # )
 
-ALL_OUTPUT = []
-
+ALL_OUTPUT = {}
 
 def directed_diffusion(
     in_prompt,
@@ -112,6 +111,10 @@ def run_it(
     progress=gr.Progress(),
 ):
     global ALL_OUTPUT
+
+    if os.getlogin() not in ALL_OUTPUT:
+        ALL_OUTPUT[os.getlogin()] = []
+
     num_affected_steps = [in_slider_ddsteps]
     noise_scale = [in_slider_gcoef]
     num_trailing_attn = [in_slider_trailings]
@@ -127,6 +130,7 @@ def run_it(
     progress(0, desc="Starting...")
     for i, element in enumerate(progress.tqdm(param_list)):
         print("=========== Arguments ============")
+        print("User:", os.getlogin())
         print("Prompt:", in_prompt)
         print("BoundingBox:", in_bb)
         print("Token indices:", in_token_ids)
@@ -153,13 +157,13 @@ def run_it(
                 ),
             )
         )
-    ALL_OUTPUT += results
-    return ALL_OUTPUT
+    ALL_OUTPUT[os.getlogin()] += results
+    return ALL_OUTPUT[os.getlogin()]
 
 def clean_gallery():
     global ALL_OUTPUT
-    ALL_OUTPUT = []
-    return ALL_OUTPUT
+    ALL_OUTPUT[os.getlogin()] = []
+    return ALL_OUTPUT[os.getlogin()]