Update app.py

app.py

@@ -2,14 +2,9 @@ from transformers import pipeline
 import gradio as gr
 import re
 
-# ============ MODELS ============
-MODEL_SIMPLIFY = "google/flan-t5-small"
-MODEL_TRANSLATE_EN_FR = "Helsinki-NLP/opus-mt-en-fr"
-MODEL_TRANSLATE_FR_EN = "Helsinki-NLP/opus-mt-fr-en"
-
-simplifier = pipeline("text2text-generation", model=MODEL_SIMPLIFY)
-translate_en_fr = pipeline("translation", model=MODEL_TRANSLATE_EN_FR)
-translate_fr_en = pipeline("translation", model=MODEL_TRANSLATE_FR_EN)
+# ============ MODEL ============
+MODEL_ID = "google/flan-t5-small"
+simplifier = pipeline("text2text-generation", model=MODEL_ID)
 
 
 # ============ HELPERS ============
@@ -17,7 +12,9 @@ translate_fr_en = pipeline("translation", model=MODEL_TRANSLATE_FR_EN)
 def too_similar(src: str, out: str) -> bool:
     src_clean = src.lower().strip()
     out_clean = out.lower().strip()
-    if out_clean.startswith(src_clean) or len(out_clean) < 40:
+    if out_clean.startswith(src_clean):
+        return True
+    if len(out_clean) < 40:
         return True
     src_words = set(src_clean.split())
     out_words = set(out_clean.split())
@@ -28,22 +25,25 @@ def too_similar(src: str, out: str) -> bool:
 
 
 def fallback_explanation(text: str, lang: str = "en") -> str:
+    txt = text.strip()
+
     if lang == "fr":
         return (
             "### 🗣️ Explication en français simple\n"
-            f"- **Texte original :** {text}\n"
+            f"- **Texte original :** {txt}\n"
             "- Ce texte contient une règle ou une obligation administrative/juridique.\n"
-            "- Vérifie qui doit agir, ce qui doit être fait, et quand.\n"
-            "- S’il mentionne une *décision définitive*, cela veut dire qu’on **ne peut plus la contester** après le délai.\n"
-            "\n**En résumé :** ce texte indique ce qu’il faut faire et ce qui se passe si on ne le fait pas."
+            "- Il faut bien vérifier qui doit faire quoi, dans quel délai, et quelles sont les conséquences.\n"
+            "- Si le texte mentionne une *décision définitive*, cela signifie qu’on **ne peut plus la contester** après ce délai.\n"
+            "\n**En résumé :** ce texte décrit ce qu’il faut faire et les conséquences si on ne le fait pas."
         )
+
     return (
         "### 🗣️ Explanation in simple English\n"
-        f"- **Original clause:** {text}\n"
-        "- This text describes a rule or obligation.\n"
+        f"- **Original clause:** {txt}\n"
+        "- This text contains a rule or an obligation.\n"
         "- Check who must act, what must be done, and when.\n"
-        "- If it mentions a *final decision*, that means it **can’t be challenged later**.\n"
-        "\n**In short:** this clause tells you what must be done and what happens if you don’t."
+        "- If it says a decision becomes *final*, that means it **cannot be challenged later**.\n"
+        "\n**In short:** this tells you what must be done and what happens if you don’t."
     )
 
 
@@ -90,23 +90,6 @@ Explanation in simple English:
     return result
 
 
-# ============ TRANSLATION FUNCTION ============
-
-def translate_text(text, lang):
-    """Translate output between FR <-> EN"""
-    if not text.strip():
-        return "Please generate an explanation first." if lang == "en" else "Veuillez d’abord générer une explication."
-    try:
-        if lang == "fr":
-            translated = translate_fr_en(text)[0]["translation_text"]
-            return f"### 🌍 Translated to English\n{translated}"
-        else:
-            translated = translate_en_fr(text)[0]["translation_text"]
-            return f"### 🌍 Traduit en français\n{translated}"
-    except Exception as e:
-        return f"⚠️ Translation error: {e}"
-
-
 # ============ INTERFACE ============
 
 with gr.Blocks(
@@ -117,14 +100,11 @@ with gr.Blocks(
     # --- HEADER ---
     gr.HTML(
         """
-        <div style="text-align:center; margin-bottom:25px;">
-            <img src="https://upload.wikimedia.org/wikipedia/commons/0/0c/Scale_of_justice_2.svg"
-                 alt="Talk2Law Logo" width="85" style="margin-bottom:10px; opacity:0.9;">
-            <h1 style="color:#283593; font-size:2.2em;">⚖️ Talk2Law</h1>
+        <div style="text-align: center; margin-bottom: 25px;">
+            <h1 style="color:#283593; font-size: 2.2em;">⚖️ Talk2Law</h1>
             <h3 style="color:#455A64;">Explain complex legal texts in simple French or English</h3>
-            <p style="color:#607D8B; font-size:0.95em;">
-                Paste a clause below — the app will simplify, reformulate, and explain it clearly.<br>
-                Collez un texte ci-dessous — l’application le reformulera et l’expliquera clairement.
+            <p style="color:#607D8B; font-size: 0.95em;">
+                Paste a clause below — the app will simplify, reformulate, and explain it clearly.
             </p>
         </div>
         """
@@ -133,7 +113,12 @@ with gr.Blocks(
     # --- INPUT AREA ---
     with gr.Row():
         with gr.Column(scale=1):
-            lang = gr.Radio(["English", "Français"], value="English", label="🌐 Language / Langue")
+            lang = gr.Radio(
+                ["English", "Français"],
+                value="English",
+                label="🌐 Language / Langue",
+                interactive=True,
+            )
 
             inp = gr.Textbox(
                 lines=8,
@@ -156,13 +141,10 @@ with gr.Blocks(
 
         with gr.Column(scale=1):
             out = gr.Markdown(label="✅ Explanation / Explication")
-            translate_btn = gr.Button("🌍 Translate / Traduire", variant="secondary")
-            translated_out = gr.Markdown(label="🌎 Translated Output / Texte traduit")
 
     # --- BUTTON ACTIONS ---
     btn.click(fn=simplify_legal, inputs=[inp, lang], outputs=out)
     clear_btn.click(fn=lambda: "", inputs=None, outputs=inp)
-    translate_btn.click(fn=translate_text, inputs=[out, lang], outputs=translated_out)
 
     # --- FOOTER ---
     gr.HTML(