Update variables.py

variables.py

@@ -74,7 +74,7 @@ def process_tweets(df,df_users):
     return df_tweets
 
 @st.experimental_singleton(suppress_st_warning=True)
-def create_vectorstore(texts,model):
+def create_vectorstore(texts,model,username,topic,creation_time):
     '''Create FAISS vectorstore'''
 
     if model == "hkunlp/instructor-large":
@@ -85,7 +85,8 @@ def create_vectorstore(texts,model):
     elif model == "sentence-transformers/all-mpnet-base-v2":
         emb = HuggingFaceEmbeddings(model_name=model)
 
-    docsearch = FAISS.from_texts(texts, emb)
+    docsearch = FAISS.from_texts(texts, emb,
+                                metadatas=[{"source": user,"topic":top,"extraction_time":tme} for user,top,tme in zip(username,topic,creation_time)])
 
     return docsearch