Update app.py

app.py

@@ -12,9 +12,11 @@ def simplify_text(text):
     return simplified[0]['summary_text']
 
 def extract_text_from_pdf(pdf_file):
-    """Extracts text from an uploaded PDF file."""
-    doc = fitz.open(pdf_file)
-    text = "\n".join(page.get_text("text") for page in doc)
+    """Extracts text from an uploaded PDF file stream."""
+    text = ""
+    with fitz.open(stream=pdf_file.read(), filetype="pdf") as doc:
+        for page in doc:
+            text += page.get_text()
     return text
 
 # Streamlit UI
@@ -26,14 +28,18 @@ if option == "Text Input":
     if st.button("Simplify") and user_text:
         simplified_text = simplify_text(user_text)
         st.subheader("Simplified Text:")
-        st.write(simplified_text)
+        st.text_area("Simplified Output", simplified_text, height=150)
 
 elif option == "Upload PDF":
     uploaded_file = st.file_uploader("Upload a PDF", type=["pdf"])
     if uploaded_file:
         extracted_text = extract_text_from_pdf(uploaded_file)
-        simplified_text = simplify_text(extracted_text[:1000])  # Limit text length for processing
-        st.subheader("Simplified Text:")
-        st.write(simplified_text)
+        st.subheader("Extracted Text from PDF:")
+        st.text_area("Extracted Text", extracted_text, height=200)
+
+        if st.button("Simplify Extracted Text"):
+            simplified_text = simplify_text(extracted_text[:1000])  # Limit length for model input
+            st.subheader("Simplified Text:")
+            st.text_area("Simplified Output", simplified_text, height=150)
 
 st.write("\nMade with ❤️ using Pretrained Models")