Update app.py

app.py

@@ -75,16 +75,15 @@ def process_documents_wrapper(inputs):
         for snip in snippets
     ]
     response = f"Uploaded and processed documents {inputs}"
-    st.session_state.messages.append((f"/add-papers {inputs}", response, "identity"))
     st.session_state.documents = documents
-    return (response, "identity")
+    return index_documents_wrapper(None, f"/add-papers {inputs}")
 
 
-def index_documents_wrapper(inputs=None):
+def index_documents_wrapper(inputs=None, arg='/library'):
     response = pd.DataFrame(
         st.session_state.index, columns=["id", "reference", "tokens"]
     )
-    st.session_state.messages.append(("/library", response, "dataframe"))
+    st.session_state.messages.append((arg, response, "dataframe"))
     return (response, "dataframe")