Minor overhaul to tabs from semi-working version

app.py

@@ -101,7 +101,6 @@ for model in ahx_model_list:
 # -----------------------------------------------------------------------------------------------
 
 
-
 #@title Dropdown Prompt Tab
 
 model_tags = [model.modelId.split("/")[1] for model in ahx_model_list]
@@ -141,10 +140,15 @@ 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")
+  gr.Markdown('''
+      # 🧑‍🚀 Advanced Concept Loader
 
-  # dropdown = gr.Dropdown(list(DROPDOWNS), label="choose style...")
-  # gr.Markdown("<u>styles</u>: check out examples of these at https://www.astronaut.horse/collaborations")
+      This tool allows you to run your own text prompts into fine-tuned artist concepts with individual parameter controls. 
+      Text prompts need to manually include artist concept / model tokens, see the examples below. 
+      The seed controls the static starting. 
+      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():
     prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
@@ -189,39 +193,83 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as dropdown_tab:
 # -----------------------------------------------------------------------------------------------
 
 
+#@title Dropdown Prompt Tab
 
-SELECT_LABEL = "Select concept"
-def assembleHTML(model):
-  return ''
-  
-def title_block(title, id):
-  return gr.Markdown(f"### [`{title}`](https://huggingface.co/{id})")
+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 default_guidance():
+  return 7.5
 
-def image_block(image_list, concept_type):
-  return gr.Gallery(
-          label=concept_type, value=image_list, elem_id="gallery"
-          ).style(grid=[2], height="auto")
+def default_steps():
+  return 30
 
-def checkbox_block():
-  checkbox = gr.Checkbox(label=SELECT_LABEL).style(container=False)
-  return checkbox
+def default_pixel():
+  return 768
 
-def infer(text):
+def random_seed():
+  return random.randint(0, 99999999999999) # <-- this is a random gradio limit, the seed range seems to actually be 0-18446744073709551615
+
+
+def simple_image_prompt(prompt, dropdown):
+  seed = random_seed()
+  guidance = 7.5
+  height = 768
+  width = 768
+  steps = 30
+
+  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)}"
+      )
+
+
+with gr.Blocks(css=".gradio-container {max-width: 650px}") as new_welcome:
+  gr.Markdown('''
+      # 🧑‍🚀 Astronaut Horse Concept Loader
+
+      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>
+  ''')
+  dropdown = gr.Dropdown(list(DROPDOWNS), label="choose style...")
+
+  # with gr.Row():
+  prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
+
+  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=simple_image_prompt, inputs=[prompt, dropdown], outputs=[output, output_text])
+
+# -----------------------------------------------------------------------------------------------
+
+
+def infer(text, dropdown):
   images_list = pipe(
-              [text],
-              num_inference_steps=50,
+              [f"{text} in the style of <{dropdown}>"],
+              num_inference_steps=30,
               guidance_scale=7.5
   )
   return images_list.images, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
-def infer_examples(text):
-  images_list = pipe(
-              [text],
-              num_inference_steps=50,
-              guidance_scale=7.5
-  )
-  return images_list.images
-  
+
 css = ""
 examples = []
 
@@ -246,31 +294,17 @@ with gr.Blocks(css=css) as demo:
   ''')
   with gr.Row():
         with gr.Column():
-          # 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():
+          text = gr.Textbox(
+              label="Enter your prompt", placeholder="Enter your prompt", show_label=False, max_lines=1, elem_id="prompt_input"
+          )
+          btn = gr.Button("generate image",elem_id="run_btn")
           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)
   checkbox_states = {}
-  inputs = [text]
+  inputs = [text, dropdown]
   btn.click(
         infer,
         inputs=inputs,
@@ -280,6 +314,5 @@ with gr.Blocks(css=css) as demo:
   # -----------------------------------------------------------------------------------------------
 
 
-
-tabbed_interface = gr.TabbedInterface([demo.queue(max_size=20), dropdown_tab], ["Welcome!", "Advanced Prompting"])
+tabbed_interface = gr.TabbedInterface([new_welcome, dropdown_tab], ["Welcome!", "Advanced Prompting"])
 tabbed_interface.launch()