Update app.py

app.py

@@ -13,9 +13,9 @@ from gradio.layouts import Column, Row
 from gradio.data_classes import InterfaceTypes
 import inspect
 
+########################################################################################################################################################################
+#### Create the custom Interface for the buttoms
 
-
-# CustomInterface class definition
 class CustomInterface(gr.Interface):
     def render_input_column(self):
         submit_btn, clear_btn, stop_btn, flag_btns, duplicate_btn = None, None, None, None, None
@@ -45,7 +45,7 @@ class CustomInterface(gr.Interface):
         # Return the appropriate tuple
         return submit_btn, clear_btn, stop_btn, flag_btns, input_component_column, duplicate_btn, None
 
-
+########################################################################################################################################################################
 # Define the device
 device = "cuda" if torch.cuda.is_available() else "cpu"
 #Define variables 
@@ -58,7 +58,6 @@ model_name = "Pclanglais/Bellay"
 
 llm = LLM(model_name)
 
-
 styles_prompts_dict = {
     "Chansons de gestes": "Tu es Bellay, le troubadour du moyen-âge. Répond à la manière d’une belle chanson de geste.",
     "Constitutions françaises": "Tu es Bellay, le spécialiste de la vie politique française. Répond dans le style d’une constitution.",
@@ -86,9 +85,9 @@ styles_prompts_dict = {
     "Encyclopédie": "Tu es Diderot, l’encyclopédiste du siècle des lumières. Répond dans un style exhaustif et détaillé.",
     "Romans classiques": "Tu es Bellay, le romancier français classique. Répond sous la forme d’un roman un peu archaïque.",
     "Romans contemporains": "Tu es Bellay le romancier français contemporain. Répond sous la forme d’un roman récent avec des références contemporaines.",
-    "Série noire": "Tu es Bellay l’auteur de roman policier. Répond sous la forme d’un style hardboiled à la manière de la série noire.",
+    "Série noire": "Tu es Bellay l’auteur de roman policier. Répond sous la forme d’un style hardboiled à la manière de la série noire.",
     "Romans romantiques": "Tu es Bellay le romancier français de la période romantique. Répond sous la forme d’une fiction inspirée.",
-    "Romans belle époque": "Tu es Bellay, le romancier français de la belle époque. Répond avec un style un peu introspectif et moderniste.",
+    "Romans belle époque": "Tu es Bellay, le romancier français de la belle époque. Répond avec un style un peu introspectif et moderniste.",
     "San Antonio": "Tu es l’auteur des romans San-Antonio. Répond avec un style un peu cru et noir.",
     "Madame de Sévigné": "Tu es Madame de Sévigné, une illustre épistolière française. Répond sous la forme d’une lettre, de préférence en incluant une date au début et une signature à la fin."
 }
@@ -98,9 +97,9 @@ class MistralChatBot:
     def __init__(self, system_prompt="Le dialogue suivant est une conversation"):
         self.system_prompt = system_prompt
 
-    def predict(self, user_message, style, temperature):
+    def predict(self, user_message, selected_style, temperature):
 
-        system_prompt = styles_prompts_dict[style]
+        system_prompt = styles_prompts_dict[selected_style]
 
         sampling_params = SamplingParams(temperature=temperature, top_p=.95, max_tokens=500, presence_penalty = 2)
         detailed_prompt = "<|im_start|>system\n" + system_prompt + "<|im_end|>\n<|im_start|>user"""
@@ -110,32 +109,31 @@ class MistralChatBot:
         generated_text = outputs[0].outputs[0].text
         return generated_text
 
-    def predict_simple(self, user_message, style):
-        system_prompt = styles_prompts_dict[style]
+    def predict_simple(self, user_message, selected_style):
+        system_prompt = styles_prompts_dict[selected_style]
         conversation = "<|im_start|>system\n" + system_prompt + "<|im_end|>\n<|im_start|>user\n" + user_message + "<|im_end|>\n<|im_start|>assistant\n"
         return conversation
 
 # Create the Mistral chatbot instance
 mistral_bot = MistralChatBot()
 
+########################################################################################################################################################################
+
 # Define the Gradio interface
-title = "Bellay, le premier LLM de la littérature française"
+title = """
+<h1 style='text-align: center; font-weight: bold;'> Bellay, le premier LLM de la littérature française</h1>
+"""
 description = """
-🌟 Bienvenue dans l'univers de Bellay! 🌟
-
-📚 Choisissez votre style ou personnage 🎭
-
-Avec Bellay, embarquez pour un voyage littéraire unique! Sélectionnez l'un des styles fascinants ou transformez-vous en une des personnages emblématiques de Bellay. Êtes-vous prêt à explorer? Voici quelques options pour commencer:
-
-Proust - Plongez dans l'élégance et la complexité de Proust. 🖋️
-La Fontaine - Laissez-vous emporter par le charme des fables de La Fontaine. 🐢🦊
-Zola - Découvrez le réalisme poignant et captivant de Zola. 🏭🌆
-Montaigne - Engagez-vous dans les réflexions profondes de Montaigne. 🤔💭
-... et bien d'autres encore!
-
-🤖 Interagissez avec Bellay 🎙️
-
-Posez vos questions, lancez des discussions ou demandez des conseils à Bellay. Il est là pour répondre avec l'esprit et le style du personnage ou de l'époque que vous avez choisi. N'hésitez pas, lancez-vous dans cette aventure littéraire passionnante!
+<p style='text-align: justify;'>
+Lorem ipsum dolor sit amet, consectetur adipiscing elit. Vivamus elementum leo a
+ tortor fringilla elementum. Nullam eget sodales lacus. Nullam non aliquet elit. 
+ Etiam enim arcu, vehicula nec enim id, laoreet sodales justo. Nulla feugiat non 
+ leo in interdum. Quisque id leo ligula. Etiam id quam urna. Morbi tincidunt eros 
+ in eros vulputate, a congue enim feugiat. Ut laoreet ex in dui ultrices elementum. 
+ Vivamus eget venenatis dui. Vestibulum dictum ipsum eu mauris convallis, sit amet 
+ sollicitudin orci lacinia. Nunc pellentesque hendrerit felis eleifend interdum. 
+ Ut ac volutpat nisi. 
+</p>
 """
 examples = [
     [
@@ -178,9 +176,46 @@ list_dropdown = [
     'Romans contemporains'
 ]
 
+# Dictionary of explanations for each artist or style
+explanations = {
+    'Proust': "Marcel Proust, known for his introspective and lengthy novel 'In Search of Lost Time', which explores themes of memory and time.",
+    'Encyclopédie': "A groundbreaking Enlightenment-era work, edited by Diderot and d'Alembert, aiming to compile and disseminate human knowledge.",
+    'Balzac': "Honoré de Balzac, a French novelist and playwright, famous for his detailed depiction of society in 'La Comédie Humaine'.",
+    'Montaigne': "Michel de Montaigne, a Renaissance philosopher known for popularizing the essay as a literary genre, exploring human nature.",
+    'Foucault': "Michel Foucault, a 20th-century philosopher and social theorist, known for his analysis of power and societal institutions.",
+    'Chansons de gestes': "Medieval epic poems celebrating heroic deeds, exemplified by 'The Song of Roland', key to French literary history.",
+    'Constitutions françaises': "Refers to the various constitutions that France has had throughout its history, each shaping its political landscape.",
+    'La Fontaine': "Jean de La Fontaine, famed for his Fables, which are moralistic stories often featuring animals and conveying ethical lessons.",
+    'Madame de Sévigné': "Marie de Rabutin-Chantal, Marquise de Sévigné, known for her detailed and insightful letters depicting French aristocratic life.",
+    'Molière': "Jean-Baptiste Poquelin, known as Molière, a playwright and actor, acclaimed for his comedies like 'Tartuffe' and 'The Misanthrope'.",
+    'Racine': "Jean Racine, a playwright of the French classical era, renowned for his tragedies like 'Phèdre' and 'Andromaque'.",
+    'Romans classiques': "Refers to classic French novels, characterized by their adherence to traditional literary standards and themes.",
+    'Romans romantiques': "Romantic-era French novels, emphasizing emotion, nature, and individualism, contrasting with the rationality of Enlightenment.",
+    'Théâtre absurde': "Absurdist theatre, a post-World War II movement exemplified by playwrights like Samuel Beckett, emphasizing the absurdity of human existence.",
+    'Romans belle époque': "Novels from the Belle Époque, a period of French history marked by optimism, cultural innovations, and artistic flourishing.",
+    'Bible Segond': "A French translation of the Bible by Louis Segond, widely used in French-speaking Protestant communities.",
+    'Zola': "Émile Zola, a key figure in the literary school of naturalism, known for his series of novels 'Les Rougon-Macquart'.",
+    'Pagnol': "Marcel Pagnol, a novelist and filmmaker, celebrated for his works depicting life in Provence, including 'Jean de Florette'.",
+    'Philosophie': "Refers to French philosophy, a rich tradition ranging from existentialism to post-structuralism, featuring thinkers like Sartre and Derrida.",
+    'Poésies': "French poetry, a diverse field ranging from the troubadours of the Middle Ages to the symbolists and modern poets.",
+    'De Gaulle': "Charles de Gaulle, a French general and statesman, pivotal in establishing the Fifth Republic and its constitution.",
+    'Série noire': "A French publishing imprint known for its translations of American hardboiled detective and noir crime fiction.",
+    'San Antonio': "A series of French detective novels by Frédéric Dard, featuring the character San Antonio, known for their humor and inventiveness.",
+    'Duras': "Marguerite Duras, a French writer and film director, known for her minimalist style and works like 'The Lover'.",
+    'Presse années 1970': "Refers to the French press in the 1970s, a period marked by political activism and the rise of new journalistic styles.",
+    'Braudel': "Fernand Braudel, a French historian who revolutionized the study of history by focusing on long-term social and geographic structures.",
+    'Bourdieu': "Pierre Bourdieu, a French sociologist, renowned for his theory of social practices and concepts like cultural capital.",
+    'Derrida': "Jacques Derrida, a philosopher best known for developing deconstruction, a critical approach that questions the relationship between text and meaning.",
+    'Baudrillard': "Jean Baudrillard, a cultural theorist known for his analysis of consumerism, simulation, and hyperreality."
+}
+
+# Function to return the explanation based on the selected artist or style
+def update_explanation(choice):
+    global selected_style
+    selected_style = choice
+    return explanations.get(choice, "No explanation available.")
 
 additional_inputs=[
-    gr.Dropdown(list_dropdown, value="Proust", label="Style", info="Le style prédominant de Bellay"),
     gr.Slider(
         label="Température",
         value=0.7,  # Default value
@@ -192,15 +227,26 @@ additional_inputs=[
     ),
 ]
 
-iface = CustomInterface(
-    fn=mistral_bot.predict,
-    title=title,
-    description=description,
-    examples=examples,
-    inputs=[gr.Textbox(label="Question ou votre instruction", type="text", lines=5, placeholder = "Commencer à écrire ici")] + additional_inputs,
-    outputs=[gr.Textbox(label="Réponse", type="text", lines=5)],
-    theme="carlosrosash/bellay_test"
-)
-
-# Launch the Gradio interface for the Mistral model
-iface.launch()
+# Use gradio.Blocks for a more dynamic interface
+with gr.Blocks() as demo:
+    # Block 1: Title, Description, Dropdown, Explanation
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Row():
+        dropdown = gr.Dropdown(list_dropdown, label="Choose an Artist or Style")
+        explanation_box = gr.Textbox(label="Explanation", type="text", lines=5, interactive=False)
+    dropdown.change(update_explanation, inputs=dropdown, outputs=explanation_box)
+
+    # Block 2: Embedding the Gradio Interface
+    iface = gr.Interface(
+        fn=mistral_bot.predict,
+        examples = examples,
+        inputs=[gr.Textbox(label="Question ou votre instruction", type="text", lines=5, placeholder="Commencer à écrire ici")] + additional_inputs,
+        outputs=gr.Textbox(label="Réponse", type="text", lines=5),
+        theme="carlosrosash/bellay_test",
+    )
+
+# Initialize the shared variable
+selected_style = "Proust"  # Default value or any value from your dropdown list
+
+demo.launch()