Update app.py

app.py

@@ -66,19 +66,15 @@ 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")
+            print("Reinitialized query successful")
         except Exception as e:
             raise ValueError(e)
     scores = [round(match['score'], 2) for match in xc['matches']]
@@ -99,14 +95,11 @@ def test_image(_id, image):
         # delete corrupted file from pinecone and cloud
         index.delete(ids=[_id])
         bucket.blob(f"images/{_id}.png").delete()
-        print(f"DELETED '{_id}'")
         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,
             filter={"image_nsfw": {"$lt": 0.5}}
@@ -117,32 +110,26 @@ 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,
                 filter={"image_nsfw": {"$lt": 0.5}}
             )
+            print("Reinitialized query successful")
         except Exception as e:
             raise ValueError(e)
     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 in ids:
         try:
             image_url = f"images/{_id}.png"
-            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