Update app.py

app.py

@@ -167,8 +167,10 @@ class ChemicalIdentifier:
       file_paths = [single_file_path.name for single_file_path in file_paths]
       return file_paths
 
-    def get_chemicals_for_file(self,state,knowledge_base):
+    def get_chemicals_for_file(self,state):
 
+        knowledge_base = state["knowledge_base"]
+        
         # Set the question for which we want to find the answer
         question = "Identify the Chemical Capabilities Only"
 
@@ -217,10 +219,10 @@ class ChemicalIdentifier:
 
            # Create a knowledge base from the loaded documents using the create_knowledge_base() method
            knowledge_base = self.create_knowledge_base(docs)
-
+           state = {"knowledge_base": knowledge_base}
            pdf_name = os.path.basename(file_obj)
            global results
-           final_ans = self.get_chemicals_for_file(state,knowledge_base)
+           final_ans = self.get_chemicals_for_file(state)
            results += pdf_name+"\n"+final_ans+"\n\n"
 
         # Return a tuple containing the file name and the knowledge base