Upload folder using huggingface_hub

.env

@@ -0,0 +1,2 @@
+REPLICATE_API_TOKEN=r8_DAzyOBdCwUdt0b26ZMPWLyvyHTh55uh2Lwb3c
+OPENAI_API_KEY=sk-proj-6lTXmIwTYmNo7uUpQwujT3BlbkFJDMVzyH5hzblFbgYLLMCP

api/index.py

@@ -46,7 +46,7 @@ def call_openai(pil_image):
                 {
                     "role": "user",
                     "content": [
-                        {"type": "text", "text": "You are a product designer. I've attached a moodboard here. In one sentence, what do all of these elements have in common? Answer from a design language perspective, if you were telling another designer to create something similar, including any repeating colors and materials and shapes and textures"},
+                        {"type": "text", "text": "You are a product designer. I've attached a moodboard here. In one sentence, what do all of these elements have in common? Answer from a design language perspective, if you were telling another designer to create something similar, including any repeating colors and materials and shapes and textures. This is for a single product, so respond as though you're applying them to a single object. Reply with a completion to the following (don't include these words please, just the rest): [A render of an object which] [your response]. Do NOT include 'A render of an object which' in your response."},
                         {
                             "type": "image_url",
                             "image_url": {
@@ -78,14 +78,12 @@ def image_classifier(moodboard, prompt):
         pil_image = Image.fromarray(moodboard.astype('uint8'))
 
         openai_response = call_openai(pil_image)
-        openai_response = openai_response.replace('moodboard', '')
-        openai_response = openai_response.replace('share', '')
-        openai_response = openai_response.replace('unified', '')
+
     else:
         raise gr.Error(f"Please upload a moodboard to control image generation style")
 
     input = {
-        "prompt": "high quality render of " + prompt + ", " + openai_response[12:],
+        "prompt": "high quality render of a " + prompt + " which " + openai_response + ", minimalist and simple mockup on a white background",
         "output_format": "jpg"
     }
     
@@ -118,16 +116,17 @@ def image_classifier(moodboard, prompt):
     except Exception as e:
         raise gr.Error(f"Second image download failed: {e}")
 
-    # Call Stable Diffusion API with the response from OpenAI
+    # Call SDXL API with the response from OpenAI
     input = {
         "width": 768,
         "height": 768,
-        "prompt": "high quality render of " + prompt + ", " + openai_response[12:],
-        "negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch, logo, buttons",
+        "prompt": "centered high quality render of a " + prompt + " which " + openai_response + ' centered on a plain white background',
+        "negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch, logo, buttons, markings, text, wires, complex, screws, nails, construction",
         "refine": "expert_ensemble_refiner",
         "apply_watermark": False,
         "num_inference_steps": 25,
-        "num_outputs": 2
+        "num_outputs": 2,
+        "guidance_scale": 8.5
     }
     
     output = replicate.run(