import tempfile
import streamlit as st
from PIL import Image
import os
from utils.ingest1 import create_vector_database

def process_uploaded_file():
    st.title("Upload File to Chat")
    uploaded_file = st.file_uploader("File upload", type="pdf")
    if uploaded_file:
        temp_dir = tempfile.mkdtemp()
        path = os.path.join(temp_dir, uploaded_file.name)
    #with open(path, "wb") as f:
    #    f.write(uploaded_file.getvalue())
        print(path)    
        st.write("Document uploaded successfully!")
        # Display the uploaded document
        st.write("Preview of the document:")
        st.write(uploaded_file)

        # Button to start parsing and vector database creation
        if st.button("Start Processing"):
            # Placeholder for processing logic
            st.write("Processing...")

            # Placeholder for progress bar
            with st.spinner('Processing...'):
                # Call your function to parse data and create vector database
                create_vector_database(path)

            st.success("Processing completed!")

            # Display success message
            st.write("Vector database created successfully!")

            # Show success image
            success_image = Image.open("success_image.jpg")
            st.image(success_image, caption="Success!", use_column_width=True)

    # Add a footer
    #st.text("Built with Streamlit")