Update app.py

app.py

@@ -1,8 +1,25 @@
+!pip install gradio
+!pip install openai
+
+# import gradio as gr
+# import openai
+# import base64
+# from PIL import Image
+# from IPython.display import display, Image
+# from PIL import Image as PILImage
+# from io import BytesIO
+# import numpy as np
+# import io
+# import cv2
+
 import gradio as gr
 import openai
 import os
 
-openai.api_key = os.environ.get("OPENAI_API_KEY")
+#Big Web Labs API Key
+openai.api_key = "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"
+
+# openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # Dalle API
 def open_ai_txt2img(prompt):
@@ -27,3 +44,41 @@ with gr.Blocks() as demo:
     submit.click(generator, inputs=[prompt], outputs=[image1, image2, image3], api_name="mmd")
 
 demo.launch()
+
+# def image_variation(input):
+#   # Resize input image to 1024x1024
+#   input = cv2.resize(input, (1024, 1024))
+  
+#   # Convert input to PIL Image
+#   input = np.uint8(input * 255)
+#   pil_image = PILImage.fromarray(input)
+  
+#   # Convert PIL Image to byte stream
+#   byte_stream = io.BytesIO()
+#   pil_image.save(byte_stream, format='PNG')
+#   byte_array = byte_stream.getvalue()
+  
+#   # Call OpenAI API to create variation
+#   response = openai.Image.create_variation(image=byte_array, n=6, size="1024x1024")
+  
+#   # Return URL of first variation
+#   urls = [variation['url'] for variation in response['data']]
+#   return urls
+
+# with gr.Blocks() as demo:
+#     with gr.Tab("Upload"):
+#         with gr.Row():
+#             image_input = gr.Image()
+#         with gr.Row():  
+#             image_submit = gr.Button("Image Variation")
+#         with gr.Row():
+#             image_output = gr.Image()
+#             image_output2 = gr.Image()
+#             image_output3 = gr.Image()
+#         with gr.Row():
+#             image_output4 = gr.Image()
+#             image_output5 = gr.Image()
+#             image_output6 = gr.Image()
+#     image_submit.click(image_variation, inputs=[image_input], outputs=[image_output, image_output2, image_output3, image_output4, image_output5, image_output6])
+
+# demo.launch(debug=True)