Update lexica.py

lexica.py

@@ -22,6 +22,7 @@ def lexica(design,n):
 
   image_urls = []
   image_prompts = []
+  image_gallery=[]
 
   for key, value in data.items():
     for i in range(n):
@@ -32,6 +33,7 @@ def lexica(design,n):
 
  
       image_prompts.append(value[i]['prompt'])
+      image_gallery.append(value[i]['gallery'])
  
   images = []
 
@@ -47,10 +49,11 @@ def lexica(design,n):
     images.append(image)
 
  
-  df = pd.DataFrame(image_prompts, columns=["Lexica Prompt"], index=range(1, len(image_prompts)+1))
+#  df = pd.DataFrame(image_prompts, columns=["Lexica Prompt"], index=range(1, len(image_prompts)+1))
 
  
-  df.index.name = "Sr. No."
+#  df.index.name = "Sr. No."
+  df = pd.DataFrame({'image_prompts': image_prompts, 'image_gallery': image_gallery})
 
  
   for image in images: