Update src/database.py

src/database.py

@@ -11,11 +11,22 @@ MONGO_URI = os.getenv("MONGO_URI")
 DB_NAME = os.getenv("MONGO_DB")
 COLLECTION_NAME = os.getenv("MONGO_COLLECTION")
 
-# Initialize MongoDB client
-client = MongoClient(MONGO_URI)
-db = client[DB_NAME]
-collection = db[COLLECTION_NAME]
+# Global variables for lazy initialization
+_client = None
+_db = None
+_collection = None
 
+def get_database_connection():
+    """Initialize MongoDB connection lazily"""
+    global _client, _db, _collection
+    
+    if _client is None:
+        _client = MongoClient(MONGO_URI, serverSelectionTimeoutMS=5000)
+        _client.admin.command('ping')
+        _db = _client[DB_NAME]
+        _collection = _db[COLLECTION_NAME]
+    
+    return _client, _db, _collection
 
 def insert_analysis_result(
     video_name: str,
@@ -26,32 +37,45 @@ def insert_analysis_result(
     response: dict
 ):
     """Insert a new video analysis result"""
-    document = {
-        "video_name": video_name,
-        "offer_details": offer_details,
-        "target_audience": target_audience,
-        "specific_hook": specific_hook,
-        "additional_context": additional_context,
-        "response": response,
-        "created_at": datetime.utcnow()
-    }
-    result = collection.insert_one(document)
-    return str(result.inserted_id)
-
+    try:
+        client, db, collection = get_database_connection()
+        document = {
+            "video_name": video_name,
+            "offer_details": offer_details,
+            "target_audience": target_audience,
+            "specific_hook": specific_hook,
+            "additional_context": additional_context,
+            "response": response,
+            "created_at": datetime.utcnow()
+        }
+        result = collection.insert_one(document)
+        return str(result.inserted_id)
+    except:
+        raise
 
 def get_all_results(limit: int = 20):
     """Fetch all saved results, sorted by newest"""
-    return list(collection.find().sort("created_at", -1).limit(limit))
-
+    try:
+        client, db, collection = get_database_connection()
+        return list(collection.find().sort("created_at", -1).limit(limit))
+    except:
+        return []
 
 def get_result_by_id(doc_id):
     """Fetch a specific analysis by _id"""
-    from bson import ObjectId
-    return collection.find_one({"_id": ObjectId(doc_id)})
-
+    try:
+        from bson import ObjectId
+        client, db, collection = get_database_connection()
+        return collection.find_one({"_id": ObjectId(doc_id)})
+    except:
+        return None
 
 def delete_result_by_id(doc_id):
     """Delete a specific analysis by _id"""
-    from bson import ObjectId
-    result = collection.delete_one({"_id": ObjectId(doc_id)})
-    return result.deleted_count > 0
+    try:
+        from bson import ObjectId
+        client, db, collection = get_database_connection()
+        result = collection.delete_one({"_id": ObjectId(doc_id)})
+        return result.deleted_count > 0
+    except:
+        return False