global streamlit config

app.py

@@ -10,6 +10,21 @@ from langchain.llms import HuggingFaceHub
 import os
 from dotenv import load_dotenv
 
+# Set the Streamlit page configuration and CSS styles
+st.set_page_config(page_title="PDF Buddy", page_icon=":coffee:")
+st.markdown(
+    """
+    <style>
+    body {
+        background-color: #fce6ef;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+st.write(css, unsafe_allow_html=True)
+st.header("PDF Buddy :coffee:")
+
 
 def get_pdf_text(pdf_docs):
     text = ""
@@ -40,6 +55,7 @@ def load_embeddings():
 
 embeddings = load_embeddings()
 
+
 def get_vectorstore(text_chunks):
     vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
     return vectorstore
@@ -49,6 +65,7 @@ def load_llm():
     llm = HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":0.5, "max_length":218})
     return llm
 
+# Load the model and store it as a global variable
 llm = load_llm()
 
 def get_conversation_chain(vectorstore):
@@ -77,25 +94,12 @@ def handle_userinput(user_question):
 
 def main():
     load_dotenv()
-    st.set_page_config(page_title="PDF Buddy", page_icon=":coffee:")
-    st.markdown(
-        """
-        <style>
-        body {
-            background-color: #fce6ef;
-        }
-        </style>
-        """,
-        unsafe_allow_html=True
-    )
-    st.write(css, unsafe_allow_html=True)
 
     if "conversation" not in st.session_state:
         st.session_state.conversation = None
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = None
 
-    st.header("PDF Buddy :coffee:")
     user_question = st.text_input("Ask a question about your documents:")
     if user_question:
         handle_userinput(user_question)