Update app.py

app.py

@@ -30,16 +30,20 @@ logger = logging.getLogger(__name__)
 # MongoDB Configuration
 MONGODB_URI = os.getenv("MONGODB_URI")
 DATABASE_NAME = os.getenv("DATABASE_NAME")
-COLLECTION_NAME = os.getenv("COLLECTION_NAME")
-
-# Initialize MongoDB Connection
-client = MongoClient(MONGODB_URI)
-db = client[DATABASE_NAME]
+COLLECTION_NAME = os.getenv("COLLECTION_NAME","invoice_collection")
 
 app = FastAPI(docs_url='/')
 use_gpu = False
 output_dir = 'output'
 
+if not MONGODB_URI:
+    raise ValueError("MONGODB_URL is not set. Please add it to Hugging Face secrets.")
+
+# Initialize MongoDB Connection
+client = MongoClient(MONGODB_URI)
+db = client[DATABASE_NAME]
+invoice_collection=db[COLLECTION_NAME]
+
 # Initialize PaddleOCR
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
@@ -58,8 +62,6 @@ s3_client = boto3.client(
     aws_access_key_id=AWS_ACCESS_KEY,
     aws_secret_access_key=AWS_SECRET_KEY
 )
-if not MONGODB_URI:
-    raise ValueError("MONGODB_URL is not set. Please add it to Hugging Face secrets.")
 
 # Function to fetch file from S3