Update fluxai.py

fluxai.py

@@ -103,33 +103,30 @@ async def fluxai_image(payload: FluxAI):
                 )
 
             if payload.auto_enhancer:
-                with Image.open(io.BytesIO(image_bytes)) as image:
+                with Image.open(BytesIO(image_bytes)) as image:
                     enhancer = ImageEnhance.Sharpness(image)
                     image = enhancer.enhance(1.5)
                     enhancer = ImageEnhance.Contrast(image)
                     image = enhancer.enhance(1.2)
                     enhancer = ImageEnhance.Color(image)
                     image = enhancer.enhance(1.1)
-                    enhanced_image_bytes = io.BytesIO()
+                    
+                    enhanced_image_bytes = BytesIO()
                     image.save(enhanced_image_bytes, format="JPEG", quality=95)
                     enhanced_image_bytes.seek(0)
+                    encoded_string = base64.b64encode(enhanced_image_bytes.getvalue()).decode("utf-8")
 
-                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, enhanced_image_bytes)
+                response = x.get_response_image(example_test, encoded_string)
                 
                 return SuccessResponse(
                     status="True",
                     randydev={"image_data": encoded_string, "caption": response}
                 )
+
             else:
-                encoded_string = base64.b64encode(image_bytes).decode("utf-8")
-                return SuccessResponse(
-                    status="True",
-                    randydev={"image_data": encoded_string}
-                )
+                return StreamingResponse(BytesIO(image_bytes), media_type="image/jpeg")
 
         except Exception as e:
             return SuccessResponse(
@@ -141,4 +138,4 @@ async def fluxai_image(payload: FluxAI):
         return SuccessResponse(
             status="False",
             randydev={"error": f"Not enough tokens. Current tokens: {tokens}. Please support @xtdevs"}
-        )
+        )