Push_Code

app.py

@@ -7,69 +7,80 @@ import tempfile
 import concurrent.futures
 import os
 
-api_key = os.environ.get("API_KEY")  
-univin_model = os.environ.get("univin")  
+api_key = os.environ.get("API_KEY")
+univin_model = os.environ.get("univin")
 
 client = genai.Client(api_key=api_key)
 
 PROMPT_VARIATIONS = [
     # 1. Professional White Background (Universal)
-    "Generate a crisp, professional studio-quality image of the original product placed precisely centered against a pure, seamless white background. Ensure the lighting is neutral, diffused, and evenly distributed to clearly accentuate all original details, including exact shape, precise color reproduction, authentic texture, printing clarity, and visibility of text and logos exactly as provided in the input image. Maintain the highest resolution for immediate use on a professional e-commerce website.",
+    "A high-resolution professional image of the product perfectly centered on a pure white seamless background, preserving exact shape, color, texture, printed text, and logo precisely as in the input image. Ideal for professional e-commerce.",
 
     # 2. Luxurious Spa Setting (Beauty, Skincare, Haircare)
-    "Create an elegant, luxurious scene featuring the original beauty, skincare, or haircare product placed gently atop a polished marble surface in a high-end spa environment. Surround with subtle, natural spa accessories (e.g., stones, plants, or towels) slightly blurred to keep the product in sharp focus. Employ soft, natural lighting that highlights the product's original shape, authentic color, rich textures, and precise logos and printed text without alteration. Convey a serene, premium aesthetic ideal for upscale beauty websites.",
+    "A luxurious spa-themed image showcasing the original beauty or skincare product elegantly positioned on polished marble surrounded by spa accessories (stones, towels, plants). Preserve all original product details exactly as input.",
 
-    # 3. Professional Model Holding Product (Skincare, Beauty)
-    "Produce a sophisticated, professional image showing a confident woman model gently holding the original skincare or beauty product in her hands. Frame the shot to focus primarily on the product, clearly displaying the model’s well-groomed hands and subtle styling without overshadowing the product. Ensure every original aspect of the product—shape, exact color, texture, and printed text or logos—is strictly preserved as in the original image. Opt for soft, flattering studio lighting, evoking elegance and authenticity suitable for premium skincare or beauty product pages.",
+    # 3. Professional Model Holding Product (Beauty/Skincare)
+    "Professional image of a sophisticated woman model holding the original beauty or skincare product gently and elegantly, focusing sharply on product details while maintaining precise original shape, color, texture, text, and logos.",
 
-    # 4. Sleek, Modern Tech Scene (Personal-use Electronics)
-    "Render an ultra-modern, technology-inspired scene showcasing the original personal-use electronic product on a sleek, reflective black or dark-gray surface. Include subtle ambient backlighting or minimal neon accents, creating a sophisticated, high-tech atmosphere. Precisely maintain the product’s original shape, exact color, texture details, and clearly readable text and logos exactly as in the original image. The aesthetic should evoke precision, innovation, and premium quality, perfect for professional electronics presentations online.",
+    # 4. Modern Tech Scene (Personal Electronics)
+    "A sleek, modern technology-oriented image featuring the electronic product placed on a reflective black or dark-gray surface with subtle neon lighting, strictly maintaining original shape, color, texture, printed text, and logos.",
 
-    # 5. Flat-lay with minimal styling
-    "Craft a warm, inviting lifestyle image featuring the original candle or soft toy professionally held by a model in a natural yet sophisticated home environment. If Candle: A woman's hands gently holding the lit candle, illuminating softly textured cozy fabrics or home décor elements around her. If Soft Toy: A smiling child warmly hugging or gently holding the soft toy in comfortable surroundings such as a bedroom or living space, subtly blurred in the background. Ensure meticulous preservation of the product’s original shape, accurate color, texture fidelity, and intact printed text and logos without any modification. The image should communicate warmth, authenticity, and premium comfort suited for a lifestyle-oriented web presentation.",
+    # 5. Lifestyle with Model (Candles/Soft Toys)
+    "Warm lifestyle image featuring either a woman professionally holding a lit candle or a child warmly hugging a soft toy. Ensure exact preservation of the original product's shape, color, texture, printed text, and logos as provided.",
 
-    # 6. High-contrast studio art shot
-    "Develop a vibrant, refreshing composition featuring the original haircare or skincare product prominently placed among fresh botanical elements (such as leaves, flowers, or gentle water droplets), set on a reflective or softly textured neutral surface. Incorporate bright, clean lighting to accentuate freshness and purity. Strictly retain the product’s original details—shape, exact colors, textures, clearly legible printed text, and logos exactly as presented. The final image should reflect a natural, premium aesthetic appealing directly to consumers seeking high-quality, nature-inspired personal care products."
+    # 6. Natural Refreshing Composition (Haircare & Skincare)
+    "Refreshing natural composition featuring the original skincare or haircare product among botanical elements and water droplets. Precisely preserve original product shape, color, texture, printed text, and logos from the input image."
 ]
 
-
 def process_variation(variation, input_image, product_name):
-    text_input = (
-        f"Hi, this is a picture of a product. The name of the product is {product_name}.",
-        variation
-    )
+    prompt_text = f"Product Name: {product_name}\n{variation}"
+
+    request_content = [
+        types.Part(text=prompt_text),
+        types.Part(inline_data=types.Blob(mime_type="image/png", data=input_image.getvalue()))
+    ]
+
     response = client.models.generate_content(
-        model=univin_model,  
-        contents=[text_input, input_image],
-        config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
+        model=univin_model,
+        contents=request_content,
+        config=types.GenerateContentConfig(response_modalities=['Image'])
     )
-    for part in response.candidates[0].content.parts:
-        if part.inline_data is not None:
-            generated_img = Image.open(BytesIO(part.inline_data.data))
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
-                generated_img.save(tmp_file, format="PNG")
-                return tmp_file.name
+
+    if response.candidates:
+        for part in response.candidates[0].content.parts:
+            if part.inline_data:
+                generated_img = Image.open(BytesIO(part.inline_data.data))
+                with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+                    generated_img.save(tmp_file, format="PNG")
+                    return tmp_file.name
     return None
 
-def generate_images(input_image, product_name):
-    
+def generate_images(input_pil_image, product_name):
+    input_buffer = BytesIO()
+    input_pil_image.save(input_buffer, format="PNG")
+
     with concurrent.futures.ThreadPoolExecutor() as executor:
         futures = [
-            executor.submit(process_variation, variation, input_image, product_name)
+            executor.submit(process_variation, variation, input_buffer, product_name)
             for variation in PROMPT_VARIATIONS
         ]
         output_files = [future.result() for future in futures if future.result() is not None]
+
     return output_files
 
 with gr.Blocks() as demo:
     gr.Markdown("# Uni-Imaginator")
     with gr.Row():
-        input_image = gr.Image(type="pil", label="Upload Image")
+        input_image = gr.Image(type="pil", label="Upload Product Image")
         product_name = gr.Textbox(label="Product Name", placeholder="Enter the product name")
+
     generate_button = gr.Button("Generate Images")
-    
-    gallery = gr.Gallery(label="Generated Images", elem_id="gallery", columns=4, height="auto", object_fit="contain", interactive=True)
-    
-    generate_button.click(fn=generate_images, inputs=[input_image, product_name], outputs=gallery)
-    
-demo.launch()
+    gallery = gr.Gallery(label="Generated Images", columns=3, height="auto", interactive=True)
+
+    generate_button.click(
+        fn=generate_images,
+        inputs=[input_image, product_name],
+        outputs=gallery
+    )
+
+demo.launch(debug=True)