Spaces:

multimodalart
/

mariogpt

Running on Zero

App Files Files Community

multimodalart HF Staff commited on Feb 17, 2023

Commit

ee2e0b7

1 Parent(s): 6845a5c

Update app.py

Browse files

Files changed (1) hide show

app.py +21 -19

app.py CHANGED Viewed

@@ -7,25 +7,24 @@ from mario_gpt.prompter import Prompter
 from mario_gpt.lm import MarioLM
 from mario_gpt.utils import view_level, convert_level_to_png
-import os
-import subprocess
-from pyngrok import ngrok
 mario_lm = MarioLM()
 device = torch.device('cuda')
 mario_lm = mario_lm.to(device)
 TILE_DIR = "data/tiles"
-subprocess.Popen(["python3","-m","http.server","7861"])
-ngrok.set_auth_token(os.environ.get('NGROK_TOKEN'))
-http_tunnel = ngrok.connect(7861,bind_tls=True)
 def make_html_file(generated_level):
     level_text = f"""{'''
 '''.join(view_level(generated_level,mario_lm.tokenizer))}"""
     unique_id = uuid.uuid1()
-    with open(f"demo-{unique_id}.html", 'w', encoding='utf-8') as f:
         f.write(f'''<!DOCTYPE html>
 <html lang="en">
@@ -42,7 +41,7 @@ def make_html_file(generated_level):
         cheerpjAddStringFile("/str/mylevel.txt", `{level_text}`);
     }});
     cheerpjCreateDisplay(512, 500);
-    cheerpjRunJar("/app/mario.jar");
 </script>
 </html>''')
     return f"demo-{unique_id}.html"
@@ -61,9 +60,9 @@ def generate(pipes, enemies, blocks, elevation, temperature = 2.0, level_size =
     filename = make_html_file(generated_level)
     img = convert_level_to_png(generated_level.squeeze(), TILE_DIR, mario_lm.tokenizer)[0]
-    gradio_html = f'''<div style="border: 2px solid;">
-        <iframe width=512 height=512 style="margin: 0 auto" src="{http_tunnel.public_url}/{filename}"></iframe>
-        <p style="text-align:center">Press the arrow keys to move. Press <code>s</code> to jump and <code>a</code> to shoot flowers</p>
     </div>'''
     return [img, gradio_html]
@@ -72,16 +71,16 @@ with gr.Blocks() as demo:
     [[Github](https://github.com/shyamsn97/mario-gpt)], [[Paper](https://arxiv.org/abs/2302.05981)]
     ''')
     with gr.Tabs():
-        with gr.TabItem("Type prompt"):
-            text_prompt = gr.Textbox(value="", label="Enter your MarioGPT prompt. ex: 'many pipes, many enemies, some blocks, low elevation'")
         with gr.TabItem("Compose prompt"):
             with gr.Row():
-                pipes = gr.Radio(["no", "little", "some", "many"], label="pipes")
-                enemies = gr.Radio(["no", "little", "some", "many"], label="enemies")
             with gr.Row():
-                blocks = gr.Radio(["little", "some", "many"], label="blocks")
-                elevation = gr.Radio(["low", "high"], label="elevation")
     with gr.Accordion(label="Advanced settings", open=False):
         temperature = gr.Number(value=2.0, label="temperature: Increase these for more diverse, but lower quality, generations")
         level_size = gr.Number(value=1399, precision=0, label="level_size")
@@ -104,4 +103,7 @@ with gr.Blocks() as demo:
         fn=generate,
         cache_examples=True,
     )
-demo.launch()

 from mario_gpt.lm import MarioLM
 from mario_gpt.utils import view_level, convert_level_to_png
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+import os
+import uvicorn
 mario_lm = MarioLM()
 device = torch.device('cuda')
 mario_lm = mario_lm.to(device)
 TILE_DIR = "data/tiles"
+app = FastAPI()
 def make_html_file(generated_level):
     level_text = f"""{'''
 '''.join(view_level(generated_level,mario_lm.tokenizer))}"""
     unique_id = uuid.uuid1()
+    with open(f"static/demo-{unique_id}.html", 'w', encoding='utf-8') as f:
         f.write(f'''<!DOCTYPE html>
 <html lang="en">
         cheerpjAddStringFile("/str/mylevel.txt", `{level_text}`);
     }});
     cheerpjCreateDisplay(512, 500);
+    cheerpjRunJar("/app/static/mario.jar");
 </script>
 </html>''')
     return f"demo-{unique_id}.html"
     filename = make_html_file(generated_level)
     img = convert_level_to_png(generated_level.squeeze(), TILE_DIR, mario_lm.tokenizer)[0]
+    gradio_html = f'''<div>
+        <iframe width=512 height=512 style="margin: 0 auto" src="static/{filename}"></iframe>
+        <p style="text-align:center">Press the arrow keys to move. Press <code>a</code> to run, <code>s</code> to jump and <code>d</code> to shoot fireflowers</p>
     </div>'''
     return [img, gradio_html]
     [[Github](https://github.com/shyamsn97/mario-gpt)], [[Paper](https://arxiv.org/abs/2302.05981)]
     ''')
     with gr.Tabs():
         with gr.TabItem("Compose prompt"):
             with gr.Row():
+                pipes = gr.Radio(["no", "little", "some", "many"], label="How many pipes?")
+                enemies = gr.Radio(["no", "little", "some", "many"], label="How many enemies?")
             with gr.Row():
+                blocks = gr.Radio(["little", "some", "many"], label="How many blocks?")
+                elevation = gr.Radio(["low", "high"], label="Elevation?")
+        with gr.TabItem("Type prompt"):
+            text_prompt = gr.Textbox(value="", label="Enter your MarioGPT prompt. ex: 'many pipes, many enemies, some blocks, low elevation'")
     with gr.Accordion(label="Advanced settings", open=False):
         temperature = gr.Number(value=2.0, label="temperature: Increase these for more diverse, but lower quality, generations")
         level_size = gr.Number(value=1399, precision=0, label="level_size")
         fn=generate,
         cache_examples=True,
     )
+app.mount("/static", StaticFiles(directory="static", html=True), name="static")
+app = gr.mount_gradio_app(app, demo, "/", gradio_api_url="http://localhost:7860/")
+uvicorn.run(app, host="0.0.0.0", port=7860)