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