Update app.py

app.py

@@ -146,32 +146,83 @@ def chat_completions():
                 }
             )
         else:
-            text_matches = re.findall(r'0:"(.*?)"', response.text)
-            parsed_text = "".join(text_matches)
-
-            return Response(
-                json.dumps({
-                    "object": "chat.completion",
-                    "created": int(time.time() * 1000),
-                    "model": data.get('model', "DeepSeek-R1"),
-                    "choices": [
-                        {
-                            "index": 0,
-                            "message": {
-                                "role": "user",
-                                "content": parsed_text
-                            },
-                            "finish_reason": "stop"
-                        }
-                    ]
-                },ensure_ascii=False),
-                status=response.status_code,
-                headers={
-                    'Cache-Control': 'no-cache',
-                    'Connection': 'keep-alive',
-                    'Content-Type': 'application/json'
-                }
-            )
+            if data.get('model', "DeepSeek-R1") != "AkashGen":
+                text_matches = re.findall(r'0:"(.*?)"', response.text)
+                parsed_text = "".join(text_matches)
+                
+                return Response(
+                    json.dumps({
+                        "object": "chat.completion",
+                        "created": int(time.time() * 1000),
+                        "model": data.get('model', "DeepSeek-R1"),
+                        "choices": [
+                            {
+                                "index": 0,
+                                "message": {
+                                    "role": "user",
+                                    "content": parsed_text
+                                },
+                                "finish_reason": "stop"
+                            }
+                        ]
+                    },ensure_ascii=False),
+                    status=response.status_code,
+                    headers={
+                        'Cache-Control': 'no-cache',
+                        'Connection': 'keep-alive',
+                        'Content-Type': 'application/json'
+                    }
+                )
+            else:
+                match = re.search(r"jobId='([^']+)'", response.text)
+                job_id = None
+                if match:
+                    job_id = match.group(1)
+                    print("jobId:", job_id)
+                    while True:
+                        try:
+                            _img_response = requests.get(
+                                'https://chat.akash.network/api/image-status?ids='+job_id,
+                                headers=headers
+                            )
+                            _data = _img_response.json()
+                            # 检查响应状态码是否为200
+                            if _data[0]["status"] == "completed":
+                                print("图片生成完成:", job_id)
+                                return Response(
+                                    json.dumps({
+                                        "object": "chat.completion",
+                                        "created": int(time.time() * 1000),
+                                        "model": data.get('model', "DeepSeek-R1"),
+                                        "choices": [
+                                            {
+                                                "index": 0,
+                                                "message": {
+                                                    "role": "user",
+                                                    "content": _data[0]["result"]
+                                                },
+                                                "finish_reason": "stop"
+                                            }
+                                        ]
+                                    },ensure_ascii=False),
+                                    status=response.status_code,
+                                    headers={
+                                        'Cache-Control': 'no-cache',
+                                        'Connection': 'keep-alive',
+                                        'Content-Type': 'application/json'
+                                    }
+                                )
+                            else:
+                                print("图片生成中:", job_id)
+                        except Exception as e:
+                            print("请求过程中出现异常:", e)
+                        
+                        # 每隔5秒请求一次
+                        time.sleep(5)
+                else:
+                    return jsonify({"error": "当前官方服务异常"}), 500
+                
+                
     
     except Exception as e:
         print(e)