eRAG-PDPA-v1 / app.py
amornpan's picture
Update app.py
051469e verified
raw
history blame
9.68 kB
import streamlit as st
import requests
import base64
import os
import re
st.title("AI สนับสนุนความรู้ด้านกฏหมาย PDPA")
st.write("AI อัจฉริยะเพื่อการเรียนรู้พระราชบัญญัติคุ้มครองข้อมูลส่วนบุคคล (PDPA) ได้รับการออกแบบมาเพื่อช่วยให้ผู้ใช้งานเข้าถึงข้อมูลสำคัญได้อย่างรวดเร็วและเข้าใจง่าย โดยระบบสามารถค้นหาและวิเคราะห์ความหมายในเชิงบริบทได้อย่างลึกซึ้ง ทำให้สามารถเข้าใจความเชื่อมโยงและความหมายของข้อมูลได้อย่างแม่นยำ รองรับการเรียนรู้เชิงลึกในประเด็นเกี่ยวกับ PDPA และการคุ้มครองข้อมูลส่วนบุคคลได้อย่างมีประสิทธิภาพ [Powered by NT Lahnmah]")
st.subheader("🏷️ Tags")
st.markdown("""
- LLM   RAG   eRAG   PDPA   Machine Learning   AI   Data Privacy   GenAI   NT GenAI   ntgenai   lahnmah   NT Thai GPT   ntthaigpt   medical   medtech   HealthGPT   หลานม่า   NT Academy
""", unsafe_allow_html=True)
system_prompt = "คุณเป็นผู้ช่วยที่มีความรู้ด้านกฎหมาย พระราชบัญญัติคุ้มครองข้อมูลส่วนบุคคล (PDPA) และสามารถให้คำตอบที่เกี่ยวข้องเฉพาะตาม context ที่ได้รับ"
def clean_text_for_search(text):
return ' '.join(re.sub(r'P-\d+\s*$', '', text, flags=re.MULTILINE).split())
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.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
def format_file_size(size_in_bytes):
for unit in ['B', 'KB', 'MB', 'GB']:
if size_in_bytes < 1024:
return f"{size_in_bytes:.2f} {unit}"
size_in_bytes /= 1024
return f"{size_in_bytes:.2f} GB"
def display_search_result(result, index):
with st.expander(f"🔍 Search Result #{index + 1} (Score: {result['score']:.4f})"):
st.markdown("#### 📄 Document Information")
col1, col2 = st.columns(2)
with col1:
st.markdown("**File Details:**")
st.write(f"• File Name: {result['metadata']['file_name']}")
st.write(f"• Page: {result['metadata']['page_label']}")
st.write(f"• Type: {result['metadata']['file_type']}")
st.write(f"• Size: {format_file_size(result['metadata']['file_size'])}")
with col2:
st.markdown("**Dates:**")
st.write(f"• Created: {result['metadata']['creation_date']}")
st.write(f"• Modified: {result['metadata']['last_modified_date']}")
st.markdown("#### 📝 Content")
st.markdown(f"```\n{result['text']}\n```")
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)}")
st.markdown("""
<style>
.stTextInput>div>div>input {
background-color: #d3f9d8; /* สีเขียวอ่อน */
color: black; /* สีข้อความเป็นสีดำ */
cursor: text; /* เคอร์เซอร์เป็นรูปเส้น */
caret-color: black; /* เปลี่ยนสีเคอร์เซอร์ให้เป็นสีดำ */
}
/* ปรับสีเคอร์เซอร์ให้เป็นสีดำเมื่อมีการคลิก (focus) */
.stTextInput>div>div>input:focus {
border-color: black; /* ขอบสีดำ */
outline-color: black; /* เส้นขอบสีดำ */
}
</style>
""", unsafe_allow_html=True)
# ฟอร์มสำหรับกรอกข้อมูล
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.raise_for_status()
data = response.json()
search_results = data["results"]
st.markdown("### 🔎 Search Results")
for idx, result in enumerate(search_results):
display_search_result(result, idx)
except requests.RequestException as e:
st.error(f"Error: {str(e)}")
st.subheader("📄 ไฟล์เอกสารที่ใช้เป็นคลังข้อมูล PDPA")
st.markdown(
"""
- [FAQ-เกี่ยวกับการคุ้มครองข้อมูลส่วนบุคคล-PDPA.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/1.FAQ-%E0%B9%80%E0%B8%81%E0%B8%B5%E0%B9%88%E0%B8%A2%E0%B8%A7%E0%B8%81%E0%B8%B1%E0%B8%9A%E0%B8%81%E0%B8%B2%E0%B8%A3%E0%B8%84%E0%B8%B8%E0%B9%89%E0%B8%A1%E0%B8%84%E0%B8%A3%E0%B8%AD%E0%B8%87%E0%B8%82%E0%B9%89%E0%B8%AD%E0%B8%A1%E0%B8%B9%E0%B8%A5%E0%B8%AA%E0%B9%88%E0%B8%A7%E0%B8%99%E0%B8%9A%E0%B8%B8%E0%B8%84%E0%B8%84%E0%B8%A5-PDPA.pdf) (1.13 MB)
- [1612025563SummaryPDPA_DigitalCouncilofThailand.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/1612025563SummaryPDPA_DigitalCouncilofThailand.pdf) (1.35 MB)
- [PDPA_Guideline_v_1.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/PDPA_Guideline_v_1.pdf) (10.7 MB)
- [T_0024.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/T_0024.pdf) (81.1 kB)
- [T_0026.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/T_0026.pdf) (82.6 kB)
- [T_0028.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/T_0028.pdf) (105 kB)
- [T_0032.pdf](https://huggingface.co/spaces/amornpan/eRAG-PDPA-v1/blob/main/%E0%B9%80%E0%B8%AD%E0%B8%81%E0%B8%AA%E0%B8%B2%E0%B8%A3_PDPA/T_0032.pdf) (103 kB)
""",
unsafe_allow_html=True
)
st.subheader("👤 Authors")
st.write("""
- Amornpan Phornchaicharoen ([email protected])
- Aekanun Thongtae ([email protected])
- Montita Somsoo ([email protected])
- Jiranuwat Songpad ([email protected])
- Phongsatorn Somjai ([email protected])
- Benchawan Wangphoomyai ([email protected])
""")