Update app.py

app.py

@@ -2,49 +2,41 @@ import streamlit as st
 import requests
 from datetime import datetime
 import base64
-import fitz  # PyMuPDF
 import os
 import re
 
-# ตรวจสอบและกำหนดค่าเริ่มต้นให้ st.session_state.dialog_done
 if "dialog_done" not in st.session_state:
-    st.session_state.dialog_done = False  # ค่าเริ่มต้นเป็น False
+    st.session_state.dialog_done = False
 
-# ถ้า dialog_done เป็น False ให้แสดงหน้าคำถาม
 if not st.session_state.dialog_done:
     st.title("เริ่มต้นคำถาม")
     st.write("คำถาม: ใครคือคนที่หน้าตาดีที่สุด?")
     
     answer = st.text_input("กรุณาตอบคำถามนี้:")
-    if st.button("ส่งคำตอบ"):  # ปุ่มส่งคำตอบ
+    if st.button("ส่งคำตอบ"):
         if answer == "พี่ก้องคนหล่อ":
-            st.session_state.dialog_done = True  # อัปเดตสถานะเป็น True
+            st.session_state.dialog_done = True
             st.success("คำตอบถูกต้อง! กดปุ่ม 'เข้าสู่หน้าหลัก' เพื่อดำเนินการต่อ")
         else:
             st.error("คำตอบไม่ถูกต้อง กรุณาลองใหม่อีกครั้ง.")
     
     if st.session_state.dialog_done:
-        # เพิ่มปุ่มเพื่อให้ผู้ใช้กดเข้าสู่หน้าหลัก
         if st.button("เข้าสู่หน้าหลัก"):
-            st.experimental_rerun()  # รีเฟรชเพื่อแสดงหน้าหลัก
+            st.experimental_rerun()
 
 else:
-    # เนื้อหาของหน้าหลัก
     st.title("AI สนับสนุนความรู้ด้าน PDPA")
     st.write("เราสอบถาม AI สืบค้น และสรุป")
 
-    # Define system prompt at the top level
-    system_prompt = """คุณเป็นผู้ช่วยที่มีความรู้ด้านกฎหมาย PDPA และสามารถให้คำตอบที่เกี่ยวข้องเฉพาะตาม context ที่ได้รับ"""
+    system_prompt = "คุณเป็นผู้ช่วยที่มีความรู้ด้านกฎหมาย PDPA และสามารถให้คำตอบที่เกี่ยวข้องเฉพาะตาม context ที่ได้รับ"
 
     def clean_text_for_search(text):
-        """Clean text for better search matching"""
         text = re.sub(r'P-\d+\s*$', '', text, flags=re.MULTILINE)
         text = re.sub(r'Confidential.*$', '', text, flags=re.MULTILINE)
         text = ' '.join(text.split())
         return text
 
     def create_highlighted_pdf(pdf_path, search_text, page_number):
-        """Create a highlighted version of the PDF page"""
         try:
             search_text = clean_text_for_search(search_text)
             doc = fitz.open(pdf_path)
@@ -73,7 +65,6 @@ else:
             return None
 
     def format_file_size(size_in_bytes):
-        """Convert bytes to human readable format"""
         for unit in ['B', 'KB', 'MB', 'GB']:
             if size_in_bytes < 1024:
                 return f"{size_in_bytes:.2f} {unit}"
@@ -81,7 +72,6 @@ else:
         return f"{size_in_bytes:.2f} GB"
 
     def display_search_result(result, index):
-        """Display a single search result with metadata in an expander"""
         with st.expander(f"🔍 Search Result #{index + 1} (Score: {result['score']:.4f})"):
             st.markdown("#### 📄 Document Information")
             col1, col2 = st.columns(2)
@@ -101,9 +91,6 @@ else:
             st.markdown("#### 📝 Content")
             st.markdown(f"```\n{result['text']}\n```")
 
-            st.markdown("#### 📂 File Location")
-            st.code(result['file_path'], language='plaintext')
-
             try:
                 pdf_path = result['file_path']
                 if os.path.exists(pdf_path):
@@ -129,8 +116,6 @@ else:
                         st.markdown(pdf_display, unsafe_allow_html=True)
                     else:
                         st.error("Failed to create highlighted PDF")
-                else:
-                    st.error("PDF file not found at the specified location.")
             except Exception as e:
                 st.error(f"Error displaying PDF: {str(e)}")
 
@@ -157,7 +142,7 @@ else:
                 for idx, result in enumerate(search_results):
                     display_search_result(result, idx)
 
-                response_text = "\n\n---\n\n".join([f"Text: {result['text']}\nFile Path: {result['file_path']}" for result in search_results])
+                response_text = "\n\n---\n\n".join([f"Text: {result['text']}" for result in search_results])
 
             except requests.RequestException as e:
                 st.error(f"Error: {str(e)}")