Spaces:
Runtime error
Runtime error
| import os | |
| from flask import Flask, session | |
| from flask_pymongo import PyMongo | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| app = Flask(__name__) | |
| app.config["MONGO_URI"] = os.environ.get("MONGO_URI") | |
| if app.config["MONGO_URI"] and "mongodb.net/?" in app.config["MONGO_URI"]: | |
| app.config['MONGO_URI'] = app.config['MONGO_URI'].replace("mongodb.net/?", "mongodb.net/swiftquery?") | |
| mongo = PyMongo(app) | |
| def debug_task(): | |
| with app.app_context(): | |
| # Simulate session user_id. We don't know it exactly, but let's try 1 as it's common or implied by the error 'KeyError: 1' if it was a lookup | |
| # Or maybe the error is looking up key 1 in a dict. | |
| # Let's search for any chat sessions | |
| print("Finding chat sessions...") | |
| all_chats = list(mongo.db.chatsession.find()) | |
| print(f"Found {len(all_chats)} chats.") | |
| for chat in all_chats: | |
| print(f"Processing chat: {chat.get('id')} / {chat.get('sql_id')}") | |
| # This is the logic we added | |
| if chat.get('doctor_id'): | |
| d_id = chat['doctor_id'] | |
| print(f" Doctor ID: {d_id} (Type: {type(d_id)})") | |
| # Check find_one | |
| try: | |
| query = {"$or": [{"sql_id": d_id}, {"id": d_id}]} | |
| print(f" Query: {query}") | |
| doc = mongo.db.doctor.find_one(query) | |
| print(f" Found Doc: {doc.get('name') if doc else 'None'}") | |
| if doc: | |
| print(f" Doc Name: {doc['name']}") | |
| chat['doctor_name'] = doc['name'] if doc else "Unknown" | |
| except Exception as e: | |
| print(f" ERROR processing chat {chat.get('id')}: {e}") | |
| import traceback | |
| traceback.print_exc() | |
| else: | |
| chat['doctor_name'] = "AI Assistant" | |
| print(" Success.") | |
| if __name__ == "__main__": | |
| debug_task() | |