Update app.py

app.py

@@ -28,7 +28,6 @@ [email protected]
 
 import streamlit as st
 import requests
-import base64
 import os
 import re
 
@@ -40,10 +39,10 @@ headers = {
     "Authorization": f"Bearer {access_token}"  # Using f-string
 }
 
-st.title('🤖 eRAG-PDPA-v1')
+st.title('eRAG-PDPA-v1')
 
 st.markdown("""
-    ##### AI เพื่อการเรียนรู้พระราชบัญญัติคุ้มครองข้อมูลส่วนบุคคล (PDPA) 
+    ##### 🤖 AI เพื่อการเรียนรู้พระราชบัญญัติคุ้มครองข้อมูลส่วนบุคคล (PDPA) 
     ได้รับการออกแบบมาเพื่อช่วยให้ผู้ใช้งานเข้าถึงข้อมูลสำคัญได้อย่างรวดเร็วและเข้าใจง่าย 
     โดยระบบสามารถค้นหาและวิเคราะห์ความหมายในเชิงบริบทได้อย่างลึกซึ้ง 
     ทำให้สามารถเข้าใจความเชื่อมโยงและความหมายของข้อมูลได้อย่างแม่นยำ 
@@ -53,38 +52,6 @@ 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)
-        doc = fitz.open(pdf_path)
-        page = doc[int(page_number) - 1]
-        words = [word for word in search_text.split() if word]
-
-        for word in words:
-            if len(word) > 3:
-                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 color (yellow)
-                    highlight.update()
-
-        new_doc = fitz.open()
-        new_doc.insert_pdf(doc, from_page=int(page_number) - 1, to_page=int(page_number) - 1)
-        pdf_bytes = new_doc.write()
-
-        doc.close()
-        new_doc.close()
-
-        return pdf_bytes
-    except Exception as e:
-        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']: