Update app.py

app.py

@@ -11,17 +11,26 @@ from openai import OpenAI
 #     im1.save(r'new_name.png')
 #     return im1
 def create_var(image_path, open_api_key):
+    url_list = []
+    img_list = []
     client = OpenAI(api_key = open_api_key)
     # image2 = con_image("white_tile.jpg")
     response = client.images.create_variation(
       image=open(image_path, "rb"),
-      n=1,
+      n=4,
       size="1024x1024"
     )
-    
-    image_url = response.data[0].url
-    # print(response.data[0])
-    return(image_url)
+    for i in range(len(response.data)):
+        url1 = response.data[i].url
+        url_list.append(url1)
+
+    for j in range(len(url_list)) :
+        response = requests.get(url_list[j])
+        with open(f"image{j}.jpg", "wb") as f:
+            f.write(response.content)
+            img_list.append(f"image{j}.jpg")
+    return img_list
+
 
 block = gr.Blocks(theme=gr.themes.Glass(primary_hue="slate"))
 with block:
@@ -32,6 +41,6 @@ with block:
             api_input = gr.Textbox(label= "Enter Api-key")
             upload_button = gr.Button(value="Upload & Start Chat", interactive=True, variant="primary")
         with gr.Column():
-            text = gr.Textbox(label="Output")
-    upload_button.click(create_var, inputs=[img_input, api_input], outputs=[text])
+            img_gallery = gr.Gallery(label="Generated images", show_label=True, elem_id="gallery", object_fit="contain", height="auto",allow_preview=True)
+    upload_button.click(create_var, inputs=[img_input, api_input], outputs=[img_gallery])
 block.launch()