demo.queue() at launch

app.py

@@ -9,7 +9,6 @@ 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 = None
     
     def change_language(self, language):
         self.current_language = language
@@ -77,6 +76,7 @@ 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=[self.vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                inputs=[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=[self.vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
-                outputs=[process_output, self.vectorstore]
+                inputs=[vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                outputs=[process_output, vectorstore]
             )
             
             summarize_btn.click(
                 fn=self.summarize_interface,
-                inputs=[self.vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                inputs=[vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
                 outputs=[summary_output]
             )
 
             specialist_btn.click(
                 fn=self.specialist_opinion,
-                inputs=[self.vectorstore, ai_model_dropdown, type_model, api_key_input, project_id_watsonx, specialist_placeholder],
+                inputs=[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=[self.vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
+                inputs=[vectorstore, chat_placeholder, chatbot, ai_model_dropdown, type_model, api_key_input, project_id_watsonx],
                 outputs=[chatbot]
             )
         
@@ -245,4 +245,4 @@ class PDFProcessorUI:
 if __name__ == "__main__":
     ui = PDFProcessorUI()
     demo = ui.create_ui()
-    demo.launch()
+    demo.queue().launch()