import streamlit as st
import requests


st.set_page_config(page_title="ChatPDF", page_icon="📄")

st.title("📄 ChatPDF")
uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")

if 'pdf_processed' not in st.session_state:
    st.session_state['pdf_processed'] = False

if uploaded_file is not None and not st.session_state['pdf_processed']:
    upload_file = {'file': uploaded_file}
    response = requests.post(url='http://127.0.0.1:8000/upload_pdf', files=upload_file)
    if response.status_code == 200:
        st.session_state['pdf_processed'] = True
        st.write("PDF processed and vectors upserted!")
    else:
        st.write("Failed to process PDF")

if st.session_state['pdf_processed']:
    st.title("🤖 Ask a Question about the PDF Content")
    add_selectbox = st.sidebar.selectbox("Select Model", ("Default", "deepset/roberta-base-squad2", "llama"))
    question = st.text_input("Ask a question about the PDF content:")
    if question:
        url = f"http://127.0.0.1:8000/ask_question/?question={question}&model={add_selectbox}"
        response = requests.post(url)
        if response.status_code == 200:
            st.write(response.json()['answer'])
        else:
            st.write("Failed to get an answer")