Update pages/Visualization.py

pages/Visualization.py

@@ -15,11 +15,13 @@ 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:
-        time.sleep(3)
-        response = requests.post(API_URL_img, headers=headers, json=payload)
-    else:       
-        return response.content
+    while True:
+        if 'error' in body:
+            time.sleep(3)
+            response = requests.post(API_URL_img, headers=headers, json=payload)
+        else: 
+            break    
+    return response.content
 
 
 st.markdown('# :female-student: Персональный помощник для студентов')