Spaces:
Runtime error
Runtime error
File size: 2,070 Bytes
ed0b977 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
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()
|