Add advanced tab to working basic tab

app.py

@@ -1,5 +1,6 @@
 #@title Prepare the Concepts Library to be used
 
+
 import requests
 import os
 import gradio as gr
@@ -21,7 +22,6 @@ models_list = api.list_models(author="sd-concepts-library", sort="likes", direct
 models = []
 
 my_token = os.environ['api_key']
-# my_token = "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" # 🤫
 
 pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
 
@@ -98,9 +98,97 @@ for model in ahx_model_list:
   models.append(model_content)
 
 
+# -----------------------------------------------------------------------------------------------
+
+
+
+#@title Dropdown Prompt Tab
+
+model_tags = [model.modelId.split("/")[1] for model in ahx_model_list]
+model_tags.sort()
+
+
+import random 
+
+
+#@title Gradio Concept Loader
+DROPDOWNS = {}
+
+for model in model_tags:
+  if model != "ahx-model-1" and model != "ahx-model-2":
+    DROPDOWNS[model] = f" in the style of <{model}>"
+
+# def image_prompt(prompt, dropdown, guidance, steps, seed, height, width):
+def image_prompt(prompt, guidance, steps, seed, height, width):
+  # prompt = prompt + DROPDOWNS[dropdown]
+  generator = torch.Generator(device="cuda").manual_seed(int(seed))
+  return (
+      pipe(prompt=prompt, guidance_scale=guidance, num_inference_steps=steps, generator=generator, height=int((height // 8) * 8), width=int((width // 8) * 8)).images[0], 
+      f"prompt = '{prompt}'\nseed = {int(seed)}\nguidance_scale = {guidance}\ninference steps = {steps}\nheight = {int((height // 8) * 8)}\nwidth = {int((width // 8) * 8)}"
+      )
+
+
+def default_guidance():
+  return 7.5
+
+def default_steps():
+  return 30
+
+def default_pixel():
+  return 768
+
+def random_seed():
+  return random.randint(0, 99999999999999) # <-- this is a random gradio limit, the seed range seems to actually be 0-18446744073709551615
+
+with gr.Blocks(css=".gradio-container {max-width: 650px}") as dropdown_tab:
+  # gr.Markdown("<u>styles</u>: check out examples of these at https://www.astronaut.horse/collaborations")
+
+  # dropdown = gr.Dropdown(list(DROPDOWNS), label="choose style...")
+  # gr.Markdown("<u>styles</u>: check out examples of these at https://www.astronaut.horse/collaborations")
+
+  with gr.Row():
+    prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
+  with gr.Row():
+    seed = gr.Slider(0, 99999999999999, label="seed", dtype=int, value=random_seed, interactive=True, step=1)
+  with gr.Row():
+    with gr.Column():
+      guidance = gr.Slider(0, 10, label="guidance", dtype=float, value=default_guidance, step=0.1, interactive=True)
+    with gr.Column():
+      steps = gr.Slider(1, 100, label="inference steps", dtype=int, value=default_steps, step=1, interactive=True)
+  with gr.Row():
+    with gr.Column():
+      height = gr.Slider(50, 3500, label="height", dtype=int, value=default_pixel, step=1, interactive=True)
+    with gr.Column():
+      width = gr.Slider(50, 3500, label="width", dtype=int, value=default_pixel, step=1, interactive=True)
+  gr.Markdown("<u>heads-up</u>: height multiplied by width should not exceed about 195,000 or an error will occur so don't go too nuts")
+
+  go_button = gr.Button("generate image", elem_id="go-button")
+  output = gr.Image(elem_id="output-image")
+  output_text = gr.Text(elem_id="output-text")
+  # go_button.click(fn=image_prompt, inputs=[prompt, dropdown, guidance, steps, seed, height, width], outputs=[output, output_text])
+  go_button.click(fn=image_prompt, inputs=[prompt, guidance, steps, seed, height, width], outputs=[output, output_text])
+  gr.Markdown('''
+    ## Prompt Examples Using Artist Tokens:
+    * "an alien in the style of \<ahx-model-12>"
+    * "a painting in the style of \<ahx-model-11>"
+    * "a landscape in the style of \<ahx-model-10> and \<ahx-model-14> "
+
+    ## Valid Artist Tokens:
+    * \<ahx-model-3>
+    * \<ahx-model-4>
+    * \<ahx-model-6>
+    * \<ahx-model-7>
+    * \<ahx-model-9>
+    * \<ahx-model-10>
+    * \<ahx-model-11>
+    * \<ahx-model-12>
+    * \<ahx-model-13>
+    * \<ahx-model-14>
+  ''')
+
+# -----------------------------------------------------------------------------------------------
 
 
-#@title Prepare the Concepts Library to be used
 
 SELECT_LABEL = "Select concept"
 def assembleHTML(model):
@@ -153,31 +241,31 @@ with gr.Blocks(css=css) as demo:
   gr.Markdown('''
       # 🧑‍🚀 Astronaut Horse Concept Loader
 
-      Run your own text prompts into fine-tuned artist concepts, see the example below. Currently only loading two artist concepts while testing. Soon will automatically be able to add all concepts from astronaut horse artist collaborations. There's some buggy stuff here that'll be cleared up next week but I wanted to at least get this usable for the weekend!
-      
-      http://www.astronaut.horse
+      This tool allows you to run your own text prompts into fine-tuned artist concepts from an ongoing series of Stable Diffusion collaborations with visual artists linked below. Select an artist's fine-tuned concept / model from the dropdown and enter any desired text prompt. You can check out example output images and project details on the project's webpage. Additionally if you can play around with more controls in the Advanced Prompting tab. Enjoy!
+      <a href="http://www.astronaut.horse">http://www.astronaut.horse</a>
   ''')
   with gr.Row():
         with gr.Column():
-          with gr.Box():
-                  with gr.Row(elem_id="prompt_area").style(mobile_collapse=False, equal_height=True):
-                      text = gr.Textbox(
-                          label="Enter your prompt", placeholder="Enter your prompt", show_label=False, max_lines=1, elem_id="prompt_input"
-                      ).style(
-                          border=(True, False, True, True),
-                          rounded=(True, False, False, True),
-                          container=False,
-                      )
-                      btn = gr.Button("generate image",elem_id="run_btn").style(
-                          margin=False,
-                          rounded=(False, True, True, False),
-                      )  
-                  # with gr.Row().style():
-                  infer_outputs = gr.Gallery(show_label=False, elem_id="generated-gallery").style(grid=[1], height="512px")
-                  with gr.Row():
-                    gr.HTML("<p></p>")
-                  with gr.Row():
-                    gr.Examples(examples=examples, fn=infer_examples, inputs=[text], outputs=infer_outputs, cache_examples=True)
+          # with gr.Box():
+          dropdown = gr.Dropdown(list(DROPDOWNS), label="choose style...")
+          with gr.Row(elem_id="prompt_area").style(mobile_collapse=False, equal_height=True):
+              text = gr.Textbox(
+                  label="Enter your prompt", placeholder="Enter your prompt", show_label=False, max_lines=1, elem_id="prompt_input"
+              ).style(
+                  border=(True, False, True, True),
+                  rounded=(True, False, False, True),
+                  container=False,
+              )
+              btn = gr.Button("generate image",elem_id="run_btn").style(
+                  margin=False,
+                  rounded=(False, True, True, False),
+              )  
+          # with gr.Row().style():
+          infer_outputs = gr.Gallery(show_label=False, elem_id="generated-gallery").style(grid=[1])
+          with gr.Row():
+            gr.HTML("<p></p>")
+          with gr.Row():
+            gr.Examples(examples=examples, fn=infer_examples, inputs=[text], outputs=infer_outputs, cache_examples=True)
           with gr.Group(elem_id="share-btn-container"):
             community_icon = gr.HTML(community_icon_html, visible=False)
             loading_icon = gr.HTML(loading_icon_html, visible=False)
@@ -188,24 +276,10 @@ with gr.Blocks(css=css) as demo:
         inputs=inputs,
         outputs=[infer_outputs, community_icon, loading_icon]
     )
-  gr.Markdown('''
-      ## Prompt Examples Using Artist Token:
-      * "a painting in the style of \<ahx-model-11>"
-      * "an alien in the style of \<ahx-model-12>"
-
-      ## Valid Artist Tokens:
-      * \<afx-model-3>
-      * \<afx-model-4>
-      * \<afx-model-6>
-      * \<afx-model-7>
-      * \<afx-model-9>
-      * \<afx-model-10>
-      * \<afx-model-11>
-      * \<afx-model-12>
-      * \<afx-model-13>
-      * \<afx-model-14>
-  ''')
-# demo.queue(max_size=20).launch()
+  
+  # -----------------------------------------------------------------------------------------------
+
+
 
-tabbed_interface = gr.TabbedInterface([demo.queue(max_size=20)], ["Open Prompter"])
-tabbed_interface.launch()
+tabbed_interface = gr.TabbedInterface([demo.queue(max_size=20), dropdown_tab], ["Welcome!", "Advanced Prompting"])
+tabbed_interface.launch(share=True)