Update app.py

app.py

@@ -1,68 +1,61 @@
 # app.py
 import gradio as gr
-import spacy, re, subprocess, sys, io, os
+import spacy, re, subprocess, sys, os
 from spellchecker import SpellChecker
 from transformers import pipeline
 
-# 1. download spaCy English model
 subprocess.check_call([sys.executable, "-m", "spacy", "download", "en_core_web_sm"])
 nlp = spacy.load("en_core_web_sm")
-
-# 2. lightweight spell checker
 spell = SpellChecker(language='en')
+para = pipeline("text2text-generation", model="Vamsi/T5_Paraphrase_Paws", device=-1)
+
 def simple_correct(text):
     return " ".join(spell.correction(w) or w for w in text.split())
 
-# 3. paraphrase model
-para = pipeline("text2text-generation", model="Vamsi/T5_Paraphrase_Paws", device=-1)  # CPU
-
-def chunk_text(text, max_words=400):
-    """Split text into ~max_words chunks."""
+def chunk_text(text, max_words=380):
     words = text.split()
     for i in range(0, len(words), max_words):
-        yield " ".join(words[i:i+max_words])
+        yield " ".join(words[i:i + max_words])
 
 def humanize(text, tone, strength, freeze):
-    # lock entities if requested
     locked = [(ent.text, ent.label_) for ent in nlp(text).ents] if freeze else []
-
-    # process chunk-by-chunk
-    out_chunks = []
-    for chunk in chunk_text(text, 400):
-        paraphrased = para(
-            chunk,
-            max_length=512,
-            do_sample=True,
-            temperature=0.7 * strength / 10
-        )[0]['generated_text']
+    chunks_out = []
+    for chunk in chunk_text(text, 380):
+        paraphrased = para(chunk, max_length=512, do_sample=True,
+                           temperature=0.7 * strength / 10)[0]['generated_text']
         paraphrased = simple_correct(paraphrased)
-        # restore locked entities in each chunk
         for ent, label in locked:
             paraphrased = re.sub(re.escape(ent), ent, paraphrased, flags=re.IGNORECASE)
-        out_chunks.append(paraphrased)
+        chunks_out.append(paraphrased)
+    full_text = "\n\n".join(chunks_out)
 
-    full = " ".join(out_chunks)
-    # make a downloadable file
+    # save downloadable file
     file_path = "/tmp/humanized.txt"
     with open(file_path, "w", encoding="utf-8") as f:
-        f.write(full)
-    return full, file_path
-
-# --- Gradio interface ---
-iface = gr.Interface(
-    fn=humanize,
-    inputs=[
-        gr.Textbox(label="Paste your long text here", lines=15, placeholder="Paste up to 10 000 words…"),
-        gr.Dropdown(["Casual", "Academic", "Marketing", "Legal", "Creative"], value="Casual", label="Tone"),
-        gr.Slider(1, 10, value=5, label="Humanize Strength"),
-        gr.Checkbox(label="Lock facts / dates / names")
-    ],
-    outputs=[
-        gr.Textbox(label="Humanized text", lines=20),
-        gr.File(label="Download full text as .txt")
-    ],
-    title="AI Humanizer – Unlimited Words & Download",
-    description="Paste any length, get the full text back plus a downloadable file."
-)
-
-iface.launch()
+        f.write(full_text)
+    return full_text, file_path
+
+with gr.Blocks(title="AI Humanizer – Unlimited Words") as demo:
+    gr.Markdown("## AI Humanizer – Unlimited Words")
+    with gr.Row():
+        with gr.Column():
+            text_in = gr.Textbox(label="Paste your LONG text", lines=15, max_lines=None)
+            tone_dd = gr.Dropdown(["Casual", "Academic", "Marketing", "Legal", "Creative"], value="Casual", label="Tone")
+            strength_sl = gr.Slider(1, 10, value=5, label="Strength 1-10")
+            lock_cb = gr.Checkbox(label="Lock facts / dates / names")
+            submit_btn = gr.Button("Humanize", variant="primary")
+
+        with gr.Column():
+            text_out = gr.Textbox(label="FULL humanized text", lines=25, interactive=True)
+            with gr.Row():
+                copy_btn = gr.Button("📋 Copy Text")
+                file_out = gr.File(label="Download .txt")
+
+    submit_btn.click(humanize, inputs=[text_in, tone_dd, strength_sl, lock_cb],
+                     outputs=[text_out, file_out])
+
+    # JavaScript one-line copy
+    copy_btn.click(None, text_out, None,
+                   js="(txt) => navigator.clipboard.writeText(txt)")
+
+demo.launch()