update

app.py

@@ -1,13 +1,136 @@
+#!/usr/local/bin/python3
+# -*- coding: utf-8 -*-
+
+import time
+import os
+import requests
 import gradio as gr
+import simplejson as json
+import oss2
+
+CSS = '.gradio-container a {color:#b7adf4 !important}'
+HEADERS = {'app-key': os.environ['a3'], 'origin': os.environ['a4'], 'referer': os.environ['a4']}
+
+
+def login(loginId, password):
+    data = {'loginId': loginId, 'password': password}
+    cookies = {'720yun_v8_session': os.environ['a3']}
+
+    resp = requests.post(os.environ['a5'], data=data, headers=HEADERS, cookies=cookies)
+
+    if resp.status_code == 200:
+        return resp.json()['data']['token']
+    raise gr.Error('Error')
+
+
+def create_panorama(prompt, negative_prompt):
+    data = {'api_key': os.environ['a2'], 'generator': 'stable-skybox-trt', 'prompt': prompt, 'negative_text': negative_prompt}
+    resp = requests.post(os.environ['a1'], data=data)
+
+    if resp.status_code == 200:
+        request_id = resp.json()['request']['id']
+        flag = True
+        while flag:
+            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':
+                    flag = False
+                    return request_id, progress_data['file_url']
+            time.sleep(1)
+
+
+def upload(token, file_id, file_url):
+    now = int(time.time())
+    data = [{
+        'fileId': f'{file_id}-{now}',
+        'name': f'AI素材-{now}',
+        'watermarked': 0,
+        'size': 1024 * 1024 * 6,
+        'albumId': 0,
+        'exif': {},
+        'gps': {},
+        'panoId': 0,
+        'action': 1
+    }]
+    HEADERS['app-authorization'] = token
+    resp = requests.post(os.environ['a6'], data={'panos': json.dumps(data)}, headers=HEADERS)
+    if resp.status_code == 200:
+        resp_data = resp.json()['data'][0]
+        access_key_id = resp_data['accessKeyId']
+        security_token = resp_data['securityToken']
+        accessKey_secret = resp_data['accessKeySecret']
+        bucket_name = resp_data['bucketName']
+        path = resp_data['path'][1:]
+        endpoint = resp_data['endpointO']
+        pano_id = resp_data['panoId']
+        task_id = resp_data['taskId']
+        expired = resp_data['expired']
+
+        auth = oss2.StsAuth(access_key_id, accessKey_secret, security_token)
+        bucket = oss2.Bucket(auth, endpoint, bucket_name)
+
+        input_stream = requests.get(file_url)
+        result = bucket.put_object(f'{path}/{pano_id}.jpg', input_stream)
+        if result.status == 200:
+            resp = requests.post(f'{os.environ["a6"]}/{task_id}', data={'status': 3, 'expired': expired}, headers=HEADERS)
+            time.sleep(5)
+            if resp.status_code == 200:
+                flag = True
+                while flag:
+                    pano_ids = [pano['id'] for pano in requests.get(os.environ["a7"], headers=HEADERS).json()['data']]
+                    if pano_id not in pano_ids:
+                        flag = False
+                        break
+                    time.sleep(1)
+                data = {
+                    'name': f'AI全景作品-{now}',
+                    'materials': json.dumps([{
+                        'type': 1,
+                        'id': pano_id
+                    }]),
+                    'templateId': 2,
+                    'publishPlatform': 1,
+                    'keywords': 'AI全景',
+                    'source': 99
+                }
+                resp = requests.post(os.environ['a8'], data=data, headers=HEADERS)
+                if resp.status_code == 200:
+                    return resp.json()['data']['tid']
+
+
+def main(loginId, password, prompt, negative_prompt, state, progress=gr.Progress()):
+    if 'token' not in state:
+        state['token'] = login(loginId, password)
+    token = state['token']
+
+    file_id, image = create_panorama(prompt, negative_prompt)
+    panorama_id = upload(token, file_id, image)
+    return f'https://www.720yun.com/vr/{panorama_id}'
+
+
+with gr.Blocks(css=CSS) as demo:
+    session = gr.State({})
 
-description = "Story generation with GPT-2"
-title = "Generate your own story"
-examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
+    gr.Markdown("""
+        # Generate your own panorama with AI
+        Your **[720yun.com](https://www.720yun.com)** account is required.
+        Support by **[720yun.com](https://www.720yun.com)**.
+        Prompt example: a beautiful matte painting of northernmost continent, a gigantic square fortress covered by blizzard, texture smooth, aerial view, epic composition, post apocalyptic, sharp focus, sci-fi, futuristic, fantasy, clean background trending, by Jan Urschel and Sergey Vasnev and Emmanuel Shiu and Liam Wong and Michal Karcz, ornate, cinematic, cinematic lighting, light effect, epic, airy vibrant theme, octane render, unreal engine, concise and clear, 4k hd wallpaper, trending on artstation and cgsociety
+    """)
+    with gr.Row():
+        with gr.Column():
+            login_id = gr.Textbox(label='LoginId', placeholder='720yun loginId')
+        with gr.Column():
+            password = gr.Textbox(label='Password', type='password', placeholder='720yun password')
+    with gr.Row():
+        prompt = gr.Textbox(label='Prompt', lines=5, placeholder='Enter your prompt')
+    with gr.Row():
+        negative_prompt = gr.Textbox(label='Negative prompt', lines=2, placeholder='Enter your negative prompt')
+    with gr.Row():
+        out = gr.Textbox(label='Panorama Url')
 
-def greet(name):
-    return "Hello " + name + "!!"
+    btn = gr.Button('Run')
+    btn.click(fn=main, inputs=[login_id, password, prompt, negative_prompt, session], outputs=out, show_progress=True)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text",
-            description=description,
-            examples=examples)
-iface.launch()
+demo.queue().launch()

requirements.txt

@@ -1 +1,3 @@
-requests
+requests
+oss2
+simplejson