app updated

app.py

@@ -66,11 +66,9 @@ with st.sidebar:
         "Insert a number (top n rows to be selected):", value=5, step=1
     )
 
-    # Use dictionary
-    use_dict_format = st.checkbox("Use dictionary format.", value=True)
-
     # Clear button
     clear_button = st.sidebar.button("Clear Conversation", key="clear")
+
     # Credit
     current_year = current_year()  # This will print the current year
     st.markdown(
@@ -117,7 +115,7 @@ elif uploaded_files:
         refs_tab = query_search(
             "pful for understanding federal income", documents, query_database, sources
         )
-        refs_tab.head(math.ceil(top_n))
+        refs_tab = refs_tab.head(math.ceil(top_n))
 
         # React to user input
         if prompt := st.chat_input("What is up?"):
@@ -130,9 +128,6 @@ elif uploaded_files:
 
             # Display assistant response in chat message container
             with st.chat_message("assistant"):
-                if not use_dict_format:
-                    st.table(result)
-                else:
-                    st.write(result.to_json())
+                st.table(result)
             # Add assistant response to chat history
             st.session_state.messages.append({"role": "assistant", "content": result})