Update app.py

app.py

@@ -29,26 +29,33 @@ def process_transcripts(files, context, session_key):
 
 @spaces.GPU
 def retrieve_answer(question, goals, session_key):
-    
+    keycheck = namespace_check(session_key)
     with QAService(conf, 
                    pinecone=pc_connector, 
                    model_pipeline=pipelines, 
                    question=question, 
                    goals=goals,
-                   session_key=session_key) as q:
+                   session_key=session_key,
+                   keycheck=keycheck) as q:
         f, c = q.run()
 
     return f, c
 
+def namespace_check(arg):
+    index = pc_connector.Index(conf["embeddings"]["index_name"])
+    stats = index.DescribeIndexStats()
+    name_list = stats['namespaces'].keys()
+    arg in name_list
+    return arg in name_list
+
 
 def drop_namespace(arg):
     if conf["embeddings"]["override"]:
         pass
         print("Maintained Namespace: " + conf["embeddings"]["demo_namespace"])
     else:
-        index = pc_connector.Index(conf["embeddings"]["index_name"])
-        stats=index.DescribeIndexStats()
-        if arg in stats['namespaces'].keys():        
+        namecheck = namespace(arg)
+        if namecheck:        
             pc_connector.delete(namespace=arg, delete_all=True)
         print("Deleted namespace: " + arg)