|
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() |