Leetmonkey In Action. Darn LeetMonkey these days

app.py

@@ -89,27 +89,41 @@ def extract_and_format_code(text):
 dataset = load_dataset("sugiv/leetmonkey_python_dataset")
 val_dataset = dataset["train"].train_test_split(test_size=0.1)["test"]
 
-def gradio_interface(model_name):
-    model = load_model(gguf_models[model_name])
+def gradio_interface(btn):
+    model_name = gr.Dropdown.update(choices=list(gguf_models.keys()), label="Select GGUF Model")
     sample = random.choice(val_dataset)
     instruction = sample['instruction']
     original_output = sample['output']
     
+    model = load_model(gguf_models[model_name])
     generated_output = generate_solution(instruction, model)
     python_code = extract_and_format_code(generated_output)
     
-    return instruction, python_code, original_output
-
-iface = gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Dropdown(choices=list(gguf_models.keys()), label="Select GGUF Model"),
-    outputs=[
-        gr.Textbox(label="LeetCode Problem", lines=10),
-        gr.Code(label="Generated Solution", language="python"),
-        gr.Code(label="Original Solution", language="python")
-    ],
-    title="LeetCode Problem Solver",
-    description="Select a model to generate a solution for a random LeetCode problem."
-)
-
-iface.launch(share=True)
+    return instruction, python_code, original_output, model_name
+
+with gr.Blocks() as demo:
+    gr.Markdown("# LeetMonkey Problem Solver")
+    
+    with gr.Row():
+        with gr.Column():
+            problem_display = gr.Textbox(label="LeetCode Problem", lines=10)
+            select_problem_btn = gr.Button("Select Random Problem")
+        
+        with gr.Column():
+            model_dropdown = gr.Dropdown(choices=list(gguf_models.keys()), label="Select GGUF Model", value="Exact Copy")
+            solution_display = gr.Code(label="Generated Solution", language="python")
+            original_solution_display = gr.Code(label="Original Solution", language="python")
+            generate_btn = gr.Button("Generate Solution")
+    
+    select_problem_btn.click(
+        lambda: random.choice(val_dataset)['instruction'],
+        outputs=problem_display
+    )
+    
+    generate_btn.click(
+        gradio_interface,
+        inputs=[model_dropdown, problem_display],
+        outputs=[problem_display, solution_display, original_solution_display, model_dropdown]
+    )
+
+demo.launch(share=True)