Update app.py

app.py

@@ -125,16 +125,6 @@ elif uploaded_files:
         # Call the function
         query_database = list_to_nums(documents)
 
-        # Create reference table
-        refs_tab = query_search(
-            "pful for understanding federal income",
-            documents,
-            query_database,
-            sources,
-            q_levels,
-        )
-        refs_tab = refs_tab.head(math.ceil(top_n))
-
         # React to user input
         if prompt := st.chat_input("What is up?"):
             # Display user message in chat message container
@@ -142,10 +132,20 @@ elif uploaded_files:
             # Add user message to chat history
             st.session_state.messages.append({"role": "user", "content": prompt})
 
+            # Create reference table
+            refs_tab = query_search(
+                prompt,
+                documents,
+                query_database,
+                sources,
+                q_levels,
+            )
+            refs_tab = refs_tab.head(math.ceil(top_n))
             result = refs_tab
 
             # Display assistant response in chat message container
             with st.chat_message("assistant"):
                 st.table(result)
+
             # Add assistant response to chat history
             st.session_state.messages.append({"role": "assistant", "content": result})