structure

app.py

@@ -9,6 +9,7 @@ import my_new_openai
 import time
 import streamlit as st
 import os
+from PIL import Image
 
 
 def merge_indices(index1, index2):
@@ -147,8 +148,11 @@ def main():
                     else:
                         st.warning("unsaved embeddings will be lost.")
         else:
+            if not sst.vectorstore:
+                vec_store = setup_db.load_vectorstore_from_excel("data/KBOB_Klassifizierung.xlsx")
+                sst.vectorstore = True
+            sst.page = "home"
             file = st.file_uploader("upload file", accept_multiple_files=False)
-            vec_store = setup_db.load_vectorstore_from_excel("data/KBOB_Klassifizierung.xlsx")
             if st.button("classify me!"):
                 with st.spinner("Classifying..."):
                     query_vecs = []
@@ -158,13 +162,13 @@ def main():
                         with one:
                             st.success("text")
                         # ONE OR MULTIPLE IS THE QUESTION
-                        imgs = ingest.get_pdf_images(file.getvalue())
-                        if type(imgs) != list:
-                            imgs = [imgs]
-                        for img in imgs:
+                        images = ingest.get_pdf_images(file.getvalue())
+                        if type(images) != list:
+                            images = [images]
+                        for img in images:
                             text += my_new_openai.img_to_text(img_base64=my_new_openai.image_bytes_to_base64(img))
                         with two:
-                            st.success("imgs")
+                            st.success("images")
 
                         tabs = ingest.get_pdf_tables(file.getvalue())