Update app.py

app.py

@@ -88,16 +88,13 @@ def load_recommender(paths, start_page=1):
 
     for path in paths:
         texts = pdf_to_text(path, start_page=start_page)
-        all_texts.extend([(text, path) for text in texts])
-
-    chunks = []
-    for text, path in all_texts:
-        pdf_chunks = text_to_chunks([text], start_page=start_page)
-        pdf_chunks = [f'[PDF {os.path.basename(path)}, {chunk}]' for chunk in pdf_chunks]
-        chunks.extend(pdf_chunks)
+        all_texts.extend(texts)
 
+    chunks = text_to_chunks(all_texts, start_page=start_page)
     recommender.fit(chunks)
     return 'Corpus Loaded.'
+
+    
 def generate_text(openAI_key, prompt, engine="text-davinci-003"):
     openai.api_key = openAI_key
     completions = openai.Completion.create(
@@ -133,32 +130,33 @@ def generate_answer(question, openAI_key):
 
 def question_answer(url, files, question, openAI_key):
     if openAI_key.strip() == '':
-        return '[ERROR]: Please enter your Open AI Key. Get your key here: https://platform.openai.com/account/api-keys'
+        return '[ERROR]: Please enter your Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
     if url.strip() == '' and not files:
         return '[ERROR]: Both URL and PDF are empty. Provide at least one.'
-    
+
     if url.strip() != '' and files:
         return '[ERROR]: Both URL and PDF are provided. Please provide only one (either URL or PDF).'
 
+    pdf_paths = []
+
     if url.strip() != '':
         glob_url = url
-        download_pdf(glob_url, 'corpus.pdf')
-        load_recommender(['corpus.pdf'])
+        output_path = 'corpus.pdf'
+        download_pdf(glob_url, output_path)
+        pdf_paths.append(output_path)
 
     else:
-        file_paths = []
         for file in files:
             old_file_name = file.name
             file_name = file.name
             file_name = file_name[:-12] + file_name[-4:]
             os.rename(old_file_name, file_name)
-            file_paths.append(file_name)
-
-        load_recommender(file_paths)
+            pdf_paths.append(file_name)
 
     if question.strip() == '':
         return '[ERROR]: Question field is empty'
 
+    load_recommender(pdf_paths)
     return generate_answer(question, openAI_key)
 
 recommender = SemanticSearch()