Update main.py

main.py

@@ -227,7 +227,7 @@ class GmailRequest(BaseModel):
     to_email: str
     subject: str
     customer_name: str
-    order_id: str
+    order_id: Optional[str] = None
     status: str
     total_price: str
     products: List[ProductItem]
@@ -239,12 +239,15 @@ class CompleteRequest(BaseModel):
     mk: str
 
 @app.post("/api/v1/gmail/create")
-async def create_gmail_notif(data: GmailRequest, x_api_key: str = Header(None)):
+async def create_gmail_notif(
+    data: GmailRequest,
+    x_api_key: str = Header(None)
+):
     if x_api_key != API_KEY:
-        raise HTTPException(status_code=403, detail="Invalid Key")
+        raise HTTPException(status_code=403, detail="API key invalid.")
+
+    final_order_id = data.order_id if data.order_id else gm_app.generate_random_order_id()
 
-    final_order_id = gm_app.generate_order_id()
-    
     order_cache[final_order_id] = {
         "email": data.to_email,
         "name": data.customer_name,
@@ -262,9 +265,11 @@ async def create_gmail_notif(data: GmailRequest, x_api_key: str = Header(None)):
         total_price=data.total_price,
         from_name=data.from_name
     )
-    
-    if res["ok"]:
-        res["order_id"] = final_order_id
+
+    if not res["ok"]:
+        raise HTTPException(status_code=500, detail=res["error"])
+
+    res["order_id"] = final_order_id
     return res
 
 @app.post("/api/v1/gmail/done")