Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
+import random
 from llama_cpp import Llama
 
-# Charger le modèle GGUF (Q8_0)
+# Charger le modèle
 llm = Llama(
     model_path="model/WizzGPTv7-q8_0.gguf",
     n_threads=1,
@@ -10,14 +11,323 @@ llm = Llama(
     use_mlock=True
 )
 
-def generate(prompt):
-    output = llm(prompt, max_tokens=200, stop=["</s>"], echo=False)
-    return output["choices"][0]["text"]
+# Liste complète des débuts de prompt
+prompt_list = [
+    "A beautiful sunset over",
+    "A macro shot of",
+    "Award winning photography of",
+    "Wide shot of",
+    "The intricate details of",
+    "The portrait of a",
+    "A resin figurine of",
+    "Painting of",
+    "A delicate ",
+    "A rustic ",
+    "A stunningly beautiful",
+    "Masterpiece, best quality",
+    "Hyperrealistic shot of a",
+    "Professional fashion shot of",
+    "Photo of a classy",
+    "Photo of a glamour",
+    "(Horror) shot of",
+    "Cinematic film still of",
+    "A (monochrome:2) photography of",
+    "A ",
+    "Masterpiece, best quality, ultra-detailed, ",
+    "A hyper-realistic, cinematic shot of ",
+    "An incredibly detailed, high-resolution portrait of ",
+    "A breathtaking, photorealistic capture of ",
+    "A vivid, intricately detailed scene of ",
+    "A surreal, ultra-detailed rendering of ",
+    "An exquisitely detailed, cinematic masterpiece of ",
+    "A mesmerizing, hyper-detailed view of ",
+    "A stunning, ultra-realistic depiction of ",
+    "A visually arresting, high-definition snapshot of ",
+    "The ",
+    "The (hologram) of",
+    "A surrealist art of",
+    "The (photography of a __animal__:2), (full-body), ",
+    "The (photo of a (__animal__:__animal__:__weight__) hybrid), full-body,",
+    "A photo of an Eldritch abomination,",
+    "A (3D geometrically printed figurine) of",
+    "An overwhelmingly beautiful",
+    "An (eerie scene) featuring",
+    "A serene landscape showcasing",
+    "A photography of a magnificent ",
+    "A bokeh effect background with ",
+    "The bustling streets of",
+    "A whimsical depiction of",
+    "A dramatic aerial view of",
+    "An enchanting tableau presenting",
+    "The ancient ruins of",
+    "A night sky filled with",
+    "A macro close-up portrait of",
+    "An (abstract composition) with",
+    "A snowy scene with",
+    "The golden hour lighting up",
+    "A mystical forest where",
+    "A futuristic cityscape with",
+    "An ethereal ",
+    "A whimsical",
+    "Photo of an amazingly beautiful creature with",
+    "Photo of an iridescent ",
+    "Focus on ",
+    "A meticulously crafted",
+    "A dramatic chiaroscuro lighting of",
+    "An artistic interpretation of",
+    "A vivid depiction of",
+    "A serene moment captured in",
+    "A richly textured",
+    "An evocative scene portraying",
+    "An epic panorama of",
+    "An iconic representation of",
+    "A minimalist composition featuring",
+    "An avant-garde portrayal of",
+    "An immersive diorama of",
+    "A time-lapse sequence showing",
+    "A haunting silhouette against",
+    "An ancient mythological scene in",
+    "A contemporary reimagining of",
+    "A hyper-detailed rendering of",
+    "A nostalgic memory preserved in",
+    "A powerful juxtaposition of",
+    "A seamless blend of",
+    "A surreal",
+    "An evocative interplay of light and shadow",
+    "A preserved fragment of",
+    "An intricate lacework of",
+    "An offbeat take on",
+    "A reflective surface capturing",
+    "A slow-motion capture of",
+    "A stunning vista showcasing",
+    "A timeless black-and-white portrait of",
+    "A twisted version of",
+    "A dynamic motion blur effect in",
+    "A geometric pattern inspired by",
+    "A moody fog enveloping",
+    "A textured overlay on",
+    "A vibrant explosion of color",
+    "A carefully orchestrated symmetry in",
+    "A kaleidoscopic arrangement of",
+    "A subtle gradation of color in",
+    "The raw grandeur of",
+    "A whirlwind of movement in",
+    "A conceptual representation of",
+    "An experimental take on",
+    "A harmonious blend of",
+    "A poetic rendering of",
+    "An organic flow of",
+    "An understated elegance in",
+    "An unassuming grace in",
+    "A dynamic shift in",
+    "A gentle harmony of",
+    "A graceful curve in",
+    "A heightened sense of",
+    "A radiant glow surrounding",
+    "A resonant soundscape in visual form",
+    "A strong focal point in",
+    "An angular perspective on",
+    "A fleeting moment of",
+    "A playful interaction between",
+    "A stark minimalist design featuring",
+    "A whimsical journey through",
+    "An organic structure evolving from",
+    "A visual metaphor for",
+    "A window into another world through a Stable-Diffusion of",
+    "A delicate balance between",
+    "A seamless transition from",
+    "A tranquil scene",
+    "The vibrant colors of",
+    "A peaceful countryside morning in",
+    "An underwater world revealing",
+    "The photography of",
+    "A (2000s vintage RAW photo) of",
+    "Photoshoot of",
+    "A breathtaking",
+    "An extremely detailed",
+    "An establishing shot of",
+    "A concept art of",
+    "A breathtaking portrait of a majestic ",
+    "A charcoal sketch of",
+    "A dramatic scene of",
+    "A painting in the style of",
+    "An isolated",
+    "An (upper angle shot) of",
+    "An (aerial perspective photo) of a",
+    "A high contrast side light scene",
+    "A dark And mysterious ",
+    "A clay sculpture of ",
+    "A digital illustration featuring",
+    "A graffiti art piece depicting",
+    "A historical reenactment of ",
+    "A leading lines shot of ",
+    "A low angle view of ",
+    "A misty dawn over ",
+    "A mixed media collage showcasing",
+    "A panoramic view of ",
+    "A pastel portrait of ",
+    "A pencil drawing of ",
+    "A tight shot of ",
+    "A watercolor painting of ",
+    "A National Geographic wildlife photographer capturing",
+    "A worm's eye view of ",
+    "An ink drawing depicting",
+    "Masterpiece, ",
+    "A mesmerizing glimpse of ",
+    "A captivating display of ",
+    "A dynamic interplay of ",
+    "A sublime vision of ",
+    "A breathtaking portrayal of ",
+    "A poetic interpretation of ",
+    "A striking composition of ",
+    "A vivid tapestry of ",
+    "An enigmatic depiction of ",
+    "A remarkable fusion of ",
+    "A stunning perspective of ",
+    "A harmonious blend of ",
+    "A radical reinterpretation of ",
+    "A delicate sketch of ",
+    "A vibrant composition of ",
+    "A richly textured portrayal of ",
+    "A pioneering outlook of ",
+    "A refined portrait of ",
+    "A sharp focus on ",
+    "A monumental depiction of ",
+    "A fluid rendering of ",
+    "A kinetic expression of ",
+    "A graceful outline of ",
+    "A provocative framework of ",
+    "A subtle balance of ",
+    "A minimalist construction of ",
+    "A dramatic evolution of ",
+    "A whimsical abstraction of ",
+    "A poignant reflection of ",
+    "A pioneering illustration of ",
+    "A sophisticated visualization of ",
+    "An exuberant composition of ",
+    "A bold expression of ",
+    "A lavish interpretation of ",
+    "A masterful arrangement of ",
+    "A futuristic vision of ",
+    "A radiant convergence of ",
+    "A surrealist portrayal of ",
+    "A fragmented mosaic of ",
+    "A cryptic collage of ",
+    "A crystalline perspective on ",
+    "A dreamlike rendition of ",
+    "A kinetic study of ",
+    "A spontaneous capture of ",
+    "A reflective moment of ",
+    "An intense interplay of ",
+    "A layered narrative of ",
+    "A refined synthesis of ",
+    "A chiaroscuro play of ",
+    "A poetic exploration of ",
+    "A reimagined concept of ",
+    "A vigorous depiction of ",
+    "A textured cascade of ",
+    "A visionary abstraction of ",
+    "A monumental snapshot of ",
+    "A scintillating vignette of ",
+    "A graceful melding of ",
+    "A spectral composition of ",
+    "A radiant interplay of ",
+    "A striking juxtaposition of ",
+    "A refined montage of ",
+    "A nuanced portrayal of ",
+    "A visionary construct of ",
+    "An evocative snapshot of ",
+    "A bold fusion of ",
+    "A delicate overlay of ",
+    "A complex layering of ",
+    "A vivid gradient of ",
+    "A kinetic cascade of ",
+    "A luminous texture of ",
+    "A bold silhouette of ",
+    "A high-resolution photographic capture of ",
+    "A sharply focused photographic study of ",
+    "A meticulously framed photographic moment of ",
+    "An ultra-detailed photographic exploration of ",
+    "A dynamic high-contrast photographic depiction of ",
+    "A bold, hyper-realistic photographic rendering of ",
+    "An intimate photographic snapshot of ",
+    "A vividly composed photographic portrayal of ",
+    "A crisp, professional photographic impression of ",
+    "A timeless black-and-white photographic record of ",
+    "A vividly detailed photographic vignette of ",
+    "A dynamic, low-key photographic interpretation of ",
+    "A refined photographic montage of ",
+    "A compelling photographic narrative of ",
+    "A striking high-key photographic expression of ",
+    "A meticulously captured photographic scene of ",
+    "A visionary photographic perspective on ",
+    "An innovative photographic framing of ",
+    "A sophisticated photographic series of ",
+    "A powerful photographic study of ",
+    "A radiant photographic tableau of ",
+    "A minimalist photographic composition of ",
+    "An edgy photographic take on ",
+    "A vibrant photographic freeze-frame of ",
+    "A kinetic photographic capture of ",
+    "A meticulously arranged photographic display of ",
+    "A dramatic photographic snapshot of ",
+    "A crisp, color-rich photographic impression of ",
+    "A bold photographic experiment capturing ",
+    "An avant-garde photographic reinterpretation of ",
+    "A sharply rendered photographic moment of ",
+    "A compelling photographic sequence of ",
+    "A profound photographic observation of ",
+    "A refined photographic layering of ",
+    "A spontaneous photographic capture of ",
+    "A surreal photographic juxtaposition of ",
+    "A masterfully composed photographic glimpse of ",
+    "A radiant photographic slice of ",
+    "A detailed photographic capture of ",
+    "A visually arresting photographic study of "
+]
 
-# UI Gradio
-iface = gr.Interface(fn=generate, 
-                     inputs=gr.Textbox(lines=5, label="Prompt"),
-                     outputs=gr.Textbox(label="Answer"),
-                     title="WizzGPTv7 - CPU Demo")
+# Top 5 exemples à afficher dans le menu déroulant
+prompt_examples = [
+    "Award winning photography of",
+    "A breathtaking portrait of a majestic",
+    "A surreal, ultra-detailed rendering of",
+    "The golden hour lighting up",
+    "A concept art of"
+]
 
-iface.launch()
+# 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)
+    )
+
+# Random prompt
+def use_random_prompt(n):
+    prompt = random.choice(prompt_list)
+    return prompt, generate(prompt, n)
+
+# Sélection d’un exemple
+def use_example_prompt(example, n):
+    return example, generate(example, n)
+
+# Interface Gradio
+with gr.Blocks(title="WizzGPTv7 - CPU Demo") as demo:
+    gr.Markdown("## WizzGPTv7 - Text Completion (GGUF - CPU)")
+
+    with gr.Row():
+        prompt_box = gr.Textbox(lines=5, label="Prompt")
+        random_button = gr.Button("🎲 Random Prompt")
+    
+    with gr.Row():
+        example_dropdown = gr.Dropdown(choices=prompt_examples, label="📌 Prompt Examples")
+        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")
+
+    submit_button.click(generate, inputs=[prompt_box, num_slider], outputs=output_box)
+    random_button.click(use_random_prompt, inputs=[num_slider], outputs=[prompt_box, output_box])
+    example_dropdown.change(use_example_prompt, inputs=[example_dropdown, num_slider], outputs=[prompt_box, output_box])
+
+demo.launch()