Update app.py

app.py

@@ -6,13 +6,20 @@ from google.genai import types
 import gradio as gr
 import io
 from PIL import Image
-import tempfile
 
 def save_binary_file(file_name, data):
     f = open(file_name, "wb")
     f.write(data)
     f.close()
 
+# Function to convert a PIL Image to a base64 string
+def image_to_base64(img):
+    img_byte_arr = io.BytesIO()
+    img.save(img_byte_arr, format="PNG")
+    img_bytes = img_byte_arr.getvalue()
+    base64_string = base64.b64encode(img_bytes).decode("utf-8")
+    return f"data:image/png;base64,{base64_string}"
+
 def generate_image(prompt, image=None, output_filename="generated_image"):
     # Initialize client with the API key
     client = genai.Client(
@@ -89,11 +96,10 @@ def generate_image(prompt, image=None, output_filename="generated_image"):
 def chat_handler(user_input, user_image, chat_history):
     # Add user message to chat history
     if user_image:
-        # Save the uploaded image to a temporary file so Gradio can display it
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
-            user_image.save(tmp_file.name)
-            # Add the image to the chat history
-            chat_history.append({"role": "user", "content": tmp_file.name})
+        # Convert the uploaded image to base64
+        img_base64 = image_to_base64(user_image)
+        # Add the image to the chat history
+        chat_history.append({"role": "user", "content": img_base64})
     
     # Add the text prompt to the chat history
     if user_input:
@@ -109,11 +115,10 @@ def chat_handler(user_input, user_image, chat_history):
     
     # Add AI response to chat history
     if img:
-        # Save the PIL Image to a temporary file so Gradio can display it
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
-            img.save(tmp_file.name)
-            # Add the image as a file path that Gradio can serve
-            chat_history.append({"role": "assistant", "content": tmp_file.name})
+        # Convert the generated image to base64
+        img_base64 = image_to_base64(img)
+        # Add the image to the chat history
+        chat_history.append({"role": "assistant", "content": img_base64})
     
     # Add the status message
     chat_history.append({"role": "assistant", "content": status})
@@ -128,7 +133,7 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
     # Chatbot display area for the conversation thread
     chatbot = gr.Chatbot(
         label="Chat",
-        height=300,  # Reduced height from 400 to 300
+        height=300,  # Already reduced to 300
         type="messages",  # Explicitly set to 'messages' format
         avatar_images=(None, None)  # No avatars for simplicity
     )