Update app.py

app.py

@@ -1,42 +1,31 @@
 # app.py
+import gradio as gr
+import openai
 import os
-openai.api_key = os.getenv("OPENAI_API_KEY")
-from spellchecker import SpellChecker
-from transformers import pipeline
-
-# download spaCy model once
-subprocess.check_call([sys.executable, "-m", "spacy", "download", "en_core_web_sm"])
-nlp = spacy.load("en_core_web_sm")
-spell = SpellChecker(language='en')
-para = pipeline("text2text-generation", model="Vamsi/T5_Paraphrase_Paws", max_new_tokens=2000)
 
-def correct(t): return " ".join(spell.correction(w) or w for w in t.split())
+# load key from Hugging Face secret
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def humanize(text, tone, strength, lock):
-    locked = [(ent.text, ent.label_) for ent in nlp(text).ents] if lock else []
-    out = []
-    for chunk in (lambda x: [" ".join(x[i:i+380]) for i in range(0, len(x), 380)])(text.split()):
-        p = para(chunk, max_new_tokens=2000, temperature=0.7*strength/10)[0]['generated_text']
-        p = correct(p)
-        for e, l in locked: p = re.sub(re.escape(e), e, p, flags=re.IGNORECASE)
-        out.append(p)
-    full = " ".join(out)
-    with open("/tmp/full.txt", "w", encoding="utf-8") as f: f.write(full)
+def humanize(text, tone, strength):
+    prompt = f"Rewrite the text in a {tone} tone without shortening:\n\n{text}"
+    res = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": prompt}],
+        max_tokens=len(text.split()) + 200,
+        temperature=0.7 * strength / 10
+    )
+    full = res.choices[0].message.content.strip()
+    with open("/tmp/full.txt", "w", encoding="utf-8") as f:
+        f.write(full)
     return full, "/tmp/full.txt"
 
-with gr.Blocks(title="AI Humanizer – Unlimited") as demo:
-    with gr.Row():
-        with gr.Column():
-            txt = gr.Textbox(label="Paste text (any length)", lines=15)
-            tone = gr.Dropdown(["Casual","Academic","Marketing","Legal","Creative"], value="Casual")
-            strg = gr.Slider(1,10,value=5,label="Strength")
-            lock = gr.Checkbox(label="Lock facts")
-            btn = gr.Button("Humanize", variant="primary")
-        with gr.Column():
-            out = gr.Textbox(label="Full humanized text", lines=25, interactive=True)
-            with gr.Row():
-                copy = gr.Button("📋 Copy")
-                file = gr.File(label="Download .txt")
-    btn.click(humanize,[txt,tone,strg,lock],[out,file])
-    copy.click(None,out,None,js="(t)=>navigator.clipboard.writeText(t)")
+with gr.Blocks() as demo:
+    txt = gr.Textbox(label="Paste text", lines=15)
+    tone = gr.Dropdown(["Casual","Academic","Marketing","Legal","Creative"], value="Casual")
+    strg = gr.Slider(1,10,value=5,label="Strength")
+    btn = gr.Button("Humanize")
+    out = gr.Textbox(label="Humanized", lines=25)
+    file = gr.File(label="Download")
+    btn.click(humanize,[txt,tone,strg],[out,file])
+
 demo.launch()