Update app.py

app.py

@@ -11,14 +11,14 @@ import gradio as gr
 import simplejson as json
 import oss2
 
-LOGGER = logging.getLogger('skybox')
-HANDLER = logging.handlers.RotatingFileHandler(
-    './logs/skybox.log', mode='a', maxBytes=20 * 1000 * 1000, backupCount=5, encoding='utf-8')
-FORMATTER = logging.Formatter('%(asctime)s - %(funcName)s - %(lineno)d - %(levelname)s - %(message)s')
-HANDLER.setFormatter(FORMATTER)
-LOGGER.propagate = 0
-LOGGER.addHandler(HANDLER)
-LOGGER.setLevel(logging.DEBUG)
+# LOGGER = logging.getLogger('skybox')
+# HANDLER = logging.handlers.RotatingFileHandler(
+#     './logs/skybox.log', mode='a', maxBytes=20 * 1000 * 1000, backupCount=5, encoding='utf-8')
+# FORMATTER = logging.Formatter('%(asctime)s - %(funcName)s - %(lineno)d - %(levelname)s - %(message)s')
+# HANDLER.setFormatter(FORMATTER)
+# LOGGER.propagate = 0
+# LOGGER.addHandler(HANDLER)
+# LOGGER.setLevel(logging.DEBUG)
 
 CSS = '.gradio-container a {color:#b7adf4 !important}'
 HEADERS = {'app-key': os.environ['a3'], 'origin': os.environ['a4'], 'referer': os.environ['a4']}
@@ -33,7 +33,7 @@ def login(loginId, password):
         if resp.status_code == 200:
             return resp.json()['data']['token']
     except Exception as e:
-        LOGGER.error(e)
+        # LOGGER.error(e)
         raise gr.Error('登录错误，请确认你的720yun账号是否正确')
 
 
@@ -46,7 +46,7 @@ def create_panorama(prompt, negative_prompt):
             'negative_text': negative_prompt.strip()[0:398]
         }
         resp = requests.post(os.environ['a1'], data=data)
-        LOGGER.info(resp.text)
+        # LOGGER.info(resp.text)
         if resp.status_code == 200:
             time.sleep(20)
             request_id = resp.json()['request']['id']
@@ -60,12 +60,12 @@ def create_panorama(prompt, negative_prompt):
                 resp = requests.get(f'{os.environ["a1"]}/{request_id}?api_key={os.environ["a2"]}')
                 if resp.status_code == 200:
                     progress_data = resp.json()['request']
-                    if progress_data['progress'] == 100 and progress_data['status'] == 'complete':
+                    if progress_data['status'] == 'complete':
                         flag = False
                         return request_id, progress_data['file_url']
                 time.sleep(1)
     except Exception as e:
-        LOGGER.error(e)
+        # LOGGER.error(e)
         raise gr.Error('AI老师罢工了，请稍后重试')
 
 
@@ -125,10 +125,11 @@ def upload(token, file_id, file_url):
                         'source': 99
                     }
                     resp = requests.post(os.environ['a8'], data=data, headers=HEADERS)
+                    # LOGGER.error(resp.text)
                     if resp.status_code == 200:
                         return resp.json()['data']['tid']
     except Exception as e:
-        LOGGER.error(e)
+        # LOGGER.error(e)
         raise gr.Error('AI老师罢工了，请稍后重试')