random

llm_inference.py

@@ -99,7 +99,7 @@ If there are buildings or other structures, describe their architecture, conditi
 Include sensory details beyond just visual elements - mention sounds, smells, and the overall atmosphere or mood of the scene. 
 Describe any notable features like bodies of water, geological formations, or sky phenomena. 
 Consider the perspective from which the scene is viewed and how this affects the composition. 
-Your description should transport the reader to this location, allowing them to vividly imagine the scene without any living subjects present.
+Your description should transport the reader to this location, allowing them to vividly imagine the scene without any living subjects present. 
  Your output is only the caption itself, no comments or extra formatting. The caption is in a single long paragraph."""
 
             landscape_prompt = """Create an immersive and detailed description of a landscape, 
@@ -145,6 +145,10 @@ Your output is only the caption itself, no comments or extra formatting. The cap
 
             # Determine the base prompt
             print(f"Received prompt_type: '{prompt_type}'")  # Debug print
+            if prompt_type == "Random":
+                prompt_type = random.choice(list(prompt_types.keys()))
+                print(f"Randomly selected prompt type: {prompt_type}")
+            
             if prompt_type and prompt_type.strip() and prompt_type in prompt_types:
                 base_prompt = prompt_types[prompt_type]
                 print(f"Using {prompt_type} prompt")
@@ -153,9 +157,7 @@ Your output is only the caption itself, no comments or extra formatting. The cap
                 print("Using custom base prompt")
             else:
                 base_prompt = default_long_prompt
-                print(
-                    f"Warning: Unknown or empty prompt type '{prompt_type}'. Using default long prompt."
-                )
+                print(f"Warning: Unknown or empty prompt type '{prompt_type}'. Using default long prompt.")
 
             # Handle compression if applicable
             if compress and not poster: