Update variables.py

variables.py

@@ -72,8 +72,8 @@ def process_tweets(df,df_users):
     return df_tweets
 
 @st.experimental_singleton(suppress_st_warning=True)
-def embed_tweets(file,model,query,_prompt):
-    '''Process file with latest tweets'''
+def create_vectorstore(file,model):
+    '''Create FAISS vectorstore'''
 
     # Split tweets int chunks
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50)
@@ -90,6 +90,14 @@ def embed_tweets(file,model,query,_prompt):
 
     docsearch = FAISS.from_texts(texts, emb)
 
+    return docsearch
+
+    
+@st.experimental_singleton(suppress_st_warning=True)
+def embed_tweets(query,_prompt):
+    '''Process file with latest tweets'''
+
+
     chain_type_kwargs = {"prompt": _prompt}
     chain = VectorDBQA.from_chain_type(
     ChatOpenAI(temperature=0), 
@@ -97,7 +105,7 @@ def embed_tweets(file,model,query,_prompt):
     vectorstore=docsearch,
     chain_type_kwargs=chain_type_kwargs,
     return_source_documents=True,
-    k=5
+    k=3
     )
 
     result = chain({"query": query})