Update variables.py

variables.py

@@ -94,7 +94,7 @@ def create_vectorstore(file,model):
 
     
 @st.experimental_singleton(suppress_st_warning=True)
-def embed_tweets(query,_prompt):
+def embed_tweets(query,_prompt,_docsearch):
     '''Process file with latest tweets'''
 
 
@@ -102,7 +102,7 @@ def embed_tweets(query,_prompt):
     chain = VectorDBQA.from_chain_type(
     ChatOpenAI(temperature=0), 
     chain_type="stuff", 
-    vectorstore=docsearch,
+    vectorstore=_docsearch,
     chain_type_kwargs=chain_type_kwargs,
     return_source_documents=True,
     k=3
@@ -145,136 +145,6 @@ topics = {
     "19": "Stock Movement",
 }
 
-user_name = [
-    "Investing.com",
-    "(((The Daily Shot)))",
-    "Bloomberg Markets",
-    "FirstSquawk",
-    "MarketWatch",
-    "markets",
-    "FinancialTimes",
-    "CNBC",
-    "ReutersBiz",
-    "BreakingNews",
-    "LiveSquawk",
-    "NYSE",
-    "WSJmarkets",
-    "FT",
-    "TheStreet",
-    "ftfinancenews",
-    "BloombergTV",
-    "Nasdaq",
-    "NYSE",
-    "federalreserve",
-    "NewYorkFed",
-    "sffed",
-    "WSJCentralBanks",
-    "RichmondFed",
-    "ecb",
-    "stlouisfed",
-    "WorldBank",
-    "MarketCurrents",
-    "OpenOutcrier",
-    "BullTradeFinder",
-    "WallStChatter",
-    "Briefingcom",
-    "SeekingAlpha",
-    "realDonaldTrump",
-    "AswathDamodaran",
-    "ukarlewitz",
-    "alphatrends",
-    "Investor666",
-    "ACInvestorBlog",
-    "ZorTrades",
-    "ScottNations",
-    "TradersCorner",
-    "TraderGoalieOne",
-    "option_snipper",
-    "jasonleavitt",
-    "LMT978",
-    "OptionsHawk",
-    "andrewbtodd",
-    "Terri1618",
-    "SunriseTrader",
-    "traderstewie",
-    "TMLTrader",
-    "IncredibleTrade",
-    "NYFedResearch",
-    "YahooFinance",
-    "business",
-    "economics",
-    "IMFNews",
-    "Market_Screener",
-    "QuickTake",
-    "NewsFromBW",
-    "BNCommodities",
-]
-
-user_id = [
-    "988955288",
-    "423769635",
-    "69620713",
-    "59393368",
-    "3295423333",
-    "624413",
-    "69620713",
-    "4898091",
-    "20402945",
-    "15110357",
-    "6017542",
-    "21323268",
-    "28164923",
-    "18949452",
-    "15281391",
-    "11014272",
-    "35002876",
-    "18639734",
-    "21323268",
-    "26538229",
-    "15072071",
-    "117237387",
-    "327484803",
-    "16532451",
-    "83466368",
-    "71567590",
-    "27860681",
-    "15296897",
-    "2334614718",
-    "2222635612",
-    "3382363841",
-    "72928001",
-    "23059499",
-    "25073877",
-    "33216611",
-    "37284991",
-    "15246621",
-    "293458690",
-    "55561590",
-    "18560146",
-    "244978426",
-    "85523269",
-    "276714687",
-    "2806294664",
-    "16205561",
-    "1064700308",
-    "61342056",
-    "184126162",
-    "405820375",
-    "787439438964068352",
-    "52166809",
-    "2715646770",
-    "47247213",
-    "374672240",
-    "19546277",
-    "34713362",
-    "144274618",
-    "25098482",
-    "102325185",
-    "252751061",
-    "976297820532518914",
-    "804556370",
-]
-
 sentiment_classifier, topic_classifier = load_models()
 
 def convert_user_names(user_name: list):