Update app.py

app.py

@@ -28,16 +28,6 @@ base_model = AutoModelForSeq2SeqLM.from_pretrained(
     torch_dtype=torch.float32
 )
 
-
-# checkpoint = "LaMini-T5-738M"
-# tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-# base_model = AutoModelForSeq2SeqLM.from_pretrained(
-#     checkpoint,
-#     device_map="auto",
-#     torch_dtype = torch.float32,
-#     from_tf=True
-# )
-
 persist_directory = "db"
 
 @st.cache_resource
@@ -100,28 +90,28 @@ def get_file_size(file):
     file.seek(0)
     return file_size
 
-# @st.cache_data
+@st.cache_data
 #function to display the PDF of a given file 
-# 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')
 
-#     # Embedding PDF in HTML
-#     pdf_display = F'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="600" type="application/pdf"></iframe>'
+    # Embedding PDF in HTML
+    pdf_display = F'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="600" type="application/pdf"></iframe>'
 
-#     # Displaying File
-#     st.markdown(pdf_display, unsafe_allow_html=True)
+    # Displaying File
+    st.markdown(pdf_display, unsafe_allow_html=True)
 
 # Display conversation history using Streamlit messages
 def display_conversation(history):
-    for i in range(len(_history["generated"])):
-        st.message(_history["past"][i], is_user=True, key=str(i) + "_user")
-        st.message(_history["generated"][i], key=str(i))
+    for i in range(len(history["generated"])):
+        message(history["past"][i], is_user=True, key=str(i) + "_user")
+        message(history["generated"][i],key=str(i))
 
 def main():
     st.markdown("<h1 style='text-align: center; color: blue;'>Chat with your PDF 🦜📄 </h1>", unsafe_allow_html=True)
-    st.markdown("<h3 style='text-align: center; color: grey;'>Built by <a href='https://github.com/manas95826'>The Manas with ❤️ </a></h3>", unsafe_allow_html=True)
+    st.markdown("<h3 style='text-align: center; color: grey;'>Built by <a href='https://github.com/AIAnytime'>AI Anytime with ❤️ </a></h3>", unsafe_allow_html=True)
 
     st.markdown("<h2 style='text-align: center; color:red;'>Upload your PDF 👇</h2>", unsafe_allow_html=True)
 
@@ -140,8 +130,8 @@ def main():
         with col1:
             st.markdown("<h4 style color:black;'>File details</h4>", unsafe_allow_html=True)
             st.json(file_details)
-            # st.markdown("<h4 style color:black;'>File preview</h4>", unsafe_allow_html=True)
-            # pdf_view = displayPDF(filepath)
+            st.markdown("<h4 style color:black;'>File preview</h4>", unsafe_allow_html=True)
+            pdf_view = displayPDF(filepath)
 
         with col2:
             with st.spinner('Embeddings are in process...'):
@@ -168,12 +158,7 @@ def main():
             # Display conversation history using Streamlit messages
             if st.session_state["generated"]:
                 display_conversation(st.session_state)
-        
-
-        
-
-
-
-
+                
 if __name__ == "__main__":
     main()
+