Chatbot development using transformers library

I am developing a chatbot where upon the users question ,i need to go to mongodb database and retrieve the answer for his question.
what i have done till now:
i tried reading the collection into csv file and then answer based on users question in chatbot.It is working partially
example:when i ask the employees with skill as python,it is returning only the first match.Also when i ask about the count of employees with skill as python it is giving not giving count

attached is my code and csv data:

import csv

from pandas import read_csv
from pymongo import MongoClient
from transformers import pipeline

# Function to read data from MongoDB and store it in a CSV file in a table format
def mongo_to_csv(mongo_uri, database_name, collection_name, csv_file_path):
    # Connect to MongoDB
    client = MongoClient(mongo_uri)
    db = client[database_name]
    collection = db[collection_name]

    # Query the collection (you can modify the query as needed)
    data = collection.find()

    # Extract all field names from the documents to handle varying fields
    all_keys = set()
    for document in data:

    # Re-query the collection since the previous iteration exhausted the cursor
    data = collection.find()

    # Open a CSV file to write
    with open(csv_file_path, mode='w', newline='', encoding='utf-8') as csv_file:
        # Create a CSV DictWriter
        csv_writer = csv.DictWriter(csv_file, fieldnames=all_keys)
        # Write the header (field names)

        # Write the data rows
        for document in data:

    print(f"Data from {collection_name} collection in {database_name} database has been written to {csv_file_path}")
    import pandas as pd



    query="what is  name of all employees with  skills as Python"
  "_id": {
    "$oid": "66882351d969e86b0b22d603"
  "id": "1",
  "name": "abc",
 "skills": "Python",

  "_id": {
    "$oid": "66882352d969e86b0b22d603"
  "id": "2",
  "name": "ert",
 "skills": "python",


is there any better way of solving this or can i fix the existing code