Update app.py

app.py

@@ -110,14 +110,16 @@ def main():
             unique_sentences = set(response_sentences)
             filtered_response = '. '.join(unique_sentences)
                 
-            # Check if the response ends with a sentence-ending punctuation
-            while not response.strip().endswith(('.', '!', '?')) and max_tokens < MAX_TOKEN_LIMIT:
+            # Check if the filtered response ends with a sentence-ending punctuation
+            while not filtered_response.strip().endswith(('.', '!', '?')) and max_tokens < MAX_TOKEN_LIMIT:
                 max_tokens += 50  # Increase the max_tokens limit
                 chain = load_chatbot(max_tokens=max_tokens)
                 additional_response = chain.run(input_documents=docs, question=query)
-                response += additional_response  # Append the additional response to the original response
+                filtered_response += additional_response  # Append the additional response to the filtered response
+
+            st.session_state['chat_history'].append(("Bot", filtered_response, "new"))
+
 
-            st.session_state['chat_history'].append(("Bot", response, "new"))
 
             # Display new messages at the bottom
             new_messages = st.session_state['chat_history'][-2:]