Update app.py

app.py

@@ -41,8 +41,11 @@ def main(input_text, max_length, temperature, model_name, history):
     global model, tokenizer
     if model is None or tokenizer is None:
         raise Exception("Modèle non chargé. Veuillez charger un modèle avant de générer du texte.")
-    generated_text = generate_text(model, tokenizer, input_text, max_length, temperature)
-    return generated_text
+    generated_text = generate_text(model, tokenizer, input_text, max_length, temperature)    
+    # Mettre à jour l'historique
+    updated_history = append_history(history, f"Entrée: {input_text}\nSortie: {generated_text}")   
+    # Renvoie le texte généré et l'historique mis à jour
+    return generated_text, updated_history  
 
 # Variables globales pour stocker le modèle et le tokenizer
 model = None
@@ -84,9 +87,11 @@ with demo:
         gr.Markdown("# TrySmallModels Quickly!\n"
                     "NB : Work Better with Models Under 5B Parameters")
 
+    with gr.Row():
+        gr.Markdown("## Sélectionner Un Modèle")
+        
     with gr.Row():
         with gr.Column():
-            gr.Markdown("## Sélectionner Un Modèle")
             # Textbox pour entrer le nom du modèle
             model_name_textbox = gr.Textbox(label="Nom du modèle", placeholder="Depot/NomModele Exemple kurakurai/Luth-LFM2-1.2B", interactive=True)
         with gr.Column():
@@ -95,11 +100,14 @@ with demo:
         with gr.Column():
             # Textbox pour Logs et Historique
             history = gr.Textbox(label="Historique", lines=5)
-    
+            
     with gr.Row():
         gr.Markdown("## Entrer Un Prompt")
+        
+    with gr.Row():
         # Textbox pour entrer le texte d'entrée
         input_text = gr.Textbox(label="Texte d'entrée", lines=5)
+        
     with gr.Row():
         # Sliders pour ajuster la longueur maximale et la température
         max_length_slider = gr.Slider(50, 4500, label="Longueur maximale", value=200)
@@ -111,6 +119,8 @@ with demo:
 
     with gr.Row():
         gr.Markdown("## Sortie LLM")
+    
+    with gr.Row():
         # Textbox pour afficher le texte généré
         output_text = gr.Textbox(label="Texte généré", lines=5)