Attempt to remove inputs from artbot image function

app.py

@@ -181,6 +181,12 @@ for model in model_tags:
   if model != "ahx-model-1" and model != "ahx-model-2":
     DROPDOWNS[model] = f" in the style of <{model}>"
 
+TOKENS = []
+
+for model in model_tags:
+  if model != "ahx-model-1" and model != "ahx-model-2":
+    TOKENS.append(f"<{model}>")
+
 # def image_prompt(prompt, dropdown, guidance, steps, seed, height, width, negative_prompt=""):
 def image_prompt(prompt, guidance, steps, seed, height, width, negative_prompt=""):
   # prompt = prompt + DROPDOWNS[dropdown]
@@ -215,6 +221,62 @@ def image_prompt(prompt, guidance, steps, seed, height, width, negative_prompt="
       f"Prompt violates Hugging Face's Terms of Service"
     )
 
+
+# New ArtBot image function -------------------------------------------------
+# def image_prompt(prompt, dropdown, guidance, steps, seed, height, width, negative_prompt=""):
+# def artbot_image(prompt, guidance, steps, seed, height, width, negative_prompt=""):
+def artbot_image():
+  guidance = 7.5
+  steps = 30
+  height = 768
+  width = 768
+  negative_prompt = ""
+
+  all_models = [token for token in TOKENS if 'ahx-' in token]
+  model_1 = random.choice(all_models)
+  model_2 = random.choice(all_models)
+
+  prompt = f"{model_1} {model_2}"
+
+  seed = random_seed()
+    
+    
+  square_pixels = height * width
+  if square_pixels > 640000:
+      height = 640000 // width
+  generator = torch.Generator(device="cuda").manual_seed(int(seed))
+
+  height=int((height // 8) * 8)
+  width=int((width // 8) * 8)
+
+  # image_count += 1
+  curr_time = datetime.datetime.now()
+
+  is_clean = check_prompt(prompt)
+
+  print("----- advanced tab prompt ------------------------------")
+  print(f"prompt: {prompt}, size: {width}px x {height}px, guidance: {guidance}, steps: {steps}, seed: {int(seed)}")
+  # print(f"image_count: {image_count}, datetime: `{e}`")
+  print(f"datetime: `{curr_time}`")
+  print(f"is_prompt_clean: {is_clean}")
+  print("-------------------------------------------------------")
+
+  if is_clean:
+    return (
+      pipe(prompt=prompt, guidance_scale=guidance, num_inference_steps=steps, generator=generator, height=height, width=width, negative_prompt=negative_prompt).images[0], 
+      f"prompt: '{prompt}', seed = {int(seed)},\nheight: {height}px, width: {width}px,\nguidance: {guidance}, steps: {steps}, negative prompt: {negative_prompt}"
+    )
+  else:
+    return (
+      pipe(prompt="", guidance_scale=0, num_inference_steps=1, generator=generator, height=32, width=32).images[0], 
+      f"Prompt violates Hugging Face's Terms of Service"
+    )
+
+
+
+
+      
+
 def default_guidance():
   return 7.5
 
@@ -436,19 +498,12 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as artbot_1:
   #   size_dropdown = gr.Dropdown(['square', 'portrait', 'landscape'], label="choose size...")
   # prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
 
-  all_models = [dropdown for dropdown in list(DROPDOWNS) if 'ahx-' in dropdown]
-  model_1 = random.choice(all_models)
-  model_2 = random.choice(all_models)
-
-  prompt = f"{model_1} {model_2}"
-
-  seed = random_seed()
 
   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, size_dropdown], outputs=[output, output_text])
-  go_button.click(fn=image_prompt, inputs=[prompt, 7, 30, seed, 768, 768, ""], outputs=[output, output_text])
+  go_button.click(fn=artbot_image, inputs=[], outputs=[output, output_text])