Update app.py

app.py

@@ -113,6 +113,24 @@ def test_model_availability(api_key, model_name):
             f"API Key：{api_key}，错误信息：{e}"
         )
         return False
+    
+def create_base64_markdown_image(image_url):
+    try:
+        response = requests.get(image_url, stream=True)
+        response.raise_for_status()=
+
+        image_data = response.content
+        base64_encoded = base64.b64encode(image_data).decode('utf-8')
+        mime_type = response.headers.get('Content-Type', 'image/png')
+        markdown_image_link = f"![image](data:{mime_type};base64,{base64_encoded})"
+        logging.info(f"Created base64 markdown image link.")
+        return markdown_image_link
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error downloading image: {e}")
+        return None
+    except Exception as e:
+        logging.error(f"Error during processing: {e}")
+        return None
 
 def refresh_models():
     global text_models, free_text_models
@@ -838,8 +856,8 @@ def handsome_chat_completions():
         siliconflow_data = {
             "model": model_name,
             "prompt": user_content,
-            
         }
+
         if model_name == "black-forest-labs/FLUX.1-pro":
             siliconflow_data["width"] = data.get("width", 1024)
             siliconflow_data["height"] = data.get("height", 768)
@@ -935,7 +953,7 @@ def handsome_chat_completions():
                             image_url = images[0]
                             logging.info(f"Extracted image URL: {image_url}")
                         
-                        markdown_image_link = f"![image]({image_url})"
+                        markdown_image_link = create_base64_markdown_image(image_url)
                         if image_url:
                             chunk_data = {
                                 "id": f"chatcmpl-{uuid.uuid4()}",