utils.py

import os
import json
import itertools
from groq import Groq
from dotenv import load_dotenv
from pymongo import MongoClient
import random
import certifi
load_dotenv()


def get_random():
    lst = ["R1","R2","R3","S1","RD1"]
    i = random.randint(0,4)
    return lst[i]



def get_mongo_collection():
    CONNECTION_STRING = os.getenv("CONNECTION_STRING")
    DB_NAME = os.getenv("DB_NAME")
    COLLECTION_NAME = os.getenv("COLLECTION_NAME")
    try:
        # Connect with certifi to avoid SSL errors
        client = MongoClient(CONNECTION_STRING, tlsCAFile=certifi.where())
        db = client[DB_NAME]
        return db[COLLECTION_NAME]
    except Exception as e:
        print(f"Error connecting to Mongo: {e}")
        return None