Update app.py

app.py

@@ -36,14 +36,25 @@ 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.")
+# Check if environment variables are set
+if not all([MONGODB_URI, AWS_ACCESS_KEY, AWS_SECRET_KEY, S3_BUCKET_NAME, API_KEY, openai.api_key]):
+    raise ValueError("One or more required environment variables are not set.")
 
 # Initialize MongoDB Connection
 client = MongoClient(MONGODB_URI)
 db = client[DATABASE_NAME]
 invoice_collection=db[COLLECTION_NAME]
 
+@app.on_event("startup")
+def startup_db():
+    try:
+        # Create a MongoDB client and fetch server info to verify connection
+        client = MongoClient(MONGODB_URI)
+        server_info = client.server_info()  # This is a blocking call
+        print("MongoDB connection successful:", server_info)
+    except Exception as e:
+        print("MongoDB connection failed:", str(e))
+
 # Initialize PaddleOCR
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
@@ -166,6 +177,7 @@ This JSON format will be used to store and manage invoices in a structured and u
     #         return None
     # except Exception as e:
     #     return f"Error in summarization: {str(e)}"
+
 # Dependency to check API Key
 def verify_api_key(api_key: str = Header(...)):
     if api_key != API_KEY: