prompt_types

huggingface_inference_node.py

@@ -62,14 +62,16 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 "fantasy": fantasy_prompt
             }
 
-            if prompt_type in prompt_types:
+            # Update this part to handle the prompt_type correctly
+            if prompt_type and prompt_type in prompt_types:
                 base_prompt = prompt_types[prompt_type]
                 print(f"Using {prompt_type} prompt")
             elif custom_base_prompt.strip():
                 base_prompt = custom_base_prompt
+                print("Using custom base prompt")
             else:
                 base_prompt = default_happy_prompt
-                print(f"Warning: Unknown prompt type '{prompt_type}'. Using default happy prompt.")
+                print(f"Warning: Unknown or empty prompt type '{prompt_type}'. Using default happy prompt.")
 
             if compress and not poster:
                 compression_chars = {

ui_components.py

@@ -163,8 +163,12 @@ def create_interface():
             outputs=[output]
         )
 
+        def generate_text_with_llm(output, happy_talk, compress, compression_level, prompt_type, custom_base_prompt):
+            print(f"Prompt type selected: {prompt_type}")  # Debug print
+            return huggingface_node.generate(output, happy_talk, compress, compression_level, prompt_type, custom_base_prompt)
+
         generate_text_button.click(
-            huggingface_node.generate,
+            generate_text_with_llm,
             inputs=[output, happy_talk, compress, compression_level, prompt_type, custom_base_prompt],
             outputs=text_output
         )