Update app.py

app.py

@@ -295,30 +295,34 @@ prompt_examples = [
     "A concept art of"
 ]
 
-# Prompt d'exemple visibles
-prompt_examples = prompt_list[:5]
+# Afficher TOUTE la liste dans la dropdown
+prompt_examples = prompt_list
 
-# Fonction de génération
 def generate(prompt, n):
     return "\n\n---\n\n".join(
         prompt + llm(prompt, max_tokens=200, stop=["</s>"], echo=False)["choices"][0]["text"]
         for _ in range(n)
     )
 
-# Fonction: randomise un prompt (sans générer)
 def pick_random_prompt():
     return random.choice(prompt_list)
 
-# Fonction: sélectionne dans la liste (sans générer)
 def pick_example_prompt(example):
     return example
 
-# Interface Gradio
+def update_button_labels(n):
+    suffix = "Prompt" if n == 1 else "Prompts"
+    return f"Generate {suffix}", f"🎲 Random {suffix}"
+
+# UI Gradio
 with gr.Blocks(title="WizzGPTv7 - Text Completion (GGUF - CPU)") as demo:
     gr.Markdown("## WizzGPTv7 - Text Completion (GGUF - CPU)")
 
     with gr.Row():
         prompt_box = gr.Textbox(lines=5, label="Prompt", placeholder="Enter your prompt here...")
+
+    with gr.Row():
+        generate_button = gr.Button("Generate Prompt")
         random_button = gr.Button("🎲 Random Prompt")
 
     with gr.Row():
@@ -326,15 +330,11 @@ with gr.Blocks(title="WizzGPTv7 - Text Completion (GGUF - CPU)") as demo:
         num_slider = gr.Slider(minimum=1, maximum=7, step=1, value=1, label="Number of Responses")
 
     output_box = gr.Textbox(label="Completions", lines=15)
-    submit_button = gr.Button("Submit")
 
-    # Génération manuelle
-    submit_button.click(generate, inputs=[prompt_box, num_slider], outputs=output_box)
-
-    # Random remplissage uniquement
+    # Fonctions reliées
+    generate_button.click(generate, inputs=[prompt_box, num_slider], outputs=output_box)
     random_button.click(fn=pick_random_prompt, outputs=prompt_box)
-
-    # Remplir via dropdown uniquement
     example_dropdown.change(fn=pick_example_prompt, inputs=example_dropdown, outputs=prompt_box)
+    num_slider.change(fn=update_button_labels, inputs=num_slider, outputs=[generate_button, random_button])
 
 demo.launch()