Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import fitz
 from transformers import pipeline, MBart50TokenizerFast, MBartForConditionalGeneration
 from multiprocessing import Pool, cpu_count
+import tempfile
 
 # Load summarization pipeline
 summarizer = pipeline("summarization", model="Falconsai/text_summarization")
@@ -62,10 +63,17 @@ def translate_summary(summary, lang):
 
     return " ".join(translated_chunks)
 
+
+
 # Function to read PDF and summarize and translate chunk by chunk
 def summarize_and_translate_pdf(uploaded_file, lang):
+    # Save uploaded PDF to a temporary file
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        temp_file.write(uploaded_file.read())
+        temp_file_path = temp_file.name
+
     try:
-        doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
+        doc = fitz.open(temp_file_path)
     except FileNotFoundError:
         st.error("File not found. Please make sure the file path is correct.")
         return []
@@ -82,6 +90,9 @@ def summarize_and_translate_pdf(uploaded_file, lang):
     with Pool(cpu_count()) as pool:
         translated_chunks = pool.starmap(summarize_and_translate_chunk, [(chunk, lang) for chunk in chunks])
 
+    # Delete temporary file
+    os.unlink(temp_file_path)
+
     return translated_chunks