Update app.py

app.py

@@ -22,14 +22,6 @@ import requests
 import os
 import re
 
-# Access the variable 'eRAG_access_token'
-access_token = os.getenv('eRAG_access_token')
-
-# Correct way to use the access token value
-headers = {
-    "Authorization": f"Bearer {access_token}"  # Using f-string
-}
-
 st.title('eRAG-PDPA-v1')
 
 st.markdown("""
@@ -75,35 +67,13 @@ def display_search_result(result, index):
                 <pre>{}</pre>
             </div>
         """.format(result['text']), unsafe_allow_html=True)
+# Access the variable 'eRAG_access_token'
+access_token = os.getenv('eRAG_access_token')
 
-        try:
-            pdf_path = result['file_path']
-            if os.path.exists(pdf_path):
-                st.markdown("#### 📄 PDF Preview (with highlighted text)")
-
-                highlighted_pdf = create_highlighted_pdf(
-                    pdf_path,
-                    result['text'],
-                    result['metadata']['page_label']
-                )
-
-                if highlighted_pdf:
-                    base64_pdf = base64.b64encode(highlighted_pdf).decode('utf-8')
-                    pdf_display = f'''
-                        <iframe 
-                            src="data:application/pdf;base64,{base64_pdf}" 
-                            width="100%" 
-                            height="800px" 
-                            type="application/pdf"
-                            style="border: 1px solid #ccc; border-radius: 5px;"
-                        ></iframe>
-                    '''
-                    st.markdown(pdf_display, unsafe_allow_html=True)
-                else:
-                    st.error("Failed to create highlighted PDF")
-        except Exception as e:
-            st.error(f"Error displaying PDF: {str(e)}")
-
+# Correct way to use the access token value
+headers = {
+    "Authorization": f"Bearer {access_token}"  # Using f-string
+}
 # Custom styles for Streamlit inputs
 st.markdown("""
     <style>