Update app.py

app.py

@@ -61,7 +61,17 @@ def encode_text(text: str):
 
 def prompt_query(text: str):
     embeds = encode_text(text)
-    xc = index.query(embeds, top_k=30, include_metadata=True)
+    try:
+        xc = index.query(embeds, top_k=30, include_metadata=True)
+    except Exception as e:
+        print(f"Error during query: {e}")
+        # reinitialize connection
+        pinecone.init(api_key=PINECONE_KEY, environment='us-west1-gcp')
+        index = pinecone.Index(index_id)
+        try:
+            xc = index.query(embeds, top_k=30, include_metadata=True)
+        except Exception as e:
+            raise ValueError(e)
     prompts = [
         match['metadata']['prompt'] for match in xc['matches']
     ]
@@ -115,7 +125,17 @@ def test_image(_id, image):
 
 def prompt_image(text: str):
     embeds = encode_text(text)
-    xc = index.query(embeds, top_k=9, include_metadata=True)
+    try:
+        xc = index.query(embeds, top_k=9, include_metadata=True)
+    except Exception as e:
+        print(f"Error during query: {e}")
+        # reinitialize connection
+        pinecone.init(api_key=PINECONE_KEY, environment='us-west1-gcp')
+        index = pinecone.Index(index_id)
+        try:
+            xc = index.query(embeds, top_k=9, include_metadata=True)
+        except Exception as e:
+            raise ValueError(e)
     image_urls = [
         match['metadata']['image_url'] for match in xc['matches']
     ]