display pdf

app.py

@@ -9,6 +9,7 @@ from transformers import pipeline
 import torch
 import os
 import presentation as customClass
+from streamlit_pdf_viewer import pdf_viewer
 
 
 api_key = os.getenv("API_KEY")
@@ -81,7 +82,7 @@ def extract_data(feed):
         pages = pdf.pages
         for p in pages:
             data.append(p.extract_text())
-        print(data)
+        
     return None 
 
 
@@ -94,6 +95,10 @@ def main():
     uploaded_file = st.file_uploader('Choose your .pdf file', type="pdf")
     if uploaded_file is not None:
         extract_data(uploaded_file)
+        
+        binary_data = uploaded_file.getvalue()
+        pdf_viewer(input=binary_data,
+                width=700)
     if st.button("Make PPT"):
         ai_ppt(data)
 

requirements.txt

@@ -6,4 +6,5 @@ pydantic_ai
 nest_asyncio
 torch==2.2.2
 transformers==4.39.1
+streamlit_pdf_viewer
 #pydantic_ai.models.groq