Update app.py

app.py

@@ -92,21 +92,31 @@ def diffuse(text: str):
         # add image to Cloud Storage
         im = out.images[0]
         im.save(f'{_id}.png', format='png')
+        added_gcp = False
         # push to storage
         try:
             blob = bucket.blob(f'images/{_id}.png')
             blob.upload_from_filename(f'{_id}.png')
+            added_gcp = True
+            # add embedding and metadata to Pinecone
+            embeds = encode_text(text)
+            meta = {
+                'prompt': text,
+                'image_url': f'images/{_id}.png'
+            }
+            try:
+                index.upsert([(_id, embeds, meta)])
+            except Exception as e:
+                try:
+                    pinecone.init(api_key=PINECONE_KEY, environment='us-west1-gcp')
+                    index2 = pinecone.Index(index_id)
+                    index2.upsert([(_id, embeds, meta)])
+                except Exception as e:
+                    print(f"PINECONE_ERROR: {e}")
         except Exception as e:
-            print("New image not uploaded due to error...")
+            print(f"ERROR: New image not uploaded due to error with {'Pinecone' if added_gcp else 'Cloud Storage'}")
         # delete local file
         os.remove(f'{_id}.png')
-        # add embedding and metadata to Pinecone
-        embeds = encode_text(text)
-        meta = {
-            'prompt': text,
-            'image_url': f'images/{_id}.png'
-        }
-        index.upsert([(_id, embeds, meta)])
         return out.images[0]
 
 def get_image(url: str):