Update app.py

app.py

@@ -649,6 +649,7 @@ def handsome_chat_completions():
                             image_url = images[0]
                             logging.info(f"Extracted image URL: {image_url}")
                         
+                        markdown_image_link = f"![image]({image_url})"
                         if image_url:
                             chunk_data = {
                                 "id": f"chatcmpl-{uuid.uuid4()}",
@@ -660,14 +661,14 @@ def handsome_chat_completions():
                                         "index": 0,
                                         "delta": {
                                             "role": "assistant",
-                                            "content": image_url
+                                            "content": markdown_image_link
                                         },
                                         "finish_reason": None
                                     }
                                 ]
                             }
                             yield f"data: {json.dumps(chunk_data)}\n\n".encode('utf-8')
-                            full_response_content = image_url
+                            full_response_content = markdown_image_link
                         else:
                             chunk_data = {
                                 "id": f"chatcmpl-{uuid.uuid4()}",
@@ -764,6 +765,7 @@ def handsome_chat_completions():
                         image_url = images[0]
                         logging.info(f"Extracted image URL: {image_url}")
                     
+                    markdown_image_link = f"![image]({image_url})"
                     # Construct the expected JSON output - Mimicking OpenAI
                     response_data = {
                         "id": f"chatcmpl-{uuid.uuid4()}",
@@ -775,7 +777,7 @@ def handsome_chat_completions():
                             "index": 0,
                             "message": {
                                 "role": "assistant",
-                                "content": image_url if image_url else "Failed to generate image", # Directly return the URL in content
+                                "content": markdown_image_link if image_url else "Failed to generate image", # Directly return the URL in content
                             },
                             "finish_reason": "stop",
                             }