Update app.py

app.py

@@ -33,6 +33,17 @@ def generate_caption_and_image(image):
     img = image.convert("RGB")
     # reader = easyocr.Reader(['en'])  
     # result = reader.readtext(img)
+    import random
+
+    # Define lists for the three variables
+    fabrics = ['cotton', 'silk', 'denim', 'linen', 'polyester', 'wool', 'velvet']
+    patterns = ['striped', 'floral', 'geometric', 'abstract', 'solid', 'polka dots']
+    textile_designs = ['woven texture', 'embroidery', 'printed fabric', 'hand-dyed', 'quilting']
+    
+    # Randomly select one from each category
+    selected_fabric = random.choice(fabrics)
+    selected_pattern = random.choice(patterns)
+    selected_textile_design = random.choice(textile_designs)
     
     
 
@@ -44,7 +55,8 @@ def generate_caption_and_image(image):
     inputs = {key: val.to(device) for key, val in inputs.items()}
     out = model.generate(**inputs)
     caption = processor.decode(out[0], skip_special_tokens=True)
-    prompt = f"Create a highly realistic design of a clothing item based on the following description: 'The design should he realted to following {caption}. The clothing should look realistic, modern, and stylish. Use high-quality fabric textures and realistic lighting to give the design a lifelike appearance. The colors, patterns, and materials should reflect the essence of the caption and extracted text.'"
+    prompt = f"Create a highly realistic design of a clothing item based on the following description: The design should reflect {caption}, featuring a highly realistic and modern piece of clothing that incorporates stylish and high-quality textures, exuding sophistication with realistic fabric lighting and fine details, subtly hinting at {selected_fabric}, with a {selected_pattern} motif and a {selected_textile_design} style.'''
+"
 
 
     # Generate image based on the caption