Update app.py

app.py

@@ -1,25 +1,8 @@
-# !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
 
-#Big Web Labs API Key
-openai.api_key = "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"
-
-# openai.api_key = os.environ.get("OPENAI_API_KEY")
+openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # Dalle API
 def open_ai_txt2img(prompt):
@@ -43,42 +26,4 @@ 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)
+demo.launch()