Update app.py

app.py

@@ -60,16 +60,22 @@ def encode_text(text: str):
     return text_embeds
 
 def prompt_query(text: str):
+    print(f"Running prompt_query('{text}')")
     embeds = encode_text(text)
     try:
+        print("Try query pinecone")
         xc = index.query(embeds, top_k=30, include_metadata=True)
+        print("query successful")
     except Exception as e:
         print(f"Error during query: {e}")
         # reinitialize connection
+        print("Try reinitialize Pinecone connection")
         pinecone.init(api_key=PINECONE_KEY, environment='us-west1-gcp')
         index2 = pinecone.Index(index_id)
         try:
+            print("Now try querying pinecone again")
             xc = index2.query(embeds, top_k=30, include_metadata=True)
+            print("query successful")
         except Exception as e:
             raise ValueError(e)
     prompts = [
@@ -95,7 +101,9 @@ def diffuse(text: str):
         added_gcp = False
         # push to storage
         try:
+            print("try push to Cloud Storage")
             blob = bucket.blob(f'images/{_id}.png')
+            print("try upload_from_filename")
             blob.upload_from_filename(f'{_id}.png')
             added_gcp = True
             # add embedding and metadata to Pinecone
@@ -105,12 +113,17 @@ def diffuse(text: str):
                 'image_url': f'images/{_id}.png'
             }
             try:
+                print("now try upsert to pinecone")
                 index.upsert([(_id, embeds, meta)])
+                print("upsert successful")
             except Exception as e:
                 try:
+                    print("hit exception, now trying to reinit Pinecone connection")
                     pinecone.init(api_key=PINECONE_KEY, environment='us-west1-gcp')
                     index2 = pinecone.Index(index_id)
+                    print(f"reconnected to pinecone '{index_id}' index")
                     index2.upsert([(_id, embeds, meta)])
+                    print("upsert successful")
                 except Exception as e:
                     print(f"PINECONE_ERROR: {e}")
         except Exception as e:
@@ -137,8 +150,10 @@ def test_image(_id, image):
         return False
 
 def prompt_image(text: str):
+    print(f"prompt_image('{text}')")
     embeds = encode_text(text)
     try:
+        print("try query pinecone")
         xc = index.query(embeds, top_k=9, include_metadata=True)
     except Exception as e:
         print(f"Error during query: {e}")
@@ -146,6 +161,7 @@ def prompt_image(text: str):
         pinecone.init(api_key=PINECONE_KEY, environment='us-west1-gcp')
         index2 = pinecone.Index(index_id)
         try:
+            print("try query pinecone after reinit")
             xc = index2.query(embeds, top_k=9, include_metadata=True)
         except Exception as e:
             raise ValueError(e)
@@ -155,15 +171,21 @@ def prompt_image(text: str):
     scores = [match['score'] for match in xc['matches']]
     ids = [match['id'] for match in xc['matches']]
     images = []
+    print("Begin looping through (ids, image_urls)")
     for _id, image_url in zip(ids, image_urls):
         try:
+            print("download_as_string from GCP")
             blob = bucket.blob(image_url).download_as_string()
+            print("downloaded successfully")
             blob_bytes = io.BytesIO(blob)
             im = Image.open(blob_bytes)
+            print("image opened successfully")
             if test_image(_id, im):
                 images.append(im)
+                print("image accessible")
             else:
                 images.append(missing_im)
+                print("image NOT accessible")
         except ValueError:
             print(f"ValueError: '{image_url}'")
     return images, scores
@@ -185,6 +207,7 @@ def set_images(text: str):
     else:
         print("NO MATCH FOUND")
         diffuse(text)
+        print(f"diffusion for '{text}' complete")
         images, scores = prompt_image(text)
         return gr.Gallery.update(value=images)