Update app.py

app.py

@@ -13,12 +13,6 @@ from footer import footer
 # Set the Streamlit page configuration and theme
 st.set_page_config(page_title="In-Legal-IPC", layout="centered")
 
-# Display the logo image
-#col1, col2, col3 = st.columns([1, 30, 1])
-#with col2:
-  #  st.image("https://raw.githubusercontent.com/shiv4321/not/refs/heads/main/Banner.png", use_column_width=True)
-
-
 # Display the logo image
 col1, col2, col3 = st.columns([1, 30, 1])
 with col2:
@@ -39,7 +33,7 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
 
 if "memory" not in st.session_state:
-    st.session_state.memory = ConversationBufferWindowMemory(k=2, memory_key="chat_history", return_messages=True)
+    st.session_state.memory = ConversationBufferWindowMemory(k=5, memory_key="chat_history", return_messages=True)
 
 @st.cache_resource
 def load_embeddings():
@@ -59,7 +53,6 @@ As a legal chatbot specializing in the Indian Penal Code, you are tasked with pr
 - Limit responses to essential information that directly addresses the user's question, providing concise yet comprehensive explanations.
 - Avoid assuming specific contexts or details not provided in the query, focusing on delivering universally applicable legal interpretations unless otherwise specified.
 - Conclude with a brief summary that captures the essence of the legal discussion and corrects any common misinterpretations related to the topic.
-
 CONTEXT: {context}
 CHAT HISTORY: {chat_history}
 QUESTION: {question}
@@ -72,8 +65,6 @@ ANSWER:
 </s>[INST]
 """
 
-
-
 prompt = PromptTemplate(template=prompt_template,
                         input_variables=['context', 'question', 'chat_history'])
 
@@ -95,11 +86,18 @@ def reset_conversation():
     st.session_state.messages = []
     st.session_state.memory.clear()
 
+# Add the PDF button just above the chat input
+st.markdown(
+    "[📄 PDF](https://drive.google.com/file/d/1Xkq64r4Id8qSyb5woVzdvArhzUxvKJk8/view?usp=drive_link)",
+    unsafe_allow_html=True
+)
+
+# Display previous messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.write(message["content"])
 
-
+# Chat input area
 input_prompt = st.chat_input("Say something...")
 if input_prompt:
     with st.chat_message("user"):
@@ -125,9 +123,5 @@ if input_prompt:
         if st.button('🗑️ Reset All Chat', on_click=reset_conversation):
             st.experimental_rerun()
 
-
-
 # Define the CSS to style the footer
-footer()
-
-
+footer()