trial

app.py

@@ -89,7 +89,14 @@ def create_interface():
             try:
                 # Step 1: Generate Prompt
                 dynamic_seed = random.randint(0, 1000000)
-                prompt = llm_node.generate_prompt(dynamic_seed, prompt_type, custom_input)
+                if custom_input and custom_input.strip():
+                    prompt = llm_node.generate_prompt(dynamic_seed, prompt_type, custom_input)
+                    print(f"Using Custom Input Prompt.")
+                else:
+                    # Inform the system to create a random prompt based on the selected prompt_type
+                    prompt = llm_node.generate_prompt(dynamic_seed, prompt_type, f"Create a random prompt based on the '{prompt_type}' type.")
+                    print(f"No Custom Input Prompt provided. Generated prompt based on prompt_type: {prompt_type}")
+
                 print(f"Generated Prompt: {prompt}")
 
                 # Step 2: Generate Text with LLM

llm_inference.py

@@ -28,7 +28,7 @@ class LLMInferenceNode:
         if custom_input and custom_input.strip():
             prompt = custom_input
         else:
-            prompt = f"Generate a {prompt_type.lower()} prompt."
+            prompt = f"Create a random prompt based on the '{prompt_type}' type."
 
         # Additional logic can be added here if needed
         print(f"Generated prompt: {prompt}")  # Debug statement
@@ -107,7 +107,11 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
 
             # Construct messages for the LLM
             system_message = "You are a helpful assistant. Try your best to give the best response possible to the user."
-            user_message = f"{base_prompt}\nDescription: {input_text}"
+
+            if input_text.startswith("Create a random prompt based on"):
+                user_message = input_text
+            else:
+                user_message = f"{base_prompt}\nDescription: {input_text}"
 
             # Select the appropriate provider
             if provider == "Hugging Face":