Update fluxai.py

fluxai.py

@@ -115,7 +115,7 @@ def get_all_api_keys():
     return api_keys
 
 @router.post("/akeno/fluxai", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
-async def fluxai_image(payload: FluxAI, file: UploadFile):
+async def fluxai_image(payload: FluxAI):
     if deduct_tokens_gpt(payload.user_id, amount=20):
         USERS_API_KEYS = get_all_api_keys()
         if payload.api_key in USERS_API_KEYS:
@@ -134,22 +134,18 @@ async def fluxai_image(payload: FluxAI, file: UploadFile):
                         image = enhancer.enhance(1.2)
                         enhancer = ImageEnhance.Color(image)
                         image = enhancer.enhance(1.1)
-                        enhanced_image_bytes = io.BytesIO()
+                        
+                        enhanced_image_bytes = "akeno.jpg"
                         image.save(enhanced_image_bytes, format="JPEG", quality=95)
-                        enhanced_image_bytes.seek(0)
-                    ext = file.filename.split(".")[-1]
-                    unique_filename = f"{uuid.uuid4().hex}.{ext}"
-                    file_path = os.path.join("uploads", unique_filename)
-                    os.makedirs(os.path.dirname(file_path), exist_ok=True)
-                    with open(file_path, "wb") as f:
-                        f.write(enhanced_image_bytes.getvalue())
+                        with open(enhanced_image_bytes, "rb") as image_file:
+                            encoded_string = base64.b64encode(image_file.read())
+
                     example_test = "Accurately identify the baked good in the image and provide an appropriate and recipe consistent with your analysis."
                     x = GeminiLatest(api_keys=GOOGLE_API_KEY)
-                    response = x.get_response_image(example_test, file_path)
-                    url = f"https://randydev-ryuzaki-api.hf.space/{file_path}"
+                    response = x.get_response_image(example_test, enhanced_image_bytes)
                     return SuccessResponse(
                         status="True",
-                        randydev={"url": url, "caption": response}
+                        randydev={"image_data": encoded_string, "caption": response}
                     )
                 else:
                     return StreamingResponse(io.BytesIO(image_bytes), media_type="image/jpeg")