update

app.py

@@ -22,11 +22,16 @@ system_prompts = {
 # Fonction pour générer une réponse
 @spaces.GPU(duration=120)
 def generate_response(user_input, system_prompt, max_new_tokens=150, temperature=0.7):
-    # Créer la liste des messages avec le prompt système et l'entrée utilisateur
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": user_input}
-    ]
+    # Gérer le cas où le prompt système est vide
+    if not system_prompt.strip():
+        messages = [
+            {"role": "user", "content": "Translate to Wolof: Bassirou Diomaye Faye is the new Senegalese president. He is 44 years old"}
+        ]
+    else:
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_input}
+        ]
     
     # Tokeniser l'entrée
     inputs = tokenizer(user_input, return_tensors="pt").to(device)
@@ -43,40 +48,42 @@ def update_system_prompt(selected_prompt):
 
 # Interface Gradio avec Blocks
 with gr.Blocks() as iface:
-    # Titre et description de l'application
-    gr.Markdown("# Oolel Chatbot")
-    gr.Markdown("Génération de réponses basées sur des prompts système personnalisés.")
-
-    # Dropdown pour choisir un prompt système
-    dropdown = gr.Dropdown(
-        label="Choisir un prompt système",
-        choices=list(system_prompts.keys()),  # Liste des options de prompts
-        value=None,  # Pas de sélection par défaut
-        type="value",
-        interactive=True
-    )
-    
-    # Textbox pour afficher et modifier le message du prompt système
-    system_prompt_textbox = gr.Textbox(
-        label="Message du prompt système",
-        value="",  # Valeur par défaut vide
-        placeholder="Sélectionnez un prompt système pour afficher son contenu ici..."
-    )
-    
-    # Événement pour mettre à jour le prompt dans le Textbox
-    dropdown.change(update_system_prompt, inputs=[dropdown], outputs=[system_prompt_textbox])
-    
-    # Textbox pour le message utilisateur
-    user_input = gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici...")
-    
-    # Slider pour choisir le nombre maximal de tokens
-    max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
-    
-    # Bouton pour générer la réponse
-    generate_button = gr.Button("Générer une réponse")
-    
-    # Zone pour afficher la réponse générée
-    output = gr.Textbox(label="Réponse générée")
+    with gr.Row():  # Structure divisée en deux colonnes
+        with gr.Column(scale=1):  # Partie gauche : Entrées
+            gr.Markdown("# Oolel Chatbot")
+            gr.Markdown("Génération de réponses basées sur des prompts système personnalisés.")
+            
+            # Textbox pour le message utilisateur
+            user_input = gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici...")
+            
+            # Dropdown pour choisir un prompt système
+            dropdown = gr.Dropdown(
+                label="Choisir un prompt système",
+                choices=list(system_prompts.keys()),  # Liste des options de prompts
+                value=None,  # Pas de sélection par défaut
+                type="value",
+                interactive=True
+            )
+            
+            # Textbox pour afficher et modifier le message du prompt système
+            system_prompt_textbox = gr.Textbox(
+                label="Message du prompt système",
+                value="",  # Valeur par défaut vide
+                placeholder="Sélectionnez un prompt système pour afficher son contenu ici..."
+            )
+            
+            # Événement pour mettre à jour le prompt dans le Textbox
+            dropdown.change(update_system_prompt, inputs=[dropdown], outputs=[system_prompt_textbox])
+            
+            # Slider pour choisir le nombre maximal de tokens
+            max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
+            
+            # Bouton pour générer la réponse
+            generate_button = gr.Button("Générer une réponse")
+        
+        with gr.Column(scale=1):  # Partie droite : Réponse générée
+            gr.Markdown("## Réponse générée")
+            output = gr.Textbox(label="", interactive=False)  # Affichage de la réponse générée
     
     # Connecter le bouton à la fonction de génération
     generate_button.click(