Update app.py

app.py

@@ -634,25 +634,16 @@ def handsome_chat_completions():
             
             try:
                 images = response_json.get("images", [])
-                openai_images = []
-                for item in images:
-                  if isinstance(item, dict) and "url" in item:
-                    image_url = item["url"]
-                    logging.info(f"Extracted image URL: {image_url}")  # Log the URL
-                    try:
-                        image_data = requests.get(image_url, stream=True).raw
-                        image = Image.open(image_data)
-                        buffered = io.BytesIO()
-                        image.save(buffered, format="PNG")
-                        img_str = base64.b64encode(buffered.getvalue()).decode()
-                        openai_images.append({"b64_json": img_str})
-                    except Exception as e:
-                        logging.error(f"图片转base64失败: {e}")
-                        openai_images.append({"url": image_url})
-                  else:
-                    logging.error(f"无效的图片数据: {item}")
-                    openai_images.append({"url": item})
-
+                
+                # Extract the first URL if available
+                image_url = ""
+                if images and isinstance(images[0], dict) and "url" in images[0]:
+                    image_url = images[0]["url"]
+                    logging.info(f"Extracted image URL: {image_url}")
+                elif images and isinstance(images[0], str):
+                  image_url = images[0]
+                  logging.info(f"Extracted image URL: {image_url}")
+                
                 # Construct the expected JSON output - Mimicking OpenAI
                 response_data = {
                    "id": f"chatcmpl-{uuid.uuid4()}",
@@ -664,8 +655,7 @@ def handsome_chat_completions():
                         "index": 0,
                         "message": {
                             "role": "assistant",
-                            "content": "Generated Image:", # Changed content here
-                             "images": openai_images
+                            "content": image_url if image_url else "Failed to generate image", # Directly return the URL in content
                         },
                         "finish_reason": "stop",
                      }
@@ -687,8 +677,7 @@ def handsome_chat_completions():
                         "index": 0,
                         "message": {
                             "role": "assistant",
-                            "content": "Error Generating Image:", # Changed content here
-                            "images": []
+                            "content": "Failed to process image data",
                         },
                         "finish_reason": "stop",
                       }
@@ -710,7 +699,7 @@ def handsome_chat_completions():
             logging.error(f"请求转发异常: {e}")
             return jsonify({"error": str(e)}), 500
     else:
-       # Existing text-based model handling logic
+         # Existing text-based model handling logic
         try:
             start_time = time.time()
             response = requests.post(