Update app.py

app.py

@@ -13,6 +13,39 @@ 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 
@@ -25,6 +58,7 @@ 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.",
@@ -84,20 +118,24 @@ class MistralChatBot:
 # Create the Mistral chatbot instance
 mistral_bot = MistralChatBot()
 
-title = """
-<h1 style='text-align: center; font-weight: bold;'> Bellay, le premier LLM de la littérature française</h1>
-"""
+# Define the Gradio interface
+title = "Bellay, le premier LLM de la littérature française"
 description = """
-<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>
+🌟 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!
 """
 examples = [
     [
@@ -140,76 +178,29 @@ 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):
-    return explanations.get(choice, "No explanation available.")
-
-# Define the Gradio interface using Blocks
-with gr.Blocks() 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.Column():
-                clear_button = gr.Button("Clear")
-                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()
-        temperature_slider.change(0.7)
-        output_text.clear()
-        explanation_box.update("")
-        dropdown.change("Proust")
-
-    submit_button.click(on_submit, inputs=[user_input, dropdown, temperature_slider], outputs=output_text)
-    clear_button.click(on_clear)
-
-demo.launch()
+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()