Spaces:
Sleeping
Sleeping
| from pymongo import MongoClient | |
| import os | |
| password = os.environ.get('password') | |
| uri = f"mongodb+srv://cidar:{password}@cluster0.yuhfxcj.mongodb.net/?retryWrites=true&w=majority" | |
| # Create a new client and connect to the server | |
| client = MongoClient(uri) | |
| db = client['testDB'] | |
| collection = db['testCol'] | |
| def save_response(response_data): | |
| """Saves a response to the MongoDB collection.""" | |
| try: | |
| # Insert the response data into the collection | |
| print("right before insertion") | |
| collection.insert_one(response_data) | |
| print("right after insertion") | |
| except Exception as e: | |
| print(f"An error occurred: {e}") | |
| exit(0) | |
| def read_responses(filter_query=None): | |
| """Reads responses from the MongoDB collection based on an optional filter.""" | |
| try: | |
| if filter_query is None: | |
| filter_query = {} # An empty query will return all documents | |
| ranks = { | |
| 'CIDAR': [0, 0, 0], | |
| 'ALPAGASUS': [0, 0, 0], | |
| 'CHAT': [0, 0, 0], | |
| } | |
| for doc in collection.find(filter_query): | |
| for model in ranks: | |
| ranks[model][doc[model] - 1] += 1 | |
| return ranks | |
| except Exception as e: | |
| print(f"An error occurred: {e}") | |
| return [] |