checkbox added

app.py

@@ -61,6 +61,9 @@ with st.sidebar:
     # Input filter
     top_n = st.number_input("Insert a number (top n rows to be selected):")
 
+    # Use dictionary
+    use_dict_format = st.checkbox("Use dictionary format.")
+
     # Clear button
     clear_button = st.sidebar.button("Clear Conversation", key="clear")
 
@@ -126,6 +129,9 @@ elif uploaded_files:
 
         # Display assistant response in chat message container
         with st.chat_message("assistant"):
-            st.table(result)
+            if not use_dict_format:
+                st.table(result)
+            else:
+                st.write(result.to_json())
         # Add assistant response to chat history
         st.session_state.messages.append({"role": "assistant", "content": result})