changing to Gradio

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 from haystack.document_stores import InMemoryDocumentStore
 from haystack.nodes import FARMReader, PreProcessor, PDFToTextConverter, TfidfRetriever
 import logging
@@ -14,8 +14,6 @@ preprocessor = PreProcessor(
     split_overlap=3
 )
 
-uploaded_files = st.file_uploader(label='Upload a PDF Document', accept_multiple_files=True)
-logging.info(uploaded_files)
 
 def pdf_to_document_store(pdf_files):
     document_store.delete_documents()
@@ -27,6 +25,17 @@ def pdf_to_document_store(pdf_files):
     document_store.write_documents(preprocessed_docs)
     return None
 
-if uploaded_files is not None:
-    document_store.delete_all_documents()
-    pdf_to_document_store(uploaded_files)
+
+def summarize(files):
+    if files is not None:
+        document_store.delete_all_documents()
+        pdf_to_document_store(files)
+    return document_store.get_document_count()
+
+title = "Summarize one or more PDFs with a Haystack Summariser pipeline"
+iface = gr.Interface(fn=summarize,
+                    inputs=[gr.inputs.File(file_count="multiple", type="file", label="Upload some PDFs")],
+                    outputs="text",
+                    title=title,
+                    theme="huggingface")
+iface.launch()

requirements.txt

@@ -1 +1 @@
-farm-haystack==1.4.0
+farm-haystack[ocr]==1.4.0