added the Final message image rendering strcture to see Images in UI

Author: Dhruvkumar-Microsoft

src/backend/agents/image_agent.py

@@ -202,7 +202,11 @@ async def _invoke_stream(
                         self.user_id,
                         message_type=WebsocketMessageType.AGENT_MESSAGE,
                     )
-                    logger.info("ImageAgent '%s': image sent to user '%s' via WebSocket", self.agent_name, self.user_id)
+                    logger.info(
+                        "ImageAgent '%s': image sent to user '%s' via WebSocket "
+                        "(response_id=%s message_id=%s, NOT in orchestrator final text)",
+                        self.agent_name, self.user_id, response_id, message_id,
+                    )
                 except Exception as ws_exc:
                     logger.error("ImageAgent '%s': failed to send image via WebSocket: %s", self.agent_name, ws_exc)
 

src/backend/app.py

@@ -81,10 +81,16 @@ async def lifespan(app: FastAPI):
     for logger_name in packages:
         logging.getLogger(logger_name).setLevel(azure_level)
 
+for _af_logger in ("agent_framework", "agent_framework.openai"):
+    logging.getLogger(_af_logger).setLevel(azure_level)
+
 logging.getLogger("opentelemetry.sdk").setLevel(logging.ERROR)
 
 logging.getLogger("azure.core.pipeline.policies.http_logging_policy").setLevel(logging.WARNING)
 
+# Suppress noisy Azure Cosmos DB HTTP request/response logging
+logging.getLogger("azure.cosmos._cosmos_http_logging_policy").setLevel(logging.WARNING)
+
 # Suppress noisy Azure Monitor exporter "Transmission succeeded" logs
 logging.getLogger("azure.monitor.opentelemetry.exporter.export._base").setLevel(logging.WARNING)
 

src/backend/orchestration/orchestration_manager.py

@@ -3,6 +3,7 @@
 import asyncio
 import logging
 import uuid
+import re
 from typing import List, Optional
 
 import models.messages as messages
@@ -33,6 +34,26 @@
 # See localspec/bugs/framework/F1-tool-history-leak.md
 apply_tool_history_leak_patch()
 
+_BARE_IMAGE_URL_RE = re.compile(
+    r"(?<![\(\]])"                        
+    r"(?<!\]\()"                          
+    r"(https?://[^\s)]+?"
+    r"(?:/api/v4/images/[^\s)]+?|[^\s)]+?\.(?:png|jpe?g|gif|webp)))"
+    r"(?=[\s)\]]|$)",
+    re.IGNORECASE,
+)
+
+
+def _embed_bare_image_urls(text: str) -> str:
+    """Wrap bare image URLs in markdown image syntax so the UI renders them inline.
+
+    Skips URLs already inside ``![alt](url)`` or ``[text](url)`` (handled by the
+    negative lookbehinds), so it never double-wraps an existing markdown embed.
+    """
+    if not text:
+        return text
+    return _BARE_IMAGE_URL_RE.sub(r"![Generated image](\1)", text)
+
 class OrchestrationManager:
     """Manager for handling orchestration logic using agent_framework Magentic workflow."""
 
@@ -378,6 +399,21 @@ async def run_orchestration(self, user_id: str, input_task) -> None:
             # accumulated orchestrator streaming chunks.
             final_text = final_output_ref[0] or "".join(orchestrator_chunks)
 
+            final_text = _embed_bare_image_urls(final_text)
+
+            # Issue 1 diagnostic: confirm the final answer carries a renderable image
+            # embed. has_image_markdown tracks TRUE markdown (![]) — the renderable form;
+            # has_image_url tracks any image reference, even a bare URL.
+            final_source = "executor" if final_output_ref[0] else "chunks"
+            has_image_markdown = "![" in final_text
+            has_image_url = "/api/v4/images/" in final_text
+            self.logger.info(
+                "[FINAL-ASSEMBLY] job=%s user=%s source=%s len=%d "
+                "has_image_markdown=%s has_image_url=%s",
+                job_id, user_id, final_source, len(final_text),
+                has_image_markdown, has_image_url,
+            )
+            
             # Log results
             self.logger.info("\nAgent responses:")
             self.logger.info(
@@ -721,15 +757,16 @@ async def _process_event_stream(
                                     executor, cb_err,
                                 )
 
-                        try:
-                            await streaming_agent_response_callback(
-                                executor, output_data, False, user_id,
-                            )
-                        except Exception as cb_err:
-                            self.logger.error(
-                                "Error in streaming callback for %s: %s",
-                                executor, cb_err,
-                            )
+                        if executor != "magentic_orchestrator":
+                            try:
+                                await streaming_agent_response_callback(
+                                    executor, output_data, False, user_id,
+                                )
+                            except Exception as cb_err:
+                                self.logger.error(
+                                    "Error in streaming callback for %s: %s",
+                                    executor, cb_err,
+                                )
 
                 # Executor completed
                 elif (