Update fluxai.py

fluxai.py

@@ -5,6 +5,7 @@ from fastapi import HTTPException
 import io
 import requests
 import os
+import base64
 from dotenv import load_dotenv
 from pydantic import BaseModel
 from pymongo import MongoClient
@@ -100,6 +101,7 @@ async def fluxai_image(payload: FluxAI):
                     status="False",
                     randydev={"error": "Failed to generate an image"}
                 )
+
             if payload.auto_enhancer:
                 with Image.open(io.BytesIO(image_bytes)) as image:
                     enhancer = ImageEnhance.Sharpness(image)
@@ -111,24 +113,23 @@ async def fluxai_image(payload: FluxAI):
                     enhanced_image_bytes = io.BytesIO()
                     image.save(enhanced_image_bytes, format="JPEG", quality=95)
                     enhanced_image_bytes.seek(0)
-    
-                file_path = "test.jpg"
-                with open(file_path, "wb") as f:
-                    f.write(enhanced_image_bytes.getvalue())
 
-                url = "https://randydev-ryuzaki-api.hf.space/uploadfile/"
-                files = {"file": open(file_path, "rb")}
-                response_uploaded = requests.post(url, files=files).json()
-                get_url = response_uploaded.get("url")
+                encoded_string = base64.b64encode(enhanced_image_bytes.getvalue()).decode("utf-8")
+                
                 example_test = "Explain how this picture looks like."
                 x = GeminiLatest(api_keys=GOOGLE_API_KEY)
-                response = x.get_response_image(example_test, file_path)
+                response = x.get_response_image(example_test, enhanced_image_bytes)
+                
                 return SuccessResponse(
                     status="True",
-                    randydev={"url": get_url, "caption": response}
+                    randydev={"image_data": encoded_string, "caption": response}
                 )
             else:
-                return StreamingResponse(io.BytesIO(image_bytes), media_type="image/jpeg")
+                encoded_string = base64.b64encode(image_bytes).decode("utf-8")
+                return SuccessResponse(
+                    status="True",
+                    randydev={"image_data": encoded_string}
+                )
 
         except Exception as e:
             return SuccessResponse(
@@ -140,4 +141,4 @@ async def fluxai_image(payload: FluxAI):
         return SuccessResponse(
             status="False",
             randydev={"error": f"Not enough tokens. Current tokens: {tokens}. Please support @xtdevs"}
-        )
+        )