Spaces:
Build error
Build error
File size: 1,114 Bytes
b438494 04eb7e1 | 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 | from pymongo import MongoClient
import os
def data_field(data_src):
mongodb_uri = os.getenv("MONGODB_URI")
myclient = MongoClient(mongodb_uri)
mydb = myclient.get_database()
mycol = mydb["df_data"]
x = mycol.find_one({"data_field": data_src})
x = x["result"]["question"]
#st.write(x)
return x
def get_analyst_response(data_src):
try:
mongodb_uri = os.getenv("MONGODB_URI")
myclient = MongoClient(mongodb_uri)
mydb = myclient.get_database()
mycol = mydb["df_response"]
x = mycol.find_one({"data_field": data_src},
sort=[('timestamp', -1)])
if x and "result" in x:
return x["result"]
else:
print(f"No matching document or 'result' field found for data_src: {data_src} in df_response")
return None # Return None if no doc or 'result' field found
finally:
if myclient:
myclient.close()
#
|