Update app.py

app.py

@@ -21,13 +21,13 @@ def llm_pipeline(filepath, max_length, min_length):
 
     return summary
 
-# def displayPDF(file):
-#     # Opening file from file path
-#     with open(file, "rb") as f:
-#         base64_pdf = base64.b64encode(f.read()).decode('utf-8')
+def displayPDF(file):
+    # Opening file from file path
+    with open(file, "rb") as f:
+        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
 
-#     pdf_display = F'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="600" type="application/pdf"></iframe>'
-#     st.markdown(pdf_display, unsafe_allow_html=True)
+    pdf_display = F'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="600" type="application/pdf"></iframe>'
+    st.markdown(pdf_display, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     st.set_page_config(layout="wide")
@@ -42,6 +42,8 @@ if __name__ == "__main__":
         if st.button("Summarize"):
             col1, col2 = st.columns(2)
             filepath = uploaded_file.name
+            with open(filepath, "wb") as temp_file:
+                temp_file.write(uploaded_file.read())
             # with col1:
             #     st.info("Uploaded File")
             #     pdf_view = displayPDF(filepath)