update interface app

app.py

@@ -1,8 +1,10 @@
 from transformers import BartForConditionalGeneration, BartTokenizer
-
-
+from transformers.utils import logging
 import gradio as gr
 
+#define the logger instance
+logger = logging.get_logger("transformers")
+
 # Charger le modèle BART et le tokenizer
 model_name = "facebook/bart-large-cnn"
 tokenizer = BartTokenizer.from_pretrained(model_name)
@@ -18,9 +20,38 @@ def generate_text(prompt):
 #model.save_pretrained("model")
 #tokenizer.save_pretrained("model")
 
-# Créer une interface de saisie avec Gradio
-interface = gr.Interface(fn=generate_text, inputs="text", outputs="text",title="TeLLMyStory",description="Enter your story idea and the model will generate the story based on it.")
+#for the app functions
+def clear_save_textbox(message):
+    return " ", message
+
+def show_input_text(message,history:list[tuple[str,str]]):
+    history.append((message,""))
+    story = generate_text(message)
+    history[-1] = (message,story)
+    return history
 
+def delete_previous_text(history:list[tuple[str,str]]):
+    try:
+        message, _ = history.pop()
+    except IndexError:
+        message = " "
+    return history, message
 
-#Lancer l'interface
-interface.launch(share=True)
+# Créer une interface de saisie avec Gradio
+#interface = gr.Interface(fn=generate_text, inputs="text", outputs="text",title="TeLLMyStory",description="Enter your story idea and the model will generate the story based on it.")
+with gr.Blocks() as demo:
+    gr.Markdown("TeLLMyStory chatbot")
+    #input_text = blocks.text(name="input_text", label="Enter your story idea here", default="Once upon a time, there was")
+    with gr.Row():
+        input_text = gr.Textbox(name="input_text", label="Enter your story idea here", default="Once upon a time, there was")
+        clear_button = demo.button(name="clear_button", label="Clear", fn=clear_save_textbox, inputs=[input_text],variants=["secondary"])
+        retry_button = demo.button(name="retry_button", label="Retry", fn=delete_previous_text, inputs=[input_text],variants=["secondary"])
+
+    with gr.Row():
+        gr.Markdown("History of your story ideas")
+        gen_story = gr.Textbox(name="story", label="History", default=" ", type="text",readonly=True)
+    
+    send_button = demo.button(name="send_button", label="Send", fn=show_input_text, inputs=[input_text],outputs=[gen_story],variants=["primary"])
+
+# Lancer l'interface
+demo.launch()