prompt_types

huggingface_inference_node.py

@@ -62,9 +62,10 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 "fantasy": fantasy_prompt
             }
 
-            # 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"Received prompt_type: '{prompt_type}'")  # Debug print
+
+            if prompt_type and prompt_type.strip() in prompt_types:
+                base_prompt = prompt_types[prompt_type.strip()]
                 print(f"Using {prompt_type} prompt")
             elif custom_base_prompt.strip():
                 base_prompt = custom_base_prompt

ui_components.py

@@ -170,9 +170,13 @@ def create_interface():
         generate_text_button.click(
             generate_text_with_llm,
             inputs=[output, happy_talk, compress, compression_level, prompt_type, custom_base_prompt],
-            outputs=text_output
+            outputs=text_output,
+            api_name="generate_text"  # Add this line
         )
 
+        # Add this line to disable caching for the generate_text_with_llm function
+        generate_text_with_llm.cache_examples = False
+
         def update_all_options(choice):
             updates = {}
             if choice == "Disabled":