Update app.py

app.py

@@ -21,8 +21,9 @@ div.stButton > button:first-child {
 div.stButton > button:active {
     background-color: #ff6262;
 }
-   div[data-testid="stStatusWidget"] div button {display: none;
-}
+   div[data-testid="stStatusWidget"] div button {
+        display: none;
+        }
     
     .reportview-container {
             margin-top: -2em;
@@ -31,7 +32,8 @@ div.stButton > button:active {
         .stDeployButton {display:none;}
         footer {visibility: hidden;}
         #stDecoration {display:none;}
-    button[title="View fullscreen"]{visibility: hidden;}
+    button[title="View fullscreen"]{
+    visibility: hidden;}
         </style>
 """,
     unsafe_allow_html=True,
@@ -47,14 +49,16 @@ if "messages" not in st.session_state:
 if "memory" not in st.session_state:
     st.session_state.memory = ConversationBufferWindowMemory(k=2, memory_key="chat_history",return_messages=True) 
 
-embedings = HuggingFaceEmbeddings(model_name="nomic-ai/nomic-embed-text-v1",model_kwargs={"trust_remote_code":True,"revision":"289f532e14dbbbd5a04753fa58739e9ba766f3c7"})
-db = FAISS.load_local("ipc_vector_db", embedings)
+embeddings = HuggingFaceEmbeddings(model_name="nomic-ai/nomic-embed-text-v1",model_kwargs={"trust_remote_code":True,"revision":"289f532e14dbbbd5a04753fa58739e9ba766f3c7"})
+db = FAISS.load_local("ipc_vector_db", embeddings, allow_dangerous_deserialization=True)
 db_retriever = db.as_retriever(search_type="similarity",search_kwargs={"k": 4})
 
 prompt_template = """<s>[INST]This is a chat template and As a legal chat bot specializing in Indian Penal Code queries, your primary objective is to provide accurate and concise information based on the user's questions. Do not generate your own questions and answers. You will adhere strictly to the instructions provided, offering relevant context from the knowledge base while avoiding unnecessary details. Your responses will be brief, to the point, and in compliance with the established format. If a question falls outside the given context, you will refrain from utilizing the chat history and instead rely on your own knowledge base to generate an appropriate response. You will prioritize the user's query and refrain from posing additional questions. The aim is to deliver professional, precise, and contextually relevant information pertaining to the Indian Penal Code.
+
 CONTEXT: {context}
 CHAT HISTORY: {chat_history}
 QUESTION: {question}
+
 ANSWER:
 </s>[INST]
 """
@@ -62,8 +66,7 @@ ANSWER:
 prompt = PromptTemplate(template=prompt_template,
                         input_variables=['context', 'question', 'chat_history'])
 
-# You can also use other LLMs options from https://python.langchain.com/docs/integrations/llms. Here I have used TogetherAI API
-TOGETHER_AI_API= os.environ['TOGETHER_AI'] = '2a7c5dcdbb1049a39117ac0865c4d04008d49db31aa85a3258603817af16dbd0'
+TOGETHER_AI_API= os.environ['TOGETHER_AI']
 llm = Together(
     model="mistralai/Mistral-7B-Instruct-v0.2",
     temperature=0.5,
@@ -96,12 +99,12 @@ if input_prompt:
 
             message_placeholder = st.empty()
 
-            full_response ="Answer\n\n"
+            full_response = "⚠️ **_Note: Goodachari 511 is a legal assistance tool for general guidance._** \n\n\n"
         for chunk in result["answer"]:
             full_response+=chunk
             time.sleep(0.02)
             
-            message_placeholder.markdown(full_response)
+            message_placeholder.markdown(full_response+" ▌")
         st.button('Reset All Chat 🗑️', on_click=reset_conversation)
 
     st.session_state.messages.append({"role":"assistant","content":result["answer"]})