Update app.py

app.py

@@ -31,6 +31,11 @@ import requests
 import base64
 import os
 import re
+import fitz  # PyMuPDF
+
+headers = {
+    "Authorization": "Bearer eRAG-PDPA-App"
+}
 
 st.title('🤖 eRAG-PDPA-v1')
 
@@ -45,9 +50,11 @@ st.markdown("""
 
 system_prompt = "คุณเป็นผู้ช่วยที่มีความรู้ด้านกฎหมาย พระราชบัญญัติคุ้มครองข้อมูลส่วนบุคคล (PDPA) และสามารถให้คำตอบที่เกี่ยวข้องเฉพาะตาม context ที่ได้รับ"
 
+# Function to clean text for search
 def clean_text_for_search(text):
     return ' '.join(re.sub(r'P-\d+\s*$', '', text, flags=re.MULTILINE).split())
 
+# Function to create highlighted PDF
 def create_highlighted_pdf(pdf_path, search_text, page_number):
     try:
         search_text = clean_text_for_search(search_text)
@@ -60,7 +67,7 @@ def create_highlighted_pdf(pdf_path, search_text, page_number):
                 text_instances = page.search_for(word)
                 for inst in text_instances:
                     highlight = page.add_highlight_annot(inst)
-                    highlight.set_colors(stroke=(1, 1, 0))
+                    highlight.set_colors(stroke=(1, 1, 0))  # Highlight color (yellow)
                     highlight.update()
 
         new_doc = fitz.open()
@@ -75,6 +82,7 @@ def create_highlighted_pdf(pdf_path, search_text, page_number):
         st.error(f"Error in create_highlighted_pdf: {str(e)}")
         return None
 
+# Function to format file size
 def format_file_size(size_in_bytes):
     for unit in ['B', 'KB', 'MB', 'GB']:
         if size_in_bytes < 1024:
@@ -82,6 +90,7 @@ def format_file_size(size_in_bytes):
         size_in_bytes /= 1024
     return f"{size_in_bytes:.2f} GB"
 
+# Function to display search results
 def display_search_result(result, index):
     with st.expander(f"🔍 Search Result #{index + 1} (Score: {result['score']:.4f})"):
         st.markdown("#### 📄 Document Information")
@@ -100,7 +109,6 @@ def display_search_result(result, index):
             st.write(f"• Modified: {result['metadata']['last_modified_date']}")
 
         st.markdown("#### 📝 Content")
-        #st.markdown(f"```\n{result['text']}\n```")
         st.markdown("""
             <div style="background-color:#d3f9d8; padding: 10px;">
                 <pre>{}</pre>
@@ -135,6 +143,7 @@ def display_search_result(result, index):
         except Exception as e:
             st.error(f"Error displaying PDF: {str(e)}")
 
+# Custom styles for Streamlit inputs
 st.markdown("""
     <style>
         .stTextInput>div>div>input {
@@ -152,19 +161,17 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# ฟอร์มสำหรับกรอกข้อมูล
+# Form for user input
 with st.form(key="input_form"):
     user_input = st.text_input("ป้อนคำ ข้อความ หรือประโยคที่ต้องการค้นหา:", value="ข้อมูลส่วนบุคคล คืออะไร มีกี่ประเภท", key="input")
     submit_button = st.form_submit_button("Send")
 
-    # แสดงผลเมื่อปุ่ม 'Send' ถูกคลิก
     if submit_button:
         st.write(f"คุณป้อนข้อความ: {user_input}")
-    
 
 if submit_button and user_input:
     try:
-        response = requests.post("http://113.53.253.50:8002/search", json={"query": user_input})
+        response = requests.post("http://113.53.253.50:8002/search", json={"query": user_input}, headers=headers)
         response.raise_for_status()
         data = response.json()
         search_results = data["results"]
@@ -176,7 +183,7 @@ if submit_button and user_input:
     except requests.RequestException as e:
         st.error(f"Error: {str(e)}")
 
-
+# Document repository section
 st.subheader("📄 ไฟล์เอกสารที่ใช้เป็นคลังข้อมูล PDPA")
 st.markdown(
     """