Upload 2 files

main.py

@@ -78,7 +78,8 @@ import logging
 import functions as code
 
 logging.basicConfig(level=logging.ERROR)
-
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 load_dotenv()
 
@@ -98,6 +99,7 @@ SOURCE_WHAT_GAY_URL = os.environ["SOURCE_WHAT_GAY_URL"]
 SOURCE_ASSISTANT_GOOGLE_AI = os.environ["SOURCE_ASSISTANT_GOOGLE_AI"]
 SOURCE_MONITOR_URL = os.environ["SOURCE_MONITOR_URL"]
 SOURCE_OPENAI_ACCESS_URL = os.environ["SOURCE_OPENAI_ACCESS_URL"]
+SOURCE_PICSART_URL = os.environ["SOURCE_PICSART_URL"]
 
 # api keys
 REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
@@ -107,6 +109,7 @@ HUGGING_TOKEN = os.environ["HUGGING_TOKEN"]
 ASSISTANT_GOOGLE_API_KEYS = os.environ["ASSISTANT_GOOGLE_API_KEYS"]
 COOKIE_BARD_TOKEN = os.environ["COOKIE_BARD_TOKEN"]
 MONITOR_API_KEYS = os.environ["MONITOR_API_KEYS"]
+PICSART_API_KEY = os.environ["PICSART_API_KEY"]
 
 client_mongo = MongoClient(MONGO_URL)
 db = client_mongo["tiktokbot"]
@@ -405,7 +408,7 @@ def get_translate(
         )
     except:
         return SuccessResponse(
-            status="True",
+            status="False",
             randydev={"message": "Error not responding"})
 
 @app.get("/ryuzaki/reverse", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
@@ -547,6 +550,48 @@ def open_dalle(
     else:
         return SuccessResponse(status="False", randydev={"data": "Not found image data"})
 
+@app.post("/ryuzaki/picsart-pro")
+def Picsart_Pro(
+    item: Picsart,
+    api_key: None = Depends(validate_api_key)
+):
+    API_URL = SOURCE_PICSART_URL
+    try:
+        if not os.path.exists(item.image_path):
+            raise HTTPException(status_code=400, detail="Image file not found")
+
+        files = {"image": ("brosur.png", open(item.image_path, "rb"), "image/png")}
+        payload = {"format": "PNG", "output_type": "cutout"}
+        headers = {"accept": "application/json", "x-picsart-api-key": PICSART_API_KEY}
+
+        with requests.post(url, headers=headers, data=payload, files=files) as response:
+            response.raise_for_status()
+            response_data = response.json()
+            urls = response_data["data"]["url"]
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error in Picsart request: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+
+    try:
+        response_two = requests.get(urls)
+        response_two.raise_for_status()
+
+        encoded_string = base64.b64encode(response_two.content).decode("utf-8")
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error in Picsart image retrieval: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+
+    if encoded_string:
+        return SuccessResponse(status="True", randydev={"data": encoded_string})
+    else:
+        return SuccessResponse(status="False", randydev={"data": "Not found image data"})
+
 @app.post("/ryuzaki/anime-styled")
 def Anime_Styled(
     item: AnimeStyled,

models.py

@@ -147,3 +147,6 @@ class Chatgpt3Texts(BaseModel):
 
 class TextCustom(BaseModel):
     query: str
+
+class Picsart(BaseModel):
+    image_path: str