reworked Color Guided

app.py

@@ -28,7 +28,7 @@ with gr.Blocks() as demo:
     gr.HTML("""
     <style>
         #component-0 {
-            max-width: 800px;
+            max-width: 850px;
             margin: 0 auto; 
         }
         .center-markdown {
@@ -147,15 +147,17 @@ with gr.Blocks() as demo:
         with gr.Column():
             gr.Markdown("""
             <div style="text-align: center;">
-                Generate an image using a color palette.  If you must include an image and text prompt, the subject and style will be used as a reference. 
-                The colors of the image will also be incorporated, along with the colors from the colors list. A color list is always required but one has been provided.
+                Generate an image using a color palette.  You must include a text prompt if you choose to include an image, the subject and style will be used as a reference. 
+                The colors of the image will also be incorporated, along with the colors from the colors list. A color list is also required if you choose not to include one,
+                a generic list has been provided.
                 </div>
             """)
-            reference_image = gr.Image(type='pil', label="Reference Image")     
+              
             colors = gr.Textbox(label="Colors", placeholder="Enter up to 10 colors as hex values, e.g., #00FF00,#FCF2AB", max_lines=1)
-            
             prompt = gr.Textbox(label="Text", placeholder="Enter a text prompt (1-1024 characters)", max_lines=4)
             gr.Button("Generate Prompt").click(generate_nova_prompt, outputs=prompt)
+            with gr.Accordion("Optional Reference Image", open=False):
+                reference_image = gr.Image(type='pil', label="Reference Image")   
             error_box = gr.Markdown(visible=False, label="Error", elem_classes="center-markdown")
             output = gr.Image()
             with gr.Accordion("Advanced Options", open=False):

functions.py

@@ -166,19 +166,22 @@ def image_conditioning(condition_image, text, negative_text=None, control_mode="
     return check_return(result)
 
 def color_guided_content(text=None, reference_image=None, negative_text=None, colors=None, height=1024, width=1024, quality="standard", cfg_scale=8.0, seed=0):
-    # Encode the reference image if provided
-    reference_image_encoded = process_images(primary=reference_image)
-    for value in reference_image_encoded.values():
-        if len(value) < 200:
-            return None, gr.update(visible=True, value=value)
-        
+    reference_image_str = None
+
+    if reference_image is not None and not isinstance(reference_image, type(None)):
+       
+        reference_image_encoded = process_images(primary=reference_image)
+        for value in reference_image_encoded.values():
+            if len(value) < 200:
+                return None, gr.update(visible=True, value=value)
+            reference_image_str = reference_image_encoded.get('image')
     if not colors:
         colors = "#FF5733,#33FF57,#3357FF,#FF33A1,#33FFF5,#FF8C33,#8C33FF,#33FF8C,#FF3333,#33A1FF"
     
     color_guided_generation_params = {
         "text": text,
         "colors": colors.split(','),
-        "referenceImage": reference_image_encoded.get('image'),
+        **({"referenceImage": reference_image_str} if reference_image_str is not None else {}),    
         **({"negativeText": negative_text} if negative_text not in [None, ""] else {})
     }
 

generate.py

@@ -173,7 +173,7 @@ def check_rate_limit(body):
             raise ImageError(rate_limit_message.format('Premium'))
         rate_data['premium'].append(current_time)
     else:  # standard
-        if len(rate_data['standard']) >= 6:
+        if len(rate_data['standard']) >= 12:
             raise ImageError(rate_limit_message.format('Standard'))
         rate_data['standard'].append(current_time)