| 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 |
|
|
| |
| csv_loader = CSVLoader(file_path = temp_file_path, encoding = 'latin-1', csv_args={'delimiter':','}) |
|
|
| |
| data = csv_loader.load() |
|
|
| |
| 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() |