import streamlit as st
from io import StringIO
from pypdf import PdfReader
#from PyPDF2 import PdfReader
from mainapp import parse_document, parse_url

files= st.file_uploader(label="upload a file", accept_multiple_files=True, type="pdf")
st.write("if you prefer to interact with a particular URL such as a docs e.g https://docs.python.org,\n")
url = st.text_input("provide the URL in the field below, then press the enter key")
st.write(url)

# st.write(file_upload[0]._file_urls.upload_url)

all_docs = []
docs = ""
input_text = st.text_input(label="Ask your documents/url a question:")
# st.write(input_text)
    
pressed = st.button(label="Get Response", type="primary")

user_query = "inurl: " + url + " " + input_text

if len(url) > 0 and pressed is True:
    #st.write(url)
    #input_text

    response = parse_url(user_query)
    
    if 'response' not in st.session_state:
        st.session_state['response'] = response

    st.write(st.session_state.response)

else:
    try:        
        for file in files:
            if file is not None:
                file_data = PdfReader(file)
                # extract text from the pdf file
                for page in file_data.pages:
                    docs += page.extract_text()
            #all_docs.append(docs)
            if len(input_text) > 0:
                response = parse_document(docs=docs, question= input_text)
                if 'file_response' not in st.session_state:
                    st.session_state['file_response'] = response
                    st.write(st.session_state.file_response)
            else:
                st.write("Ask a question")
    except:
        st.write("No answer")