Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import ai | |
| import github_manager | |
| import webbrowser | |
| def main(): | |
| pre_prompt = """using pygame and no sprites, create: """ | |
| initial_prompt = """Conway's game of life, but the user controls one of the squares. If the square dies, the game is over.""" | |
| add_example = False | |
| post_prompt = "" | |
| if add_example: | |
| example_code_file_path = "resources/main.py" | |
| with open(example_code_file_path, "r") as file: | |
| example_code = file.read() | |
| post_prompt = example_code | |
| with gr.Blocks() as app: | |
| inp = gr.Textbox(label="Input Prompt", | |
| value=initial_prompt) | |
| with gr.Row(): | |
| model_choice = gr.Dropdown(label="Select Model", | |
| choices=[m for m in ai.models], | |
| value=ai.models[0], | |
| type="index", | |
| interactive=True) | |
| no_tokens = gr.Number(label="Tokens", value=1000) | |
| btn = gr.Button(value="Generate Game") | |
| with gr.Column(visible=False) as save_col: | |
| save_button = gr.Button("Build The Game") | |
| with gr.Column(visible=False) as play_col: | |
| play_button = gr.Button("Play The Game") | |
| out = gr.Code(label="Generated Code") | |
| def generate(prompt, model_index, max_tokens): | |
| final_prompt = f"""{pre_prompt} {prompt}. {post_prompt}""" | |
| generated_code = ai.generate_code(prompt=final_prompt, | |
| model_index=model_index, | |
| max_tokens=max_tokens) | |
| return { | |
| out: generated_code, | |
| save_col: gr.Column(visible=True), | |
| } | |
| def save(text): | |
| github_manager.update_repo(text) | |
| return { | |
| play_col: gr.Column(visible=True) | |
| } | |
| def play(): | |
| webbrowser.open(github_manager.game_link()) | |
| btn.click(generate, inputs=[inp, model_choice, no_tokens], outputs=[out, save_col], queue=True) | |
| save_button.click(save, inputs=[out], outputs=[play_col]) | |
| play_button.click(play, inputs=[], outputs=[]) | |
| app.launch(share=True) | |
| if __name__ == "__main__": | |
| main() | |