Update app.py

app.py

@@ -96,18 +96,17 @@ class DocumentQA:
         Raises:
             ValueError: If an error occurs during the process.
         """
-
-        total_chemical=[]
-
+        webpage_text =''
         for url in urls.split(','):
-          webpage_text = self.get_content_from_url(url)
-          knowledge_base = self.create_knowledge_base(webpage_text)
-          state = {"knowledge_base": knowledge_base}
-          chemicals = self.get_chemicals_for_file(state,input_qus)
-          total_chemical.append(str(url)+"\n"+chemicals+"\n\n")
-        list_of_chemicals = "".join(total_chemical)
+          webpage_text += self.get_content_from_url(url)
+        
+        knowledge_base = self.create_knowledge_base(webpage_text)
+        
+        state = {"knowledge_base": knowledge_base}
+            
+        chemicals = self.get_chemicals_for_file(state,input_qus)
 
-        return list_of_chemicals
+        return chemicals
 
 
 
@@ -155,24 +154,23 @@ class DocumentQA:
 
 
         file_paths = [single_file_path.name for single_file_path in file_paths]
-        results =''
+        docs =''
         for file_obj in file_paths:
 
            loader = UnstructuredFileLoader(file_obj, strategy="fast")
 
            # Load the contents of the file using the loader
-           docs =loader.load()
-
-           # 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)
-           
-           final_ans = self.get_chemicals_for_file(state,question)
-           results += pdf_name+"\n"+final_ans+"\n\n"
-
+           docs +=loader.load()
+
+        # 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)
+       
+        final_ans = self.get_chemicals_for_file(state,question)
+        
         # Return a tuple containing the file name and the knowledge base
-        return results
+        return final_ans
 
     def get_final_result(self,urls,file_paths,state,input_qus):