Update image URL handling in format_history_as_messages function to support SPACE_HOST environment variable

app/app.py

@@ -77,9 +77,12 @@ def format_history_as_messages(history: list):
 
         if isinstance(content, tuple):  # Handle file paths
             for path in content:
-                data_uri = _convert_path_to_data_uri(path)
+                if space_host := os.getenv("SPACE_HOST"):
+                    url = f"https://{space_host}/gradio_api/file%3D/{path}"
+                else:
+                    url = _convert_path_to_data_uri(path)
                 current_message_content.append(
-                    {"type": "image_url", "image_url": {"url": data_uri}}
+                    {"type": "image_url", "image_url": {"url": url}}
                 )
         elif isinstance(content, str):  # Handle text
             current_message_content.append({"type": "text", "text": content})
@@ -398,3 +401,5 @@ with gr.Blocks(css=css) as demo:
     )
 
 demo.launch()
+
+# /private/var/folders/9t/msy700h16jz3q35qvg4z1ln40000gn/T/gradio/a5013b9763ad9f2192254540fee226539fbcd1382cbc2317b916aef469bb01b9/Screenshot 2025-01-13 at 08.02.26.png