Update app.py

app.py

@@ -90,7 +90,19 @@ def get_formatted_text_for_annotation(output):
         annotated_texts.append(text[next_index-1:len(text)-1])
         
     return tuple(annotated_texts)
-    
+
+def displayPDF(file):
+    # Opening file from file path
+    with open(file, "rb") as f:
+        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
+
+    # Embedding PDF in HTML
+    pdf_display = F'<iframe src="data:application/pdf;base64,{base64_pdf}" width="700" height="1000" type="application/pdf"></iframe>'
+
+    # Displaying File
+    st.markdown(pdf_display, unsafe_allow_html=True)    
+
+
 # Model initialization    
 pipeline_summarization = init_text_summarization_model()
 pipeline_zsl, template = init_zsl_topic_classification()
@@ -116,6 +128,7 @@ if selected_menu == "Upload Document":
         
         with open(file_path,"wb") as f: 
             f.write(uploaded_file.getbuffer())  
+        displayPDF(file_path)
         
 elif selected_menu == "Extract Text":
     with st.spinner("Extracting Text..."):