markytools commited on
Commit
d398fc9
·
1 Parent(s): 89171ff

upgraded stock market simulator

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -116,14 +116,18 @@ else:
116
  fig.add_trace(go.Bar(x=df['Date'], y=df['Volume'][ticker_symbol], name='Volume'))
117
  fig.update_layout(title="Volume Analysis", xaxis_title="Date", yaxis_title="Volume")
118
  st.plotly_chart(fig)
 
 
119
 
120
- loader = []
121
- index = VectorstoreIndexCreator(
122
- embedding=GooglePalmEmbeddings(),
123
- text_splitter=CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)).from_loaders(loader)
 
124
  chain = RetrievalQA.from_chain_type(llm=llm,
125
  chain_type="stuff",
126
  retriever=index.vectorstore.as_retriever(),
 
127
  input_key="question")
128
 
129
  # Additional Insights
 
116
  fig.add_trace(go.Bar(x=df['Date'], y=df['Volume'][ticker_symbol], name='Volume'))
117
  fig.update_layout(title="Volume Analysis", xaxis_title="Date", yaxis_title="Volume")
118
  st.plotly_chart(fig)
119
+
120
+ from langchain.document_loaders import DataFrameLoader
121
 
122
+ loader = DataFrameLoader(df)
123
+ index = VectorstoreIndexCreator(embedding=GooglePalmEmbeddings()).from_loaders([loader])
124
+ # index = VectorstoreIndexCreator(
125
+ # embedding=GooglePalmEmbeddings(),
126
+ # text_splitter=CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)).from_loaders(loader)
127
  chain = RetrievalQA.from_chain_type(llm=llm,
128
  chain_type="stuff",
129
  retriever=index.vectorstore.as_retriever(),
130
+
131
  input_key="question")
132
 
133
  # Additional Insights