Update app.py

app.py

@@ -10,14 +10,16 @@ def clean_(s):
     s = s.replace("\n0: ", "=")
     return re.split('=', s, maxsplit=1)[-1].strip()
 
-def similarity_search2(vectordb, query, k=1, unique="True"):
+def similarity_search2(vectordb, query, k, unique):
     print(f"\nQuery Key: {query}, \nrows requested:{k}\nUnique values:{unique}")
     if unique == "False":
         vals = vectordb.similarity_search(query,k=k)
     else:
         vals = vectordb.similarity_search(query,k=1)
+    temp = []
     for val in vals:
-        return clean_(val.page_content)
+        temp.append(clean_(val.page_content))
+    return str(np.array(temp))[1:-1]
     
 with gr.Blocks() as demo:
     gr.Markdown(