Update main.py

main.py

@@ -124,7 +124,14 @@ contact_support = """
 We are aware that AI is currently offline. This seems to be caused by the API
 We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
 
-Contact Support @xtdevs
+~ Contact Support @xtdevs
+"""
+
+internal_error = """
+There has been an Internal error. We are aware of this error and notice that it can be
+caused by your search terms being to explict, too confusing, or it can be caused by the API.
+Please modify your search terms and/or try again later thank you for your understanding.
+~ @xtdevs
 """
 
 def get_all_api_keys():
@@ -548,7 +555,7 @@ def dalle_3xl(
     except requests.exceptions.RequestException:
         raise HTTPException(
             status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
+            detail=CustomErrorResponseModel(detail=[{"Error": internal_error}])
         )
 
     try:
@@ -556,7 +563,7 @@ def dalle_3xl(
     except Exception:
         raise HTTPException(
             status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
+            detail=CustomErrorResponseModel(detail=[{"Error": contact_support}])
         )
 
     if encoded_string:
@@ -578,14 +585,14 @@ def open_dalle(
     except requests.exceptions.RequestException:
         raise HTTPException(
             status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
+            detail=CustomErrorResponseModel(detail=[{"Error": internal_error}])
         )
     try:
         encoded_string = base64.b64encode(response.content).decode("utf-8")
     except Exception:
         raise HTTPException(
             status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
+            detail=CustomErrorResponseModel(detail=[{"Error": contact_support}])
         )
     if encoded_string:
         return SuccessResponse(status="True", randydev={"data": encoded_string})
@@ -788,7 +795,7 @@ def v1beta3_google_ai(
             message = results.get("output")
         return SuccessResponse(status="True", randydev={"message": message})
     except:
-        return SuccessResponse(status="False", randydev={"message": contact_support})
+        return SuccessResponse(status="False", randydev={"message": internal_error})
 
 @app.post("/ryuzaki/gemini-ai-pro", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def gemini_pro(item: GeminiPro):
@@ -1369,6 +1376,17 @@ description = """
 
 •Developed by [@xtdevs](https://t.me/xtdevs)
 """
+
+def custom_exception_handler(request: Request, exc: HTTPException) -> JSONResponse:
+    error_detail = [{"error": str(exc.detail)}]
+    custom_error_model = CustomErrorResponseModel(detail=error_detail)
+
+    return JSONResponse(
+        status_code=exc.status_code,
+        content=custom_error_model.dict(),
+        headers=exc.headers,
+    )
+
 def custom_openapi():
     if app.openapi_schema:
         return app.openapi_schema
@@ -1386,3 +1404,5 @@ def custom_openapi():
     return app.openapi_schema
 
 app.openapi = custom_openapi
+
+app.add_exception_handler(HTTPException, custom_exception_handler)