Update app.py

app.py

@@ -21,7 +21,6 @@ os.environ["LANGCHAIN_API_KEY"] = os.getenv('LANGSMITH_API')
 
 UPLOAD_DIR = "uploaded_files"
 
-
 def cleanup_files():
     if os.path.isdir(UPLOAD_DIR):
         shutil.rmtree(UPLOAD_DIR, ignore_errors=True)
@@ -123,7 +122,7 @@ if uploaded_file is not None:
         embedding
     )
 
-    template = '''You are WattBot, Wattpad's friendly chatbot assistant. Your role is to assist readers and writers with insightful answers about stories, creative writing, and using Wattpad. Answer the {question} based only on the provided {context}. After answering the question, recommend Wattpad stories that may interest the user based on the content of the PDF or the question. Be friendly, creative, and concise. Use a maximum of three sentences for the answer, and add one or two relevant story recommendations with a brief description and a link. If you're unsure about the answer, respond with "I'm not sure about that, but feel free to explore more on Wattpad!"'''
+    template = '''You are WattBot, Wattpad's friendly chatbot assistant, designed to help readers and writers with insightful answers about stories, writing tips, and the Wattpad platform. Please answer the {question} based only on the given {context}. If the question is unrelated to the context or beyond your knowledge, respond with "I'm not sure about that, but feel free to explore more on Wattpad!" Keep your responses concise, using a maximum of three sentences.'''
 
     prompt = ChatPromptTemplate.from_template(template)
     retriever = vector_base.as_retriever()
@@ -160,5 +159,11 @@ if uploaded_file is not None:
                 unsafe_allow_html=True
             )
 
+# Reset functionality
+if st.button("Reset and Upload a New PDF"):
+    st.session_state.clear()
+    st.session_state.cleanup_done = False
+    st.experimental_rerun()
+
 if st.session_state.cleanup_done:
     cleanup_files()