Update app.py

app.py

@@ -80,25 +80,25 @@ def generate_caption_and_image(image, f, p, d):
         out = model.generate(**inputs)
         caption1 = processor.decode(out[0], skip_special_tokens=True)
     
-        prompt = f"Generate a clothing item using the following details: 1. {caption1} 2. {caption2} 3. Fabric: {f} 4. Pattern: {p} 5. Design Style: {d}. "
-    
-      
-        prompt +="The image should have a clean, minimalistic grey or white background, with realistic lighting and fine details, ensuring a sophisticated and polished appearance"
+        prompt = (f"Create a visually stunning clothing item inspired by: 1. Primary Context: {caption1}, describing the mood and thematic elements of the image. 2. Secondary Insights: {caption2}, providing complementary attributes and textures. 3. Fabric: '{f}', highlighting its qualities. 4. Pattern: '{p}', enhancing visual harmony. 5. Design Style: '{d}', for a refined finish. Use a clean grey/white background with realistic lighting and intricate details for a polished presentation.")
+
+
     
     
       
     
         # Generate image based on the caption
         generated_image = pipe(prompt).images[0]
+        generated_image1 =pipe(prompt).images[0]
     
-        return generated_image
+        return generated_image, generated_image1
 
 # Gradio UI
 iface = gr.Interface(
     fn=generate_caption_and_image,
     inputs=[gr.Image(type="pil", label="Upload Image"), gr.Radio(fabrics, label="Select Fabric"), gr.Radio(patterns, label="Select Pattern"), gr.Radio(textile_designs, label="Select Textile Design")],
             
-    outputs=[gr.Image(label="Generated Design")],
+    outputs=[gr.Image(label="Generated Design 1"), gr.Image(label="Generated Design 2")],
     live=True
 )
 iface.launch(share=True)