csv / app.py
Parthx10's picture
Upload 4 files
1cc9481 verified
raw
history blame contribute delete
934 Bytes
import streamlit as st
from langchain.document_loaders.csv_loader import CSVLoader
import tempfile
from bd import get_response
def main():
st.title("Chat with CSV")
uploaded_file = st.sidebar.file_uploader("choose your csv file",type='csv')
if uploaded_file is not None:
with tempfile.NamedTemporaryFile(delete=False) as temp_file:
temp_file.write(uploaded_file.getvalue())
temp_file_path = temp_file.name
#Intializing CSVLoader
csv_loader = CSVLoader(file_path = temp_file_path, encoding = 'latin-1', csv_args={'delimiter':','})
#load csv data
data = csv_loader.load()
#input
user_ip = st.text_input("Enter Your Query/Message")
print(user_ip)
if user_ip:
response=get_response(data,user_ip)
st.write(response)
if __name__ == "__main__" :
main()