Update app.py

app.py

@@ -57,15 +57,7 @@ def generate_caption_and_image(image):
 
         
 
-    # 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)
+ 
     text = "a picture of "
     inputs = processor(img, text, return_tensors="pt").to(device)
     
@@ -97,9 +89,21 @@ def generate_caption_and_image(image):
 # Gradio UI
 iface = gr.Interface(
     fn=generate_caption_and_image,
-    inputs=gr.Image(type="pil", label="Upload 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.Textbox(label="Generated Caption"), gr.Image(label="Generated Design")],
     live=True
 )
 iface.launch(share=True)
 
+
+
+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']
+
+
+