making intialized vectorstore

app.py

@@ -9,6 +9,7 @@ class PDFProcessorUI:
         self.current_language = "English"
         self.current_ai_model = "Huggingface / IBM granite granite 3.1 8b Instruct"
         self.current_type_model = "Api Key"
+        self.vectorstore = gr.State()
     
     def change_language(self, language):
         self.current_language = language
@@ -76,7 +77,6 @@ class PDFProcessorUI:
     
     def create_ui(self):
         with gr.Blocks() as demo:
-            vectorstore = gr.State()
             title = gr.Markdown(TRANSLATIONS[self.current_language]["title"])
             
             with gr.Row():
@@ -212,31 +212,31 @@ class PDFProcessorUI:
             
             chat_placeholder.submit(
                 fn=self.qa_interface,
-                inputs=[vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                inputs=[self.vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
                 outputs=[chatbot]
             )
             
             process_btn.click(
                 fn=self.process_pdf,
-                inputs=[vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
-                outputs=[process_output, vectorstore]
+                inputs=[self.vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                outputs=[process_output, self.vectorstore]
             )
             
             summarize_btn.click(
                 fn=self.summarize_interface,
-                inputs=[vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                inputs=[self.vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
                 outputs=[summary_output]
             )
 
             specialist_btn.click(
                 fn=self.specialist_opinion,
-                inputs=[vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx, specialist_placeholder],
+                inputs=[self.vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx, specialist_placeholder],
                 outputs=[specialist_output]
             )
 
             chat_btn.click(
                 fn=self.qa_interface,
-                inputs=[vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                inputs=[self.vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
                 outputs=[chatbot]
             )
         

pdf_processor.py

@@ -188,6 +188,8 @@ class PDFProcessor:
                     collection_name="pdf_collection"
                     #persist_directory="./chroma_db"
                 )
+
+                print("vectorstore: ", vectorstore)
                 
                 return TRANSLATIONS[self.language]["pdf_processed"], vectorstore #+ f" ---- Chunks: {len(vectorstore.get()["documents"])}"