update

app.py

@@ -4,11 +4,22 @@
 import uuid
 import time
 import os
+import logging
+import logging.handlers
 import requests
 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)
+
 CSS = '.gradio-container a {color:#b7adf4 !important}'
 HEADERS = {'app-key': os.environ['a3'], 'origin': os.environ['a4'], 'referer': os.environ['a4']}
 
@@ -21,7 +32,8 @@ def login(loginId, password):
         resp = requests.post(os.environ['a5'], data=data, headers=HEADERS, cookies=cookies)
         if resp.status_code == 200:
             return resp.json()['data']['token']
-    except Exception:
+    except Exception as e:
+        LOGGER.error(e)
         raise gr.Error('登录错误，请确认你的720yun账号是否正确')
 
 
@@ -29,14 +41,14 @@ def create_panorama(prompt, negative_prompt):
     try:
         data = {
             'api_key': os.environ['a2'],
-            'generator': 'stable-skybox-trt',
-            'prompt': prompt[0:598],
-            'negative_text': negative_prompt[0:398]
+            'generator': 'stable-skybox',
+            'prompt': prompt.strip()[0:598],
+            'negative_text': negative_prompt.strip()[0:398]
         }
         resp = requests.post(os.environ['a1'], data=data)
-        print(resp.text)
+        LOGGER.info(resp.text)
         if resp.status_code == 200:
-            time.sleep(20)
+            time.sleep(15)
             request_id = resp.json()['request']['id']
             flag = True
             countdown = 50
@@ -51,8 +63,9 @@ def create_panorama(prompt, negative_prompt):
                     if progress_data['progress'] == 100 and progress_data['status'] == 'complete':
                         flag = False
                         return request_id, progress_data['file_url']
-                time.sleep(0.5)
-    except Exception:
+                time.sleep(1)
+    except Exception as e:
+        LOGGER.error(e)
         raise gr.Error('AI老师罢工了，请稍后重试')
 
 
@@ -115,7 +128,8 @@ def upload(token, file_id, file_url):
                     resp = requests.post(os.environ['a8'], data=data, headers=HEADERS)
                     if resp.status_code == 200:
                         return resp.json()['data']['tid']
-    except Exception:
+    except Exception as e:
+        LOGGER.error(e)
         raise gr.Error('AI老师罢工了，请稍后重试')
 
 
@@ -153,7 +167,7 @@ with gr.Blocks(css=CSS) as demo:
     with gr.Row():
         out = gr.Textbox(label='输出')
 
-    btn = gr.Button('Run')
+    btn = gr.Button('运行')
     btn.click(fn=main, inputs=[login_id, password, prompt, negative_prompt, session], outputs=out, show_progress=True)
 
-demo.queue(concurrency_count=3).launch()
+demo.queue(concurrency_count=8).launch()