Update app.py

app.py

@@ -1,37 +1,61 @@
-
 import re
 from langdetect import detect, DetectorFactory
 from transformers import pipeline
 import gradio as gr
 
-# Garante resultados determinísticos em detect()
+# Para resultados determinísticos na detecção de idioma
 DetectorFactory.seed = 0
 
-# Carregando o modelo GPT2 pequeno em português (CPU)
+# Pipeline de geração de texto instruído (text2text) com FLAN-T5
 generator = pipeline(
-    "text-generation",
-    model="pierreguillou/gpt2-small-portuguese",
+    "text2text-generation",
+    model="google/flan-t5-small",
     device=-1
 )
 
-# Comandos e palavras-chave em pt/en/fr
+# Palavras-chave de comando em pt/en/fr
 COMMANDS = {
-    "resumo":   ["resuma", "resumo", "resumir", "summarize", "résumé", "résumer"],
-    "reescrever": ["reescreva", "reformule", "reformular", "rewrite", "réécrire"],
-    "expandir": ["expanda", "expansão", "expandir", "detalhe", "expand", "développez"],
-    "corrigir": ["corrija", "corrigir", "melhore", "revise", "correct", "corriger"]
+    "resumo":    ["resuma", "resumo", "resumir", "summarize", "résumé", "résumer"],
+    "reescrever":["reescreva", "reformule", "reformular", "rewrite", "réécrire"],
+    "expandir":  ["expanda", "expansão", "expandir", "detalhe", "expand", "développez"],
+    "corrigir":  ["corrija", "corrigir", "melhore", "revise", "correct", "corriger"]
+}
+
+# Mensagens de instrução por comando e idioma
+INSTRUCTIONS = {
+  "resumo": {
+    "pt": "Resuma o texto a seguir de forma concisa:",
+    "en": "Please summarize the following text concisely:",
+    "fr": "Veuillez résumer le texte suivant de manière concise :"
+  },
+  "reescrever": {
+    "pt": "Reescreva este texto com mais clareza e estilo:",
+    "en": "Rewrite this text with more clarity and style:",
+    "fr": "Réécrivez ce texte avec plus de clarté et de style :"
+  },
+  "expandir": {
+    "pt": "Expanda este texto, adicionando detalhes e explicações:",
+    "en": "Expand the following text by adding details and explanations:",
+    "fr": "Développez ce texte en ajoutant des détails et des explications :"
+  },
+  "corrigir": {
+    "pt": "Corrija gramática, ortografia e estilo deste texto:",
+    "en": "Correct the grammar, spelling, and style of this text:",
+    "fr": "Corrigez la grammaire, l'orthographe et le style de ce texte :"
+  },
 }
 
-HUMANIZE_PROMPT = {
-    "pt": "Por favor, torne o texto a seguir mais natural e humano:\n\n",
-    "en": "Please make the following text more natural and human-like:\n\n",
-    "fr": "Veuillez rendre le texte suivant plus naturel et humain :\n\n"
+# Prompt de humanização por idioma
+HUMANIZE = {
+  "pt": "Por favor, torne o texto a seguir mais natural e humano:",
+  "en": "Please make the following text more natural and human-like:",
+  "fr": "Veuillez rendre le texte suivant plus naturel et humain :"
 }
 
 def detect_language(text: str) -> str:
     try:
-        lang = detect(text)
-        return lang if lang in HUMANIZE_PROMPT else "pt"
+        code = detect(text)
+        return code if code in HUMANIZE else "pt"
     except:
         return "pt"
 
@@ -41,67 +65,39 @@ def find_command(text: str) -> str:
         for kw in kws:
             if kw in low:
                 return cmd
-    return "gerar"
+    return "expandir"  # padrão se não achar comando
 
 def clean_text(text: str) -> str:
-    txt = re.sub(r"\s+", " ", text)
+    txt = re.sub(r"\s+", " ", text).strip()
     for kws in COMMANDS.values():
         for kw in kws:
             txt = re.sub(rf"\b{kw}\b", "", txt, flags=re.IGNORECASE)
     return txt.strip()
 
-def build_prompt(core: str, cmd: str, lang: str) -> str:
-    if cmd == "resumo":
-        head = {
-            "pt": "Resuma o texto a seguir de forma concisa:\n\n",
-            "en": "Summarize the following text concisely:\n\n",
-            "fr": "Résumez le texte suivant de manière concise :\n\n"
-        }
-    elif cmd == "reescrever":
-        head = {
-            "pt": "Reescreva este texto com mais clareza e estilo:\n\n",
-            "en": "Rewrite this text with more clarity and style:\n\n",
-            "fr": "Réécrivez ce texte avec plus de clarté et de style :\n\n"
-        }
-    elif cmd == "expandir":
-        head = {
-            "pt": "Expanda este texto, adicionando detalhes e explicações:\n\n",
-            "en": "Expand this text, adding details and explanations:\n\n",
-            "fr": "Développez ce texte en ajoutant des détails et des explications :\n\n"
-        }
-    elif cmd == "corrigir":
-        head = {
-            "pt": "Corrija gramática, ortografia e estilo deste texto:\n\n",
-            "en": "Correct the grammar, spelling, and style of this text:\n\n",
-            "fr": "Corrigez la grammaire, l'orthographe et le style de ce texte :\n\n"
-        }
-    else:
-        head = {"pt":"","en":"","fr":""}
-    return head[lang] + core + "\n\n"
-
 def gerar_resposta(texto: str) -> str:
     lang = detect_language(texto)
     cmd  = find_command(texto)
     core = clean_text(texto)
 
-    prompt1 = build_prompt(core, cmd, lang)
-    out1 = generator(prompt1, max_new_tokens=200, temperature=0.7, top_p=0.9)[0]["generated_text"]
-    res1 = out1.replace(prompt1, "").strip()
+    # 1) Geração principal via FLAN-T5
+    instr = INSTRUCTIONS[cmd][lang]
+    prompt1 = f"{instr}\n\n{core}"
+    out1 = generator(prompt1, max_length=512, do_sample=False)[0]["generated_text"].strip()
 
-    prompt2 = HUMANIZE_PROMPT[lang] + res1 + "\n\n"
-    out2 = generator(prompt2, max_new_tokens=100, temperature=0.6, top_p=0.8)[0]["generated_text"]
-    res2 = out2.replace(prompt2, "").strip()
+    # 2) Humanização
+    prompt2 = f"{HUMANIZE[lang]}\n\n{out1}"
+    out2 = generator(prompt2, max_length=256, do_sample=False)[0]["generated_text"].strip()
 
-    return res2
+    return out2
 
-demo = gr.Interface(
+# Interface Gradio
+app = gr.Interface(
     fn=gerar_resposta,
-    inputs=gr.Textbox(lines=6, placeholder="Digite seu texto com 'resuma', 'expanda', etc...", label="Entrada"),
-    outputs=gr.Textbox(label="Resposta AI"),
-    title="🧠 IA Multilingue Sr. Nicolas",
-    description="Detecta comando embutido, idioma e devolve resposta humanizada.",
-    allow_flagging="never"
+    inputs=gr.Textbox(lines=6, placeholder="Escreva algo com 'resuma', 'expanda', etc...", label="Entrada"),
+    outputs=gr.Textbox(label="Resposta"),
+    title="🧠 IA Instruída Multilíngue",
+    description="Detecta idioma e comando embutido, responde no mesmo idioma e humaniza o texto."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()