Upload spaghetti_ai_script.py

spaghetti_ai_script.py

@@ -997,7 +997,7 @@ if enable_longer_prompts == 1:
 
     ending_html += """
 
-Longer prompts for Stable Diffusion 1.5 and PhotoReal are available using the method <a href=\"https://github.com/huggingface/diffusers/issues/2136#issuecomment-1514969011\" target=\"_blank\">here</a>. Another method will likely eventually be added for SDXL models."""
+Longer prompts are not available for most models. When available, it is done using the method <a href=\"https://github.com/huggingface/diffusers/issues/2136#issuecomment-1514969011\" target=\"_blank\">here</a>. Another method will likely eventually be added for SDXL models in this app."""
 
 ending_html += """
 
@@ -1617,6 +1617,9 @@ stored_model_configuration_names_object = {}
 
 for this_base_model in base_model_array:
 
+    if this_base_model not in base_model_object_of_model_configuration_arrays:
+        continue
+
     default_base_model_choices_array.append(
         base_model_names_object[this_base_model]
     )
@@ -1871,6 +1874,8 @@ def model_configuration_valid(
         base_model_name_value_str = str(base_model_name_value).lower()
         model_configuration_name_value_str = str(model_configuration_name_value).lower()
         for this_base_model in base_model_array:
+            if this_base_model not in base_model_object_of_model_configuration_arrays:
+                continue
             for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:
                 if (
                     (base_model_name_value_str == this_base_model) and
@@ -5919,6 +5924,9 @@ def get_query_params(
 
     for this_base_model in base_model_array:
 
+        if this_base_model not in base_model_object_of_model_configuration_arrays:
+            continue
+
         if base_model_name_value == this_base_model:
 
             model_configuration_name_value = base_model_model_configuration_defaults_object[this_base_model]
@@ -6381,6 +6389,9 @@ def set_base_model_and_model_configuration_from_query_params(
 
         model_configuration_name_default_value_for_this_base_model = base_model_model_configuration_defaults_object[this_base_model]
 
+        if this_base_model not in base_model_object_of_model_configuration_arrays:
+            continue
+
         for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:
 
             if (
@@ -6772,8 +6783,36 @@ with gr.Blocks(
                     visible = default_allow_longer_prompts_row_visibility
                 ):
 
+                    model_types_for_longer_prompts_html = ""
+
+                    sd15_available = 0
+                    photoreal_available = 0
+
+                    if "sd_1_5_runwayml" in base_model_object_of_model_configuration_arrays:
+
+                        sd15_available = 1
+
+                        model_types_for_longer_prompts_html += base_model_names_object["sd_1_5_runwayml"]
+
+                    if "photoreal" in base_model_object_of_model_configuration_arrays:
+
+                        photoreal_available = 1
+
+                        if sd15_available == 1:
+
+                            model_types_for_longer_prompts_html += " and "
+
+                        model_types_for_longer_prompts_html += base_model_names_object["photoreal"]
+
+                    if (
+                        (sd15_available == 1) and
+                        (photoreal_available == 1)
+                    ):
+
+                        model_types_for_longer_prompts_html += " models"
+
                     allow_longer_prompts_for_sd_1_5_based_models_field = gr.Checkbox(
-                        label = "Allow longer prompts for Stable Diffusion 1.5 and PhotoReal models when not using the refiner or upscaler",
+                        label = "Allow longer prompts for " + model_types_for_longer_prompts_html + " models when not using the refiner or upscaler",
                         value = default_allow_longer_prompts_for_sd_1_5_based_models_is_selected,
                         interactive = True,
                         container = True,
@@ -6919,6 +6958,10 @@ with gr.Blocks(
 
                     this_model_configuration_choices_array = []
 
+                    if this_base_model not in base_model_object_of_model_configuration_arrays:
+
+                        continue
+
                     for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:
 
                         this_model_configuration_choices_array.append(