Update app.py

app.py

@@ -334,22 +334,27 @@ def on_model_change(model_name):
     return f"Model loaded: {model_name}"
 
 # Interface Gradio
-with gr.Blocks(css=".gr-button { font-size: 16px; }") as demo:
+with gr.Blocks(css="""
+.gr-button { font-size: 15px; padding: 0.6em 1.2em; }
+.gr-textbox textarea { font-size: 15px; }
+""") as demo:
+
     gr.Markdown("## ✨ WizzGPT — Text Completion (GGUF - CPU)")
 
     with gr.Row():
-        model_selector = gr.Dropdown(choices=list(model_versions.keys()), value="WizzGPTv7", label="🧠 Select Model")
+        model_selector = gr.Dropdown(choices=list(model_versions.keys()), value="WizzGPTv7", label="🧠 Model")
         model_status = gr.Textbox(value=f"Model loaded: {current_model}", label="Status", interactive=False)
 
-    with gr.Group():
-        with gr.Row():
-            prompt_input = gr.Textbox(label="📝 Prompt", lines=3, placeholder="Type or pick a prompt...")
-        with gr.Row():
-            prompt_dropdown = gr.Dropdown(choices=prompt_list, label="📌 Prompt Examples", scale=2)
-            random_button = gr.Button("🎲 Random Prompt", scale=1)
+    with gr.Row():
+        prompt_input = gr.Textbox(label="📝 Prompt", lines=2, placeholder="Type or select a prompt...")
 
     with gr.Row():
-        n_responses = gr.Slider(1, 7, value=1, step=1, label="Number of Completions")
+        prompt_dropdown = gr.Dropdown(choices=prompt_list, label="📌 Examples")
+        random_button = gr.Button("🎲", size="sm", elem_id="random-btn", min_width=30)
+        generate_button = gr.Button("🚀", size="sm", elem_id="generate-btn")
+
+    with gr.Row():
+        n_responses = gr.Slider(1, 7, value=1, step=1, label="Number")
 
     with gr.Accordion("⚙️ Advanced Settings", open=False):
         max_tokens = gr.Slider(10, 200, value=75, step=5, label="Max Tokens")
@@ -358,15 +363,13 @@ with gr.Blocks(css=".gr-button { font-size: 16px; }") as demo:
         top_k = gr.Slider(0, 128, value=128, step=1, label="Top-k")
         repeat_penalty = gr.Slider(0.5, 2.0, value=1.4, step=0.05, label="Repeat Penalty")
 
-    generate_button = gr.Button("🚀 Generate Prompt(s)")
-    output = gr.Textbox(label="📄 Completions", lines=12)
+    output = gr.Textbox(label="📄 Completions", lines=10)
 
-    # Interactions
+    # Event bindings
     prompt_dropdown.change(set_prompt_from_dropdown, [prompt_dropdown, n_responses], [prompt_input, random_button, generate_button])
     random_button.click(set_random_prompt, inputs=n_responses, outputs=[prompt_input, random_button, generate_button])
+    generate_button.click(generate_completions, inputs=[prompt_input, n_responses, max_tokens, temperature, top_p, top_k, repeat_penalty], outputs=output)
     n_responses.change(lambda n: (f"🎲 Random Prompt{'s' if n > 1 else ''}", f"🚀 Generate Prompt{'s' if n > 1 else ''}"), inputs=n_responses, outputs=[random_button, generate_button])
-    generate_button.click(fn=generate_completions, inputs=[prompt_input, n_responses, max_tokens, temperature, top_p, top_k, repeat_penalty], outputs=output)
-
     model_selector.change(on_model_change, inputs=model_selector, outputs=model_status)
 
 demo.launch()