Update app.py

app.py

@@ -10,27 +10,27 @@ import base64
 
 def flip_text(prompt, key):
 
-    #key_os = os.getenv("key_os")
+    key_os = os.getenv("key_os")
 
-    #if str(key).lower() == str(key_os).lower():
+    if str(key).lower() == str(key_os).lower():
     
-    print(prompt)
-    
-    headers = os.getenv("headers")
-    url_api = os.getenv("url_api")
-    
-    headers = eval(str(headers))
-    json_data = {'prompt': prompt, 'n': 1,'size': '1024x1024','response_format': 'b64_json','model': 'dall-e-3','quality': 'hd',}
-    response = requests.post(f"{url_api}", headers=headers, json=json_data)
-    js = response.json()
-    if "content_policy_violation" in str(js):
-        return "https://myneuralnetworks.ru/static/img/zp.jpg"
+        print(prompt)
+        
+        headers = os.getenv("headers")
+        url_api = os.getenv("url_api")
+        
+        headers = eval(str(headers))
+        json_data = {'prompt': prompt, 'n': 1,'size': '1024x1024','response_format': 'b64_json','model': 'dall-e-3','quality': 'hd',}
+        response = requests.post(f"{url_api}", headers=headers, json=json_data)
+        js = response.json()
+        if "content_policy_violation" in str(js):
+            return "https://myneuralnetworks.ru/static/img/zp.jpg"
+        else:
+            photo = response.json()['data'][0]['b64_json']
+            photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
+            return photo
     else:
-        photo = response.json()['data'][0]['b64_json']
-        photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
-        return photo
-    #else:
-    #    return None
+        return "https://myneuralnetworks.ru/static/img/zp2.jpg"
 
 
 css = """