aiscientist commited on
Commit
232ae3b
·
verified ·
1 Parent(s): 5c2fd9a

Upload 3 files

Browse files
Files changed (3) hide show
  1. app.py +35 -0
  2. personal.csv +0 -0
  3. requirements.txt +4 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import os
3
+ os.environ['OPENAI_API_KEY'] = "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"
4
+
5
+ from langchain.llms.openai import OpenAI
6
+ from langchain.agents.agent_types import AgentType
7
+ #from langchain.agents import create_csv_agent
8
+ from langchain_experimental.agents import create_csv_agent
9
+ import time
10
+
11
+ def main():
12
+ st.title("Personal Assistant")
13
+
14
+ csv_file_path = "personal.csv"
15
+ llm = OpenAI(temperature=0)
16
+ agent = create_csv_agent(
17
+ llm,
18
+ csv_file_path,
19
+ verbose=False,
20
+ agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
21
+ )
22
+
23
+ query = st.text_input("What would you like to know?")
24
+ if st.button("Ask"):
25
+ if query.strip() == "":
26
+ st.warning("Please enter a query.")
27
+ else:
28
+ start = time.time()
29
+ answer = agent.run(query)
30
+ end = time.time()
31
+ st.write(answer)
32
+ st.write(f"Answer (took {round(end - start, 2)} s.)")
33
+
34
+ if __name__ == "__main__":
35
+ main()
personal.csv ADDED
The diff for this file is too large to render. See raw diff
 
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ langchain
2
+ langchain_experimental
3
+ openai
4
+ streamlit