Update utils/haystack.py

utils/haystack.py

@@ -8,7 +8,7 @@ from haystack_integrations.components.retrievers.pinecone import PineconeEmbeddi
 from haystack_integrations.components.generators.cohere import CohereGenerator
 from haystack import Document
 
-def start_haystack(openai_key):
+def start_haystack(openai_key):    
     document_store = PineconeDocumentStore(dimension=1024, index="zen", environment = "gcp-starter")
     
     template = """
@@ -46,24 +46,27 @@ def start_haystack(openai_key):
 
 @st.cache_data(show_spinner=True)
 def query(prompt, _pipe):
-    try:        
-        replies = _pipe.run({
-            "text_embedder": {
-                "text": prompt
-            },
-            "prompt_builder": {
-                "query": prompt
-            },
-            "answer_builder": {
-                "query": prompt
-            }
-        })
-
-        raw = replies['answer_builder']['answers'][0]
-        result = raw.data + "\n\n -- Source: " + raw.documents[0].content + " --"
-        print(raw)
-    except Exception as e:
-        print("Hay:")
-        print(e)
-        result = ["Something went wrong!"]
-    return result
+    with st.spinner('Processing'):
+        try:        
+            replies = _pipe.run({
+                "text_embedder": {
+                    "text": prompt
+                },
+                "prompt_builder": {
+                    "query": prompt
+                },
+                "answer_builder": {
+                    "query": prompt
+                }
+            })
+    
+            raw = replies['answer_builder']['answers'][0]
+            result = raw.data + "\n\n -- Source: " + raw.documents[0].content + " --"
+            print(raw)
+            st.success('Completed!')
+        except Exception as e:
+            print("Hay:")
+            print(e)
+            result = ["Something went wrong!"]
+            st.error('Failed!')
+        return result