Update app.py

app.py

@@ -10,6 +10,8 @@ from datetime import datetime, timedelta
 import threading
 import atexit
 import uuid
+import json
+import re
 from collections import defaultdict
 
 app = Flask(__name__)
@@ -186,6 +188,27 @@ def verify_admin_password(password):
     """Verify admin password"""
     return password == ADMIN_PASSWORD
 
+def fix_malformed_json(json_str):
+    """Fix the malformed JSON from upstream server"""
+    try:x
+        # The upstream sends: {"choices":[{"delta":{"content":"text"},"index":0}],"created":123,"id":"chatcmpl-abc"123,"model":"gpt-5-nano","object":"chat.completion.chunk"}
+        # We need to fix it to: {"choices":[{"delta":{"content":"text"},"index":0}],"created":123,"id":"chatcmpl-abc","model":"gpt-5-nano","object":"chat.completion.chunk"}
+
+        # Find the pattern where id value is immediately followed by another value without comma
+        pattern = r'"id":"([^"]*)"(\d+)'
+        match = re.search(pattern, json_str)
+
+        if match:
+            id_value = match.group(1)
+            timestamp = match.group(2)
+            # Replace the malformed part with proper comma separation
+            fixed = re.sub(pattern, f'"id":"{id_value}","created":{timestamp}', json_str)
+            return fixed
+
+        return json_str
+    except:
+        return json_str
+
 def export_database():
     """Export current database to JSON"""
     try:
@@ -886,6 +909,15 @@ def chat_completions():
                     if chunk:
                         chunk_str = chunk.decode('utf-8', errors='replace')
 
+                        # Fix malformed JSON from upstream server
+                        if 'data: {' in chunk_str:
+                            # Extract JSON part
+                            json_start = chunk_str.find('data: {')
+                            if json_start != -1:
+                                json_part = chunk_str[json_start + 6:]  # Remove 'data: ' prefix
+                                fixed_json = fix_malformed_json(json_part)
+                                chunk_str = f"data: {fixed_json}"
+
                         # Replace the chat completion ID with our UUID
                         chunk_str = chunk_str.replace('"id":"chatcmpl-', f'"id":"{completion_id}"')