Make it works

app.py

@@ -4,36 +4,42 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokeniser = AutoTokenizer.from_pretrained("rbawden/CCASS-semi-auto-titrages-base")
 model = AutoModelForSeq2SeqLM.from_pretrained("rbawden/CCASS-semi-auto-titrages-base")
 
+def yes_button_func():
+    return True
 
-def generate_titre(matiere_and_titrage_prefix):
-    inputs = tokeniser(matiere_and_titrage_prefix, return_tensors="pt")
-    outputs = model.generate(inputs["input_ids"])
-    res = tokeniser.batch_decode(
-        outputs, skip_special_tokens=True, clean_up_tokenisation_spaces=True
-    )
-    if not (
-        confirm := gr.inputs.Confirm(
-            label=f"Le modèle prédit que le titre est {res}. Est-ce le cas ?",
-            default=False,
-        ).value
-    ):
-        return {"FAIL"}
-    pred_titre += f"<t> {res} <t>"  
-    inputs = tokeniser(matiere_and_titrage_prefix, return_tensors="pt")
+memory_titres = ""
+
+def generate_titre(matiere, sommaire):
+    global memory_titres
+    inp = [' '.join([matiere, memory_titres," <t> ", sommaire])]
+    inputs = tokeniser(inp, return_tensors="pt")
     outputs = model.generate(inputs["input_ids"])
     res = tokeniser.batch_decode(
         outputs, skip_special_tokens=True, clean_up_tokenisation_spaces=True
     )
-    return {"next_prediction": res}
+    memory_titres += f" <t> {res[0]}"
+    return res[0]
 
 
-input_matter = gr.inputs.Textbox(label="Matière")
-input_sommaire = gr.inputs.Textbox(label="Sommaire")
-output_text = gr.outputs.Textbox(label="Next Prediction")
 pred_titre = ""
-matiere_and_titrage_prefix = f"{input_matter} {pred_titre} {input_sommaire}"
+with gr.Blocks() as demo:
+    with gr.Tab("Titrage"):
+        input_matter = gr.Textbox(label="Matière")
+        input_sommaire = gr.Textbox(label="Sommaire")
+        output_text = gr.Textbox(label="Titre prédit")
+        pred_button = gr.Button("Prédiction du prochain titre")
+        #yes_button = gr.Button("Bon titre")
+        gr.Examples([["SECURITE SOCIALE", "L'article R. 243-59 du code de la sécurité sociale n'exige pas que l'avis préalable qu'il prévoit mentionne le caractère concerté du contrôle. Dès lors, c'est à bon droit que la cour d'appel, sans encourir les griefs du moyen, a dit que la circonstance que l'avis préalable ne fasse pas mention que le contrôle s'inscrivait dans le cadre d'un contrôle concerté, décidé par l'Agence centrale des organismes de sécurité sociale, était sans incidence sur sa régularité (1er moyen). Aux termes de l'article L. 243-7-2 du code de la sécurité sociale, afin d'en restituer le véritable caractère, les organismes mentionnés aux articles L. 213-1 et L. 752-1 sont en droit d'écarter, comme ne leur étant pas opposables, les actes constitutifs d'un abus de droit, soit que ces actes aient un caractère fictif, soit que, recherchant le bénéfice d'une application littérale des textes à l'encontre des objectifs poursuivis par leurs auteurs, ils n'aient pu être inspirés par aucun autre motif que celui d'éluder ou d'atténuer les contributions et cotisations sociales d'origine légale ou conventionnelle auxquelles le cotisant est tenu au titre de la législation sociale ou que le cotisant, s'il n'avait pas passé ces actes, aurait normalement supportées, eu égard à sa situation ou à ses activités réelles. Lorsque l'organisme de sécurité sociale écarte un acte juridique dans les conditions ci-dessus, il se place nécessairement sur le terrain de l'abus de droit. Il en résulte qu'il doit se conformer à la procédure prévue par le texte précité et les articles R. 243-60-1 et R. 243-60-3 du code de la sécurité sociale et qu'à défaut de ce faire, les opérations de contrôle et celles, subséquentes, de recouvrement sont entachées de nullité. Viole ces textes, dans leur rédaction applicable au litige, la cour d'appel qui, alors qu'elle constatait que l'organisme de recouvrement avait écarté la convention litigieuse au motif qu'elle avait pour seul objet d'éluder le paiement des cotisations sociales, ce dont il résultait qu'il s'était implicitement placé sur le terrain de l'abus de droit pour opérer le redressement, n'a pas tiré les conséquences légales de ses constatations (2e moyen)"]], [input_matter, input_sommaire])
+    matiere_and_titrage_prefix = [input_matter, input_sommaire]
+    print(matiere_and_titrage_prefix)
+
+    pred_button.click(fn=generate_titre, inputs=matiere_and_titrage_prefix, outputs=output_text)
+    #yes_button.click(fn=generate_titre, inputs=matiere_and_titrage_prefix, outputs=output_text)
 
 
+"""
 gr.Interface(
     fn=generate_titre, inputs=matiere_and_titrage_prefix, outputs=output_text
 ).launch()
+"""
+demo.launch()