switching to single file

app.py

@@ -17,23 +17,23 @@ def start_haystack():
         split_length=200,
         split_respect_sentence_boundary=True,
     )
-    summarizer = TransformersSummarizer(model_name_or_path="facebook/bart-large-cnn")
+    summarizer = TransformersSummarizer(model_name_or_path="facebook/bart-large-cnn", batch_size=5)
     return document_store, summarizer, preprocessor
 
 
-def pdf_to_document_store(pdf_files):
+def pdf_to_document_store(pdf_file):
+    document_store.delete_documents()
     converter = PDFToTextConverter(remove_numeric_tables=True, valid_languages=["en"])
-    for pdf in pdf_files:
-        with open("temp-path.pdf", 'wb') as temp_file:
-            base64_pdf = base64.b64encode(pdf.read()).decode('utf-8')
-            temp_file.write(base64.b64decode(base64_pdf))
-            doc = converter.convert(file_path="temp-path.pdf", meta=None)[0]
-            preprocessed_docs=preprocessor.process([doc])
-            document_store.write_documents(preprocessed_docs)
+    with open("temp-path.pdf", 'wb') as temp_file:
+        base64_pdf = base64.b64encode(pdf_file.read()).decode('utf-8')
+        temp_file.write(base64.b64decode(base64_pdf))
+        doc = converter.convert(file_path="temp-path.pdf", meta=None)
+        preprocessed_docs=preprocessor.process(doc)
+        document_store.write_documents(preprocessed_docs)
         temp_file.close()
 
-def summarize(files):
-    pdf_to_document_store(files)
+def summarize(file):
+    pdf_to_document_store(file)
     summaries = summarizer.predict(documents=document_store.get_all_documents(), generate_single_summary=True)
     st.write('Summary')
     for summary in summaries:
@@ -41,14 +41,8 @@ def summarize(files):
         
 document_store, summarizer, preprocessor = start_haystack()
 
-uploaded_files = st.file_uploader("Choose PDF files", accept_multiple_files=True)
+uploaded_file = st.file_uploader("Choose a PDF file", accept_multiple_files=False)
 
-if uploaded_files is not None:
-    if st.button('Summarize Documents'):
-        summarize(uploaded_files)
-
-if st.button('Calculate num of docs'):
-    st.write(document_store.get_document_count())
-
-if st.button('Clear DocumentStore'):
-    document_store.delete_documents()
+if uploaded_file is not None:
+    if st.button('Summarize Document'):
+        summarize(uploaded_file)