Update app.py

app.py

@@ -2,6 +2,9 @@ import spaces
 import os
 import torch
 import random
+from PIL import Image
+import io
+import base64
 from huggingface_hub import snapshot_download
 from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256 import StableDiffusionXLPipeline
 from kolors.models.modeling_chatglm import ChatGLMModel
@@ -81,6 +84,21 @@ def generate_image(request: gr.Request,prompt, negative_prompt, height, width, n
         generator=torch.Generator(pipe.device).manual_seed(seed)
     ).images
     # testing
+
+
+image = image[0]  # Since it's in a list
+
+# Convert the image to an in-memory byte stream
+byte_stream = io.BytesIO()
+image.save(byte_stream, format='PNG')  # You can choose other formats like 'JPEG' if needed
+byte_data = byte_stream.getvalue()
+
+# Encode the byte stream to a base64 string
+base64_str = base64.b64encode(byte_data).decode('utf-8')
+
+# For demonstration, print the base64 string
+print(base64_str)
+
     query_data: dict = request.query_params
     save_data = {
             "prompt": prompt,
@@ -91,11 +109,13 @@ def generate_image(request: gr.Request,prompt, negative_prompt, height, width, n
             "num_images_per_prompt": num_images_per_prompt,
             "use_random_seed": use_random_seed,
             "seed": seed,
-            "output": image,
+            "output": base64_str,
             "seed_used" : seed
         }
+
     print(save_data)
     print(query_data)
+    
     # url = f"https://bots.spaceship.im/save_data/{query_data['uuid']}"
     # res = requests.post(url, json=save_data)