Update app.py

app.py

@@ -13,39 +13,6 @@ from gradio.layouts import Column, Row
 from gradio.data_classes import InterfaceTypes
 import inspect
 
-
-
-# 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
-
-        with Column(variant="panel"):
-            input_component_column = Column()
-            with input_component_column:
-                for component in self.input_components:
-                    component.render()
-            with Row():
-                if self.interface_type in [InterfaceTypes.STANDARD, InterfaceTypes.INPUT_ONLY]:
-                    clear_btn = ClearButton(value="Effacer ❌")
-                    if not self.live:
-                        submit_btn = Button("Soumettez votre requête ✅", variant="primary")  # Custom label
-                        if inspect.isgeneratorfunction(self.fn) or inspect.isasyncgenfunction(self.fn):
-                            stop_btn = Button("Stop", variant="stop", visible=False)
-                elif self.interface_type == InterfaceTypes.UNIFIED:
-                    clear_btn = ClearButton(value="Effacer ❌")
-                    submit_btn = Button("Soumettez votre requête ✅", variant="primary")  # Custom label
-                    if (inspect.isgeneratorfunction(self.fn) or inspect.isasyncgenfunction(self.fn)) and not self.live:
-                        stop_btn = Button("Stop", variant="stop")
-                    if self.allow_flagging == "manual":
-                        flag_btns = self.render_flag_btns()
-                    elif self.allow_flagging == "auto":
-                        flag_btns = [submit_btn]
-        
-        # 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 +25,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.",
@@ -118,24 +84,20 @@ class MistralChatBot:
 # 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,29 +140,97 @@ 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."
+}
 
-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
-        minimum=0.05,
-        maximum=1.0,
-        step=0.05,
-        interactive=True,
-        info="Des valeurs plus élevées donne plus de créativité, mais aussi d'étrangeté",
-    ),
-]
-
-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()
+def fill_example(data):
+    question, style, temp = data
+    user_input.update(question)
+    dropdown.change(style)
+    temperature_slider.change(temp)
+    return "Il y avait eu dans le salon de l'hôtel, que j'avais quitté pour venir à la table où il se trouvait M. de Charlus et une foule d'autres personnes, une très grande madeleine, un peu grasse, mais dont la forme était fort belle, et qui avait été réservée pour moi ; elle avait été posée sur une assiette en porcelaine blanche avec des fleurs bleues dessinées dessus par un graveur français du XVIIIe siècle : cela faisait l'effet d'une madeleine faite de pâte rosée au beurre et décorée de motifs de couture en passementerie rose, sous lesquels on voyait cette couleur donner son éclat aux feuilles d'acanthe. Je pensai donc que je devrais demander à la femme de chambre si elle n'y prenait point goût, si elle ne voudrait pas bien manger ma madeleine ; je lui fis signe, après avoir pris mon cafard, et aussitôt elle s'approcha du plat, fit une pièce, en me porta la tranche avec ses doigts, comme si elle en avait fait autant de fois. Elle y mangea avec plaisir et sans gêne, à peine respectant mes yeux."
+
+
+# Function to return the explanation based on the selected artist or style
+def update_explanation(choice):
+    return explanations.get(choice, "No explanation available.")
+
+# Define the Gradio interface using Blocks
+with gr.Blocks(theme = 'carlosrosash/bellay_test') as demo:
+    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)
+
+    # Define the columns for input and output
+    with gr.Row():
+        with gr.Column():
+            user_input = gr.Textbox(label="Question or your instruction", type="text", lines=5, placeholder="Start typing here")
+            temperature_slider = gr.Slider(label="Temperature", value=0.7, minimum=0.05, maximum=1.0, step=0.05, interactive=True)
+            with gr.Row():
+                with gr.Column():
+                    clear_button = gr.Button("Clear")
+                with gr.Column():                    
+                    submit_button = gr.Button("Submit")
+        with gr.Column():
+            output_text = gr.Textbox(label="Response", type="text", lines=10)
+
+    # Function to handle the button click for submit
+    def on_submit(user_message, style, temperature):
+        return mistral_bot.predict(user_message, style, temperature)
+
+    # Function to handle the button click for clear
+    def on_clear():
+        user_input.clear()  # Clear the user input
+        temperature_slider.change(0.7)  # Reset the slider to default
+        output_text.clear()  # Clear the output text
+
+        # Reset the dropdown to a default value and update the explanation
+        default_style = list_dropdown[0]  # Assuming the first value is the default
+        dropdown.change(default_style)  # Change the dropdown to the default value
+        explanation_box.update(explanations[default_style])  # Update the explanation box
+
+    # Examples section
+    gr.Markdown("<h2>Examples</h2>")
+    examples_table = gr.Table(headers=["Question ou votre instruction", "Style", "Température"], 
+                              values=[["Comment faire une bonne madeleine ? Donne-moi une recette détaillée avec les ingrédients.", "Proust", 0.7]],
+                              interactive=True, type="array")
+    examples_table.click(fill_example, inputs=examples_table, outputs=output_text)
+
+
+    submit_button.click(on_submit, inputs=[user_input, dropdown, temperature_slider], outputs=output_text)
+    clear_button.click(on_clear)
+
+demo.launch()