Update app.py

app.py

@@ -26,6 +26,11 @@ def save_image_to_static(img, prefix="image"):
     # Return the relative path
     return filepath
 
+def get_file_url(filepath):
+    # Convert the relative file path to a URL that Gradio can serve
+    # Gradio serves files using the /file/ prefix
+    return f"/file/{filepath}"
+
 def generate_image(prompt, image=None, output_filename="generated_image"):
     # Initialize client with the API key
     client = genai.Client(
@@ -104,8 +109,10 @@ def chat_handler(user_input, user_image, chat_history):
     if user_image:
         # Save the uploaded image to the static directory
         img_path = save_image_to_static(user_image, prefix="uploaded")
-        # Add the image to the chat history
-        chat_history.append({"role": "user", "content": img_path})
+        # Convert the file path to a URL that Gradio can serve
+        img_url = get_file_url(img_path)
+        # Add the image URL to the chat history
+        chat_history.append({"role": "user", "content": img_url})
     
     # Add the text prompt to the chat history
     if user_input:
@@ -123,8 +130,10 @@ def chat_handler(user_input, user_image, chat_history):
     if img:
         # Save the generated image to the static directory
         img_path = save_image_to_static(img, prefix="generated")
-        # Add the image to the chat history
-        chat_history.append({"role": "assistant", "content": img_path})
+        # Convert the file path to a URL that Gradio can serve
+        img_url = get_file_url(img_path)
+        # Add the image URL to the chat history
+        chat_history.append({"role": "assistant", "content": img_url})
     
     # Add the status message
     chat_history.append({"role": "assistant", "content": status})