Update app.py

app.py

@@ -1,3 +1,26 @@
 import gradio as gr
+import base64
+from io import BytesIO
+from PIL import Image
 
-gr.load("models/stabilityai/stable-diffusion-3.5-large").launch()
+# Function to convert image to base64
+def image_to_base64(image: Image.Image):
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")  # You can change the format if needed
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+# Function to generate the image and return base64-encoded response
+def generate_base64_image(prompt: str):
+    # Load the pre-trained model (adjust to your needs)
+    model = gr.load("models/stabilityai/stable-diffusion-3.5-large")
+    
+    # Generate image with the model
+    img = model.predict(prompt)[0]  # Assuming the model returns the generated image
+    
+    # Convert image to base64
+    return image_to_base64(img)
+
+# Define Gradio interface
+iface = gr.Interface(fn=generate_base64_image, inputs="text", outputs="text")  # Outputs base64 text
+
+iface.launch()