Update main.py

main.py

@@ -132,48 +132,48 @@ def chat_completions():
             },
         )
     else:
-	    response = requests.post(f"{base_url}/queue/join", json=json_prompt)
-	    url = f"{base_url}/queue/data?session_hash={session_hash}"
-	    data_stream = requests.get(url, stream=True)
-
-	    time_now = int(time.time())
-	    collected_data = []
-
-	    try:
-	        for line in data_stream.iter_lines():
-	            if line:
-	                try:
-	                    decoded_line = line.decode("utf-8")
-	                    json_line = json.loads(decoded_line[6:])  # 确保这里的切片索引是正确的
-	                    collected_data.append(json_line)
-	                    if json_line.get("msg") == "process_completed":
-	                        break
-	                except json.JSONDecodeError:
-	                    print("Error decoding JSON from response")
-	                    continue  # 跳过不能解码的行
-
-	        if not collected_data:
-	            return jsonify({"error": "No data received from the server", "word_count": 0})
-
-	        # 检查数据是否有效并尝试生成最终响应
-	        try:
-	            res_data = gen_res_data(collected_data[-1], time_now=time_now) if collected_data else {"error": "No valid data processed", "word_count": 0}
-	            print("Response data:", res_data)  # 打印输出res_data
-	        except KeyError as e:
-	            res_data = {"error": f"Missing key in data: {str(e)}", "word_count": 0}
-	            print("KeyError:", res_data)  # 错误情况下也打印res_data
-	        except IndexError as e:
-	            res_data = {"error": f"Index error in data: {str(e)}", "word_count": 0}
-	            print("IndexError:", res_data)
-	        except Exception as e:
-	            res_data = {"error": f"An unexpected error occurred: {str(e)}", "word_count": 0}
-	            print("Exception:", res_data)
-
-	    except Exception as e:
-	        res_data = {"error": f"An error occurred while processing stream: {str(e)}", "word_count": 0}
-	        print("Stream processing exception:", res_data)
-
-	    return jsonify(res_data)
+        response = requests.post(f"{base_url}/queue/join", json=json_prompt)
+        url = f"{base_url}/queue/data?session_hash={session_hash}"
+        data_stream = requests.get(url, stream=True)
+
+        time_now = int(time.time())
+        collected_data = []
+
+        try:
+            for line in data_stream.iter_lines():
+                if line:
+                    try:
+                        decoded_line = line.decode("utf-8")
+                        json_line = json.loads(decoded_line[6:])  # 确保这里的切片索引是正确的
+                        collected_data.append(json_line)
+                        if json_line.get("msg") == "process_completed":
+                            break
+                    except json.JSONDecodeError:
+                        print("Error decoding JSON from response")
+                        continue  # 跳过不能解码的行
+
+            if not collected_data:
+                return jsonify({"error": "No data received from the server", "word_count": 0})
+
+            # 检查数据是否有效并尝试生成最终响应
+            try:
+                res_data = gen_res_data(collected_data[-1], time_now=time_now) if collected_data else {"error": "No valid data processed", "word_count": 0}
+                print("Response data:", res_data)  # 打印输出res_data
+            except KeyError as e:
+                res_data = {"error": f"Missing key in data: {str(e)}", "word_count": 0}
+                print("KeyError:", res_data)  # 错误情况下也打印res_data
+            except IndexError as e:
+                res_data = {"error": f"Index error in data: {str(e)}", "word_count": 0}
+                print("IndexError:", res_data)
+            except Exception as e:
+                res_data = {"error": f"An unexpected error occurred: {str(e)}", "word_count": 0}
+                print("Exception:", res_data)
+
+        except Exception as e:
+            res_data = {"error": f"An error occurred while processing stream: {str(e)}", "word_count": 0}
+            print("Stream processing exception:", res_data)
+
+        return jsonify(res_data)
 
 
 def gen_res_data(data, time_now=0, start=False):