Update app.py

app.py

@@ -29,11 +29,13 @@ date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
 #client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 history = []
 
-def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice,seed):
+def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice,seed,im_seed):
     #out_img = infer(out_prompt)
     history.clear()
     if seed == 0:
         seed = random.randint(1,1111111111111111)
+    if im_seed == 0:
+        im_seed = random.randint(1,1111111111111111)        
     out_prompt=generate(purpose,history,chat_drop,seed)
     history.append((purpose,out_prompt))
     yield (history,None)        
@@ -50,7 +52,7 @@ def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice,seed):
         num_inference_steps=None,
         guidance_scale=None,
         model=None,
-        seed=1,
+        seed=im_seed,
         )
     yield (history,out_img)
 
@@ -183,13 +185,17 @@ with gr.Blocks(css=style) as iface:
             with gr.Row():
                 agent_choice = gr.Radio(choices=["Generate","Describe"],value="Generate")
             msg = gr.Textbox()
-            model_drop=gr.Dropdown(label="Diffusion Models", type="index", choices=[m for m in models], value=models[0])
-            chat_model_drop=gr.Dropdown(label="Chatbot Models", type="index", choices=[m for m in c_models], value=c_models[0])
-            chat_seed=gr.Slider(label="Seed (0 for random)", minimum=0,maximum=1000000000000,
-                                value=random.randint(1,1000000000000),step=1,
-                                interactive=True,
-                               info="Set Seed to 0 to randomize the session")
-
+            with gr.Accordion("Controls", open=False):
+                model_drop=gr.Dropdown(label="Diffusion Models", type="index", choices=[m for m in models], value=models[0])
+                chat_model_drop=gr.Dropdown(label="Chatbot Models", type="index", choices=[m for m in c_models], value=c_models[0])
+                chat_seed=gr.Slider(label="Prompt Seed", minimum=0,maximum=1000000000000,
+                                    value=random.randint(1,1000000000000),step=1,
+                                    interactive=True,
+                                   info="Set Seed to 0 to randomize the session")
+                image_seed=gr.Slider(label="Image Seed", minimum=0,maximum=1000000000000,
+                                    value=random.randint(1,1000000000000),step=1,
+                                    interactive=True,
+                                   info="Set Seed to 0 to randomize the session")
             with gr.Group():
                 with gr.Row():
                     submit_b = gr.Button()
@@ -199,7 +205,7 @@ with gr.Blocks(css=style) as iface:
         with gr.Column(scale=2):
             sumbox=gr.Image(label="Image")
 
-    run_test = test_btn.click(gen_from_infer, [msg,chatbot,sumbox,model_drop,chat_model_drop,agent_choice,chat_seed],[chatbot,sumbox],concurrency_limit=20)
+    run_test = test_btn.click(gen_from_infer, [msg,chatbot,sumbox,model_drop,chat_model_drop,agent_choice,chat_seed,image_seed],[chatbot,sumbox],concurrency_limit=20)
     
     sub_b = submit_b.click(run, [msg,chatbot,sumbox,model_drop,chat_model_drop,agent_choice,chat_seed],[chatbot,sumbox],concurrency_limit=20)
     sub_e = msg.submit(run, [msg, chatbot,sumbox,model_drop,chat_model_drop,agent_choice,chat_seed], [chatbot,sumbox],concurrency_limit=20)