Update app.py

app.py

@@ -1,7 +1,5 @@
 import streamlit as st
 import PyPDF2
-from streamlit.runtime.scriptrunner import get_script_run_ctx
-from streamlit.source_util import get_pages
 import io
 from typing import List, Dict
 from groq import Groq
@@ -13,15 +11,6 @@ st.set_page_config(
     layout="wide"
 )
 
-# Hide Streamlit menu and footer
-hide_menu_style = """
-        <style>
-        #MainMenu {visibility: hidden;}
-        footer {visibility: hidden;}
-        </style>
-        """
-st.markdown(hide_menu_style, unsafe_allow_html=True)
-
 class PDFChatAssistant:
     def __init__(self):
         # Initialize session state variables
@@ -39,7 +28,6 @@ class PDFChatAssistant:
         try:
             pdf_reader = PyPDF2.PdfReader(pdf_file)
             text = ""
-            show_full_text()
             for page in pdf_reader.pages:
                 text += page.extract_text() + "\n\n"
             return text
@@ -51,7 +39,7 @@ class PDFChatAssistant:
         """
         Create Groq client with API key
         """
-        client = Groq(api_key=st.secrets["GROQ_API_KEY_DOCUMENTOR"])
+        client = Groq(api_key=st.secrets["GROQ_API_KEY"])
         return client
 
     def generate_ai_response(self, client: Groq, context: str, user_query: str) -> str:
@@ -63,7 +51,7 @@ class PDFChatAssistant:
                 messages=[
                     {
                         "role": "system", 
-                        "content": f"You are a helpful PDF assistant. Use the following PDF context to answer questions:\n\n{context}"
+                        "content": f"Bạn là một trợ lý PDF hữu ích. Luôn trả lời bằng tiếng Việt. Sử dụng nội dung trong file PDF đính kèm để trả lời câu hỏi sau:\n\n{context}"
                     },
                     {
                         "role": "user", 
@@ -109,37 +97,7 @@ class PDFChatAssistant:
                             "role": "user", 
                             "content": "Extract the most important key points from this PDF"
                         })
-    def show_full_text():
-        st.title("📄 Full PDF Text")
-        
-        # Try to retrieve the PDF text from the main page's session state
-        ctx = get_script_run_ctx()
-        session_info = get_pages('app.py')
-        
-        if 'pdf_text' in st.session_state and st.session_state.pdf_text:
-            st.text_area(
-                "Complete Extracted Text", 
-                value=st.session_state.pdf_text, 
-                height=600,
-                disabled=True
-            )
-            
-            # Additional buttons
-            col1, col2 = st.columns(2)
-            with col1:
-                if st.button("Copy to Clipboard"):
-                    st.code(st.session_state.pdf_text)
-                    st.success("Text copied to clipboard!")
-            
-            with col2:
-                if st.button("Back to Chat"):
-                    st.switch_page("app.py")
-        else:
-            st.warning("No PDF text available. Please upload a PDF first.")
-            if st.button("Back to Chat"):
-                st.switch_page("app.py")
 
-        
         # Main chat area
         st.header("Chat with Your PDF")