Upload folder using huggingface_hub

index.py

@@ -24,10 +24,12 @@ REPLICATE_API_TOKEN = os.getenv('REPLICATE_API_TOKEN')
 client = OpenAI()
 
 def main(img, strength, prompt):
+    print(prompt)
+
+    
     mask = img['layers'][0]
     if is_transparent(mask) == True:
         mask_img = None
-        mask_base_64 = None
     else:
         mask_img = create_mask_image(mask)
         
@@ -53,12 +55,20 @@ def main(img, strength, prompt):
         base_image = base_image.resize((new_width, new_height), Image.LANCZOS)
         if mask_img is not None:
             mask_img = mask_img.resize((new_width, new_height), Image.LANCZOS)
+    
+    if mask_img is not None:
+        mask_base_64 = img_to_base64(mask_img)
+    else:
+        mask_base_64 = None
 
     img_base_64 = img_to_base64(base_image)
-    mask_base_64 = img_to_base64(mask_img)
 
-    #prompt = call_openai(img_base_64)
+    
+    # Todo -- send just the masked out part to gpt
     prompt = prompt
+    if prompt == "":
+        prompt = call_openai(img_base_64)
+        
     #prompt = "The image shows a person wearing sleek, over-ear headphones with a matte finish and a cool, light beige color (Pantone 7527 C), captured under soft, diffused natural lighting, emphasizing the smooth and minimalist design of the headphones."
 
     output_urls = generate_image(prompt, img_base_64, mask_base_64, prompt_strength)
@@ -180,7 +190,7 @@ black_brush = gr.Brush(colors=["#000000"], default_color="#000000", color_mode="
 # Using the ImageEditor component to enable drawing on the image with limited colors
 demo = gr.Interface(
     fn=main,
-    inputs=[gr.ImageEditor(brush=black_brush), gr.Slider(0, 1, step=0.025, value=0.5, label="Image Strength"), gr.Textbox(label="Describe the object in as much detail as possible")],
+    inputs=[gr.ImageEditor(brush=black_brush), gr.Slider(0, 1, step=0.025, value=0.5, label="Image Strength"), gr.Textbox(label="Describe the object in as much detail as possible (include colors, materials, etc)")],
     #outputs=[gr.Image(type="pil"), gr.Image(type="pil"), gr.Image(type="pil"), gr.Image(type="pil")]
     outputs=["image", "image", "image", "image"]
 )