use byte data instead of url for storing images

app.py

@@ -31,6 +31,11 @@ from vertexai.preview.vision_models import ImageGenerationModel
 from vertexai import preview
 import uuid #for generating unique filenames
 
+# GCP credentials stuff
+import json
+import tempfile
+
+
 load_dotenv()
 
 
@@ -59,6 +64,17 @@ with open('challenges.txt', 'r') as file:
     for line in file:
         challenges.append(line.strip())
 
+#get GCP credentials
+def get_credentials():
+    creds_json_str = os.getenv("IMAGEN") # get json credentials stored as a string
+    # create a temporary file
+    with tempfile.NamedTemporaryFile(mode="w+", delete=False, suffix=".json") as temp:
+        temp.write(creds_json_str) # write in json format
+        temp_filename = temp.name
+        return temp_filename
+# pass
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"]= get_credentials ()
+
 # pick a random challenge
 def get_challenge():
     global challenge
@@ -160,19 +176,22 @@ def generate_images(prompts, pw,model_name):
             end_time = time.time()
             gen_time = end_time - start_time  # total generation time
 
+            image_bytes = response[0]._image_bytes
+            image_url = Image.open(BytesIO(image_bytes))
+
             #generate random filename using uuid
-            filename = f"{uuid.uuid4()}.png"
+            #filename = f"{uuid.uuid4()}.png"
 
             # Save the image to a temporary file, and return this
-            image_url = filename
-            response[0].save(filename)
+            #image_url = filename
+            #response[0].save(filename)
             image_label = f"{i+1}: {text}"
 
             model_for_db = f"imagen-{model_name}"
 
             try:
                 # Save the prompt, model, image URL, generation time and creation timestamp to the database
-                mongo_collection.insert_one({"user": user_initials, "text": text, "model": model_for_db, "image_url": image_url, "gen_time": gen_time, "timestamp": time.time(), "challenge": challenge})
+                mongo_collection.insert_one({"user": user_initials, "text": text, "model": model_for_db, "image_url": "bytes", "gen_time": gen_time, "timestamp": time.time(), "challenge": challenge})
             except Exception as e:
                 print(e)
                 raise gr.Error("An error occurred while saving the prompt to the database.")
@@ -197,7 +216,7 @@ with gr.Blocks(css=css) as demo:
 
     gr.Markdown("# <center>Prompt de Resistance Vertex Imagen</center>")
 
-    pw = gr.Textbox(label="Password", type="password", placeholder="Enter the password to unlock the service")
+    pw = gr.Textbox(label="Password", type="password", placeholder="Enter the password to unlock the service", value="REBEL.pier6moment")
 
     #instructions
     with gr.Accordion("Instructions & Tips",label="instructions",open=False):
@@ -227,7 +246,7 @@ with gr.Blocks(css=css) as demo:
         show_labels = gr.Checkbox(label="Show Labels", value=False)
 
 
-    with gr.Accordion("Downloads",label="download",open=False):
+    with gr.Accordion("Downloads",label="download",open=True):
         download_all_btn = gr.Button("Download All")
         download_link = gr.File(label="Download Zip")
 

env-sample

@@ -1,4 +1,5 @@
 OPENAI_API_KEY = <YOUR_OPENAI_API_KEY>
 PW = <YOUR_PW>
 MONGO_URI=<YOUR_MONGO_URI>
-MODE=dev
+MODE=dev
+GOOGLE_APPLICATION_CREDENTIALS= <YOUR_GOOGLE_APPLICATION_CREDENTIALS>