Spaces:

DarkDriftz
/

SuperMarioGPT

Runtime error

DarkDriftz commited on 6 days ago

Commit

41ed846

verified ·

1 Parent(s): 6f943ac

Upload 18 files

Files changed (2) hide show

app.py CHANGED Viewed

@@ -2,10 +2,10 @@ import gradio as gr
 import torch
 import uuid
 import spaces
-from MarioGPT.dataset import MarioDataset
-from MarioGPT.prompter import Prompter
-from MarioGPT.lm import MarioLM
-from MarioGPT.utils import view_level, convert_level_to_png
 from fastapi import FastAPI, HTTPException
 from fastapi.staticfiles import StaticFiles
@@ -51,7 +51,7 @@ def make_html_file(generated_level):
 <head>
     <meta charset="utf-8">
-    <title>MarioGPT</title>
     <script src="https://cjrtnc.leaningtech.com/20230216/loader.js"></script>
 </head>
@@ -126,7 +126,7 @@ with gr.Blocks().queue() as demo:
         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=True):
         temperature = gr.Slider(value=2.0, minimum=0.1, maximum=2.0, step=0.1, label="temperature: Increase these for more diverse, but lower quality, generations")
         level_size = gr.Slider(value=1399, minimum=100, maximum=2799, step=1, label="level_size")

 import torch
 import uuid
 import spaces
+from supermariogpt.dataset import MarioDataset
+from supermariogpt.prompter import Prompter
+from supermariogpt.lm import MarioLM
+from supermariogpt.utils import view_level, convert_level_to_png
 from fastapi import FastAPI, HTTPException
 from fastapi.staticfiles import StaticFiles
 <head>
     <meta charset="utf-8">
+    <title>supermariogpt</title>
     <script src="https://cjrtnc.leaningtech.com/20230216/loader.js"></script>
 </head>
         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.Slider(value=2.0, minimum=0.1, maximum=2.0, step=0.1, label="temperature: Increase these for more diverse, but lower quality, generations")
         level_size = gr.Slider(value=1399, minimum=100, maximum=2799, step=1, label="level_size")

mcp_server.py CHANGED Viewed

@@ -6,7 +6,7 @@ Provides MCP-compatible interface for generating Mario levels via HuggingChat
 import asyncio
 import logging
-from typing import Any, Optional
 from mcp.server import Server
 from mcp.types import Tool, TextContent, ImageContent, EmbeddedResource
 from pydantic import BaseModel, Field
@@ -59,7 +59,7 @@ class GenerateLevelParams(BaseModel):
     )
 @mcp_server.list_tools()
-async def list_tools() -> list[Tool]:
     """List available MCP tools"""
     return [
         Tool(
@@ -81,7 +81,7 @@ async def list_tools() -> list[Tool]:
     ]
 @mcp_server.call_tool()
-async def call_tool(name: str, arguments: Any) -> list[TextContent | ImageContent | EmbeddedResource]:
     """Handle tool calls"""
     if name == "generate_mario_level":
@@ -133,8 +133,7 @@ async def call_tool(name: str, arguments: Any) -> list[TextContent | ImageConten
                              f"Prompt: {params.prompt}\n"
                              f"Temperature: {params.temperature}\n"
                              f"Level size: {params.level_size}\n\n"
-                             f"Level representation:\n{level_text[:500]}..."
-                             if len(level_text) > 500 else level_text
                     ),
                     ImageContent(
                         type="image",

 import asyncio
 import logging
+from typing import Any, Optional, List, Union
 from mcp.server import Server
 from mcp.types import Tool, TextContent, ImageContent, EmbeddedResource
 from pydantic import BaseModel, Field
     )
 @mcp_server.list_tools()
+async def list_tools() -> List[Tool]:
     """List available MCP tools"""
     return [
         Tool(
     ]
 @mcp_server.call_tool()
+async def call_tool(name: str, arguments: Any) -> List[Union[TextContent, ImageContent, EmbeddedResource]]:
     """Handle tool calls"""
     if name == "generate_mario_level":
                              f"Prompt: {params.prompt}\n"
                              f"Temperature: {params.temperature}\n"
                              f"Level size: {params.level_size}\n\n"
+                             f"Level representation:\n{level_text[:500] + '...' if len(level_text) > 500 else level_text}"
                     ),
                     ImageContent(
                         type="image",