| import gradio as gr | |
| import visual | |
| import gen | |
| def showpuzzle(encoding): | |
| p1 = visual.parse(encoding) | |
| return visual.fig(p1) | |
| showpuzzint = gr.Interface(fn=showpuzzle, inputs=gr.Text(label="Puzzle encoding"), outputs=gr.Plot()) | |
| def genpuzzle(width,height,is_square,shuffle,seed): | |
| if is_square: | |
| height=width | |
| p = gen.genpuzzle(width,height,seed) | |
| if seed==0: | |
| seed=gen.seed1 | |
| if not shuffle: | |
| return visual.fig(p),seed,gen.h(p),'' | |
| p_ = gen.shuf(p) | |
| p2 = visual.join(p,p_) | |
| return visual.fig(p2),seed,gen.h(p),gen.h(p_) | |
| genint = gr.Interface(fn=genpuzzle, inputs=[gr.Slider(label="Width",step=1,minimum=1,maximum=40,value=3),gr.Slider(label="Height",step=1,minimum=1,maximum=40,value=3),gr.Checkbox(label="Square",value=True),gr.Checkbox(label="Shuffle"),gr.Number(label="Seed",value=0)], outputs=[gr.Plot(label="Puzzle"),gr.Number(label="Seed"),gr.Text(label="Puzzle encoding"),gr.Text(label="Shuffled encoding")]) | |
| def shuffle(encoding,show): | |
| p = visual.parse(encoding) | |
| p_ = gen.shuf(p) | |
| if show: | |
| return visual.fig(p_),gen.h(p_) | |
| return None,gen.h(p_) | |
| shufint = gr.Interface(fn=shuffle, inputs=[gr.Text(label="Puzzle encoding"),gr.Checkbox(label="Visualise")], outputs=[gr.Plot(label="Visualisation"),gr.Text(label="Shuffled encoding")]) | |
| demo = gr.TabbedInterface([genint,showpuzzint,shufint],["generate","visualise","shuffle"],title="comp2209 Tools") | |
| demo.launch() | |