Update app.py

app.py

@@ -15,8 +15,8 @@ from apscheduler.schedulers.background import BackgroundScheduler
 from flask import Flask, request, jsonify, Response, stream_with_context
 from werkzeug.middleware.proxy_fix import ProxyFix
 
-os.environ['TZ'] = 'Asia/Shanghai'
-time.tzset()
+# os.environ['TZ'] = 'Asia/Shanghai'
+# time.tzset()
 
 logging.basicConfig(level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(message)s')
@@ -123,7 +123,6 @@ def create_base64_markdown_image(image_url):
         base64_encoded = base64.b64encode(image_data).decode('utf-8')
         mime_type = response.headers.get('Content-Type', 'image/png')
         markdown_image_link = f"![](data:{mime_type};base64,{base64_encoded})"
-        # print(markdown_image_link)
         logging.info(f"Created base64 markdown image link.")
         return markdown_image_link
     except requests.exceptions.RequestException as e:
@@ -956,23 +955,26 @@ def handsome_chat_completions():
                         
                         markdown_image_link = create_base64_markdown_image(image_url)
                         if image_url:
-                            chunk_data = {
-                                "id": f"chatcmpl-{uuid.uuid4()}",
-                                "object": "chat.completion.chunk",
-                                "created": int(time.time()),
-                                "model": model_name,
-                                "choices": [
-                                    {
-                                        "index": 0,
-                                        "delta": {
-                                            "role": "assistant",
-                                            "content": markdown_image_link
-                                        },
-                                        "finish_reason": None
-                                    }
-                                ]
-                            }
-                            yield f"data: {json.dumps(chunk_data)}\n\n".encode('utf-8')
+                            chunk_size = 2048
+                            for i in range(0, len(markdown_image_link), chunk_size):
+                                chunk = markdown_image_link[i:i + chunk_size]
+                                chunk_data = {
+                                    "id": f"chatcmpl-{uuid.uuid4()}",
+                                    "object": "chat.completion.chunk",
+                                    "created": int(time.time()),
+                                    "model": model_name,
+                                    "choices": [
+                                        {
+                                            "index": 0,
+                                            "delta": {
+                                                "role": "assistant",
+                                                "content": chunk
+                                            },
+                                            "finish_reason": None
+                                        }
+                                    ]
+                                }
+                                yield f"data: {json.dumps(chunk_data)}\n\n".encode('utf-8')
                         else:
                             chunk_data = {
                                 "id": f"chatcmpl-{uuid.uuid4()}",