streamlit pdf viewer - trial

app.py

@@ -1,3 +1,5 @@
+import warnings
+warnings.filterwarnings("ignore")
 import streamlit as st
 from PyPDF2 import PdfReader
 from concurrent.futures import ThreadPoolExecutor
@@ -5,11 +7,16 @@ from base64 import b64encode
 from fpdf import FPDF
 import io, string, re, math
 from io import StringIO
-# from streamlit_pdf_viewer import pdf_viewer
+from streamlit_pdf_viewer import pdf_viewer
+#import deepcopy
 
 # Importing the Fastify Class
 from fast_reader import Fastify_Reader
 
+def text_to_binary(text):
+    return ''.join(bin(ord(char))[2:].zfill(8) for char in text)
+
+
 def pdf_extract_text(pdf_docs):
     '''
     Basic function for extracting text from the PDFs
@@ -21,10 +28,11 @@ def pdf_extract_text(pdf_docs):
             text += page.extract_text()
     return text
 
-def text_to_pdf_fastify(text):    
+def text_to_pdf_fastify(txt):    
     '''
     Basic function to apply fastification on the input text and convert it to bytes for PDF rendering
     '''
+    text = (txt + '.')[:-1]
     # Applying the Fastify Logic
     bold_text = Fastify_Reader(text).fastify()
     bold_text = bold_text.encode('latin-1', 'ignore').decode('latin-1') #since fpdf works with latin-1 encoding
@@ -36,10 +44,11 @@ def text_to_pdf_fastify(text):
     pdf.multi_cell(0, 10, txt = bold_text, markdown=True)
     return bytes(pdf.output())
 
-def text_to_pdf(text):
+def text_to_pdf(txt):
     '''
     Basic function on the input text and convert it to bytes for PDF rendering
     '''
+    text = (txt + '.')[:-1]
     text = text.encode('latin-1', 'ignore').decode('latin-1') #since fpdf works with latin-1 encoding
     # Creating the PDF
     pdf = FPDF()
@@ -80,31 +89,6 @@ with tab1:
         with st.spinner("Processing"):
             text = user_input
 
-            # Generating base64 encoded text bytes for PDF rendering
-            original_pdf = b64encode(text_to_pdf(text)).decode("utf-8")
-            base64_pdf = b64encode(text_to_pdf_fastify(text)).decode("utf-8")
-
-            # Embedding the PDFs in the HTML
-            original_display = f'<embed src="data:application/pdf;base64,{original_pdf}" width = "100%" height = 600 type="application/pdf" download="original.pdf">'
-            pdf_display = f'<embed src="data:application/pdf;base64,{base64_pdf}" width = "100%" height = 600 type="application/pdf" download="Modified.pdf">'
-
-            # original_display = f"""<embed class="pdfobject" type="application/pdf" title="Original PDF" src="data:application/pdf;base64,{original_pdf}" style=" width: 100%; height: 150%;">"""
-            # pdf_display = f"""<embed class="pdfobject" type="application/pdf" title="Modified PDF" src="data:application/pdf;base64,{base64_pdf}" style=" width: 100%; height: 150%;">"""
-            
-            # original_display = f'<iframe src="data:application/pdf;base64,{original_pdf}" width = "100%" height = 600 type="application/pdf" download="original.pdf"></iframe>'
-            # pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width = "100%" height = 600 type="application/pdf" download="Modified.pdf"></iframe>'
-            
-            # original_display = f"""
-            #     <div class="pdf-container" style="height: 500px;">
-            #         <embed class="pdfobject" type="application/pdf" title="Modified PDF" src="data:application/pdf;base64,{original_pdf}" style="width: 100%; height: 100%;">
-            #     </div>
-            #     """
-            # pdf_display = f"""
-            #     <div class="pdf-container" style="height: 500px;">
-            #         <embed class="pdfobject" type="application/pdf" title="Modified PDF" src="data:application/pdf;base64,{base64_pdf}" style="width: 100%; height: 100%;">
-            #     </div>
-            #     """
-
             # Compare Logic implementation
             if compare:
                 col1, col2, col3 = st.columns(3)
@@ -115,34 +99,32 @@ with tab1:
                 col1, col2 = st.columns([1, 1], gap="small")
                 with col1:
                     with st.container(border = True):
-                        st.markdown("<div style='text-align: center;'><strong>Original PDF viewer</strong></div>", unsafe_allow_html=True)
-                        st.markdown(original_display, unsafe_allow_html=True)
+                        st.markdown("<div style='text-align: center;'><strong>Original PDF viewer</strong></div>", unsafe_allow_html=True)               
+                        # st.markdown(original_display, unsafe_allow_html=True)
+                        pdf_viewer(input = text_to_pdf(text), width = 600)
                 with col2:
                     with st.container(border = True):
                         st.markdown("<div style='text-align: center;'><strong>Fastified PDF viewer</strong></div>", unsafe_allow_html=True)
-                        st.markdown(pdf_display, unsafe_allow_html=True)
+                        # st.markdown(pdf_display, unsafe_allow_html=True)
+                        pdf_viewer(text_to_pdf_fastify(text), width = 600)
                 
-                # Browser Cache Note
-                st.markdown(f"""
-                        <div style='background-color: #FFD580; border-radius: 5px;'>
-                            <p style='color: black;'><strong>Note</strong> - {note_text}</p>
-                        </div>
-                        """, unsafe_allow_html=True)
+                # # Browser Cache Note
+                # st.markdown(f"""
+                #         <div style='background-color: #FFD580; border-radius: 5px;'>
+                #             <p style='color: black;'><strong>Note</strong> - {note_text}</p>
+                #         </div>
+                #         """, unsafe_allow_html=True)
             else:
-                # No Comparisons
+                # # No Comparisons
                 col1, col2, col3 = st.columns(3)
-                with col2:
+                with col1:
                     st.download_button(label="Download Fastified PDF", data=text_to_pdf_fastify(text), file_name='output.pdf', mime='application/pdf')
-                with st.container(border = True):
-                    st.markdown("<div style='text-align: center;'><strong>Fastified PDF viewer</strong></div>", unsafe_allow_html=True)
-                    st.markdown(pdf_display, unsafe_allow_html=True)
+                with st.container():
+                    
+                    st.markdown("<div><strong>Fastified PDF viewer</strong></div>", unsafe_allow_html=True)
+                    # st.markdown(pdf_display, unsafe_allow_html=True)
+                    pdf_viewer(text_to_pdf_fastify(text), width = 600)
                 
-                # Browser Cache Note
-                    st.markdown(f"""
-                        <div style='background-color: #FFD580; border-radius: 5px;'>
-                            <p style='color: black;'><strong>Note</strong> - {note_text}</p>
-                        </div>
-                        """, unsafe_allow_html=True)
     
 # Added support for PDFs having text
 with tab2:
@@ -157,34 +139,6 @@ with tab2:
             with st.spinner("Processing"):
                 text = pdf_extract_text(uploaded_file)
 
-                # Generating base64 encoded text bytes for PDF rendering
-                original_pdf = b64encode(text_to_pdf(text)).decode("utf-8")
-                base64_pdf = b64encode(text_to_pdf_fastify(text)).decode("utf-8")
-
-                # Embedding the PDFs in the HTML
-                original_display = f'<embed src="data:application/pdf;base64,{original_pdf}" width = "100%" height = 600 type="application/pdf" download="original.pdf">'
-                pdf_display = f'<embed src="data:application/pdf;base64,{base64_pdf}" width = "100%" height = 600 type="application/pdf" download="Modified.pdf">'
-
-                
-                # original_display = f"""<embed class="pdfobject" type="application/pdf" title="Original PDF" src="data:application/pdf;base64,{original_pdf}" style=" width: 100%; height: 150%;">"""
-                # pdf_display = f"""<embed class="pdfobject" type="application/pdf" title="Modified PDF" src="data:application/pdf;base64,{base64_pdf}" style=" width: 100%; height: 150%;">"""
-                
-                # original_display = f'<iframe src="data:application/pdf;base64,{original_pdf}" width = "100%" height = 600 type="application/pdf" download="original.pdf"></iframe>'
-                # pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width = "100%" height = 600 type="application/pdf" download="Modified.pdf"></iframe>'
-            
-                # original_display = f"""
-                # <div class="pdf-container" style="height: 500px;">
-                #     <embed class="pdfobject" type="application/pdf" title="Modified PDF" src="data:application/pdf;base64,{original_pdf}" style="width: 100%; height: 100%;">
-                # </div>
-                # """
-                # pdf_display = f"""
-                # <div class="pdf-container" style="height: 500px;">
-                #     <embed class="pdfobject" type="application/pdf" title="Modified PDF" src="data:application/pdf;base64,{base64_pdf}" style="width: 100%; height: 100%;">
-                # </div>
-                # """
-
-
-
                 # Compare Logic implementation
                 if compare:
                     col1, col2, col3 = st.columns(3)
@@ -193,33 +147,36 @@ with tab2:
                     
                     # Side by Side comparison
                     col1, col2 = st.columns([1, 1], gap="small")
+                    
                     with col1:
                         with st.container(border = True):
                             st.markdown("<div style='text-align: center;'><strong>Original PDF viewer</strong></div>", unsafe_allow_html=True)
-                            st.markdown(original_display, unsafe_allow_html=True)
+                            pdf_viewer(text_to_pdf(text), width = 600)
                     with col2:
                         with st.container(border = True):
                             st.markdown("<div style='text-align: center;'><strong>Fastified PDF viewer</strong></div>", unsafe_allow_html=True)
-                            st.markdown(pdf_display, unsafe_allow_html=True)
-                    # Browser Cache Note
-                    st.markdown(f"""
-                        <div style='background-color: #FFD580; border-radius: 5px;'>
-                            <p style='color: black;'><strong>Note</strong> - {note_text}</p>
-                        </div>
-                        """, unsafe_allow_html=True)
+                            # st.markdown(pdf_display, unsafe_allow_html=True)
+                            pdf_viewer(text_to_pdf_fastify(text), width = 600)
+                    # # Browser Cache Note
+                    # st.markdown(f"""
+                    #     <div style='background-color: #FFD580; border-radius: 5px;'>
+                    #         <p style='color: black;'><strong>Note</strong> - {note_text}</p>
+                    #     </div>
+                    #     """, unsafe_allow_html=True)
                 else:
                     # No Comparison
                     col1, col2, col3 = st.columns(3)
-                    with col2:
+                    with col1:
                         st.download_button(label="Download Fastified PDF", data=text_to_pdf_fastify(text), file_name='output.pdf', mime='application/pdf')
-                    with st.container(border = True):
-                        st.markdown("<div style='text-align: center;'><strong>Fastified PDF viewer</strong></div>", unsafe_allow_html=True)
-                        st.markdown(pdf_display, unsafe_allow_html=True)
                     
-                    # Browser Cache Note
-                    st.markdown(f"""
-                        <div style='background-color: #FFD580; border-radius: 5px;'>
-                            <p style='color: black;'><strong>Note</strong> - {note_text}</p>
-                        </div>
-                        """, unsafe_allow_html=True)
+                    with st.container():
+                        st.markdown("<div><strong>Fastified PDF viewer</strong></div>", unsafe_allow_html=True)
+                        pdf_viewer(text_to_pdf_fastify(text), width = 500)
+                    
+                    # # Browser Cache Note
+                    # st.markdown(f"""
+                    #     <div style='background-color: #FFD580; border-radius: 5px;'>
+                    #         <p style='color: black;'><strong>Note</strong> - {note_text}</p>
+                    #     </div>
+                    #     """, unsafe_allow_html=True)
                     

requirements.txt

@@ -1,2 +1,3 @@
 PyPDF2==3.0.1
 fpdf2==2.7.8
+streamlit-pdf-viewer==0.0.7