Leetmonkey In Action. Darn LeetMonkey these days

app.py

@@ -89,20 +89,18 @@ 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(btn):
-    model_name = gr.Dropdown.update(choices=list(gguf_models.keys()), label="Select GGUF Model")
+def gradio_interface(model_name):
+    model = load_model(gguf_models[model_name])
     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, model_name
+    return instruction, python_code
 
 with gr.Blocks() as demo:
-    gr.Markdown("# LeetMonkey Problem Solver")
+    gr.Markdown("# LeetCode Problem Solver")
     
     with gr.Row():
         with gr.Column():
@@ -112,18 +110,18 @@ with gr.Blocks() as demo:
         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
-    )
+    def update_problem():
+        sample = random.choice(val_dataset)
+        return sample['instruction']
+    
+    def update_solution(problem, model_name):
+        model = load_model(gguf_models[model_name])
+        generated_output = generate_solution(problem, model)
+        return extract_and_format_code(generated_output)
     
-    generate_btn.click(
-        gradio_interface,
-        inputs=[model_dropdown, problem_display],
-        outputs=[problem_display, solution_display, original_solution_display, model_dropdown]
-    )
+    select_problem_btn.click(update_problem, outputs=problem_display)
+    generate_btn.click(update_solution, inputs=[problem_display, model_dropdown], outputs=solution_display)
 
-demo.launch(share=True)
+demo.launch(share=True)