Upload folder using huggingface_hub

index.py

@@ -13,9 +13,29 @@ from dotenv import load_dotenv, find_dotenv
 # Locate the .env file
 dotenv_path = find_dotenv()
 load_dotenv(dotenv_path)
-
 REPLICATE_API_TOKEN = os.getenv('REPLICATE_API_TOKEN')
 
+def generate_sd3_image(prompt):
+    prompt = "texture sample zoomed in of " + prompt  + ", perfect"
+
+    output = replicate.run(
+        "stability-ai/stable-diffusion-3",
+        input={
+            "cfg": 3.5,
+            "steps": 28,
+            "prompt": prompt,
+            "aspect_ratio": "1:1",
+            "output_format": "jpg",
+            "output_quality": 80,
+            "negative_prompt": "",
+            "prompt_strength": 0.85
+        }
+    )
+
+    response = requests.get(output[0])
+
+    return Image.open(io.BytesIO(response.content))
+
 def generate_pattern(image, prompt):
 
     if image is not None:
@@ -36,17 +56,16 @@ def generate_pattern(image, prompt):
             
             # Resize the image
             starter_image_pil = starter_image_pil.resize((new_width, new_height), Image.LANCZOS)
+    else:
+        starter_image_pil = generate_sd3_image(prompt)   
 
 
-        # Move the image horizontally and vertically by 50%
-        width, height = starter_image_pil.size
-        horizontal_shift = width // 2
-        vertical_shift = height // 2
-
-        transformed_image_pil = ImageChops.offset(starter_image_pil, horizontal_shift, vertical_shift)
-    else:
-        raise gr.Error(f"Please upload an image")        
+    # Move the image horizontally and vertically by 50%
+    width, height = starter_image_pil.size
+    horizontal_shift = width // 2
+    vertical_shift = height // 2
 
+    transformed_image_pil = ImageChops.offset(starter_image_pil, horizontal_shift, vertical_shift)
 
     # Create a new image with black background and white cross
     cross_image_pil = Image.new('RGB', (width, height), 'black')