Update app.py

app.py

@@ -636,29 +636,29 @@ def handsome_chat_completions():
                 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"Fetched 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})
+                  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})
 
                 # Construct the expected JSON output
                 response_data = {
-                    "images": openai_images, # use this key so the frontend can parse
+                    "images": openai_images,
                     "created": int(time.time()),
                     "timings": {
-                        "inference": total_time  # add inference time as shown in example
+                        "inference": total_time
                     }
                 }
                 if "seed" in response_json:
@@ -693,8 +693,8 @@ def handsome_chat_completions():
             logging.error(f"请求转发异常: {e}")
             return jsonify({"error": str(e)}), 500
     else:
-       # Existing text-based model handling logic
-       try:
+        # Existing text-based model handling logic
+        try:
             start_time = time.time()
             response = requests.post(
                 TEST_MODEL_ENDPOINT,
@@ -707,7 +707,6 @@ def handsome_chat_completions():
                 return jsonify(response.json()), 429
 
             if data.get("stream", False):
-              # ... (Existing stream response handling)
                 def generate():
                     first_chunk_time = None
                     full_response_content = ""
@@ -815,14 +814,10 @@ def handsome_chat_completions():
                         request_timestamps.append(time.time())
                         token_counts.append(prompt_tokens+completion_tokens)
 
-                    
-
-                    
-
-                    return Response(
-                       stream_with_context(generate()),
-                       content_type=response.headers['Content-Type']
-                   )
+                return Response(
+                    stream_with_context(generate()),
+                    content_type=response.headers['Content-Type']
+                )
             else:
                 response.raise_for_status()
                 end_time = time.time()
@@ -859,7 +854,8 @@ def handsome_chat_completions():
                                     item.get("type") == "text"
                                 ):
                                     user_content += (
-                                        item.get("text", "") + " "
+                                        item.get("text", "") +
+                                        " "
                                     )
 
                 user_content = user_content.strip()