Update pages/Visualization.py

pages/Visualization.py

@@ -13,16 +13,14 @@ headers = {"Authorization": os.getenv("api_token")}
 
 # Функция генерации изображения
 def generate_img(payload):
-    response = requests.post(API_URL_img, headers=headers, json=payload)
-    body = response.json()
-    if 'error' in body:
-        if 'estimated_time' in body:
-            time.sleep(body['estimated_time'])
-        else:
-            print(body)
-            return
+    try:
+        response = requests.post(API_URL_img, headers=headers, json=payload)
+    except json.JSONDecodeError as e:
+        print("Ошибка декодирования JSON:", e)
+        time.sleep(3)
         generate_img(payload)
-    return body
+    else:
+        return response.content
 
 
 st.markdown('# :female-student: Персональный помощник для студентов')