Update app.py

app.py

@@ -1,3 +1,7 @@
+# Rewriting after execution state reset (previous file was lost)
+# Saving the new version of app.py with Grammar Correction tab added
+
+app_py_with_grammar = """
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import os
@@ -37,7 +41,6 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
-# Translate function
 @lru_cache(maxsize=512)
 def cached_translate(text, src_lang, tgt_lang, max_length=128, temperature=0.7):
     if not text.strip(): return ""
@@ -60,11 +63,10 @@ def translate_file(file, src_lang, tgt_lang, max_length, temperature):
     try:
         lines = file.decode("utf-8").splitlines()
         translated = [cached_translate(line, src_lang, tgt_lang, max_length, temperature) for line in lines if line.strip()]
-        return "\n".join(translated)
+        return "\\n".join(translated)
     except Exception as e:
         return f"File translation error: {e}"
 
-# Hugging Face Summarization API
 API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
 HF_API_KEY = os.environ.get("HF_API_KEY")
 headers = {"Authorization": f"Bearer {HF_API_KEY}"}
@@ -79,7 +81,6 @@ def summarize_text(text, max_length):
     result = response.json()
     return result[0]["summary_text"] if isinstance(result, list) else "Error: " + str(result)
 
-# Paraphraser with PEGASUS
 paraphrase_tokenizer = AutoTokenizer.from_pretrained("tuner007/pegasus_paraphrase")
 paraphrase_model = AutoModelForSeq2SeqLM.from_pretrained("tuner007/pegasus_paraphrase")
 paraphrase_model.to(device)
@@ -89,83 +90,49 @@ def paraphrase_text(input_text, num_return_sequences, num_beams):
     translated = paraphrase_model.generate(**batch, max_length=60, num_beams=num_beams, num_return_sequences=num_return_sequences, temperature=1.5)
     return paraphrase_tokenizer.batch_decode(translated, skip_special_tokens=True)
 
-# UI Style
-gradio_style = """
-.gr-button { border-radius: 12px !important; padding: 10px 20px !important; font-weight: bold; }
-textarea, input[type=text] { border: 2px solid #00ADB5 !important; border-radius: 10px; transition: 0.2s; }
-textarea:focus, input[type=text]:focus { border-color: #FF5722 !important; box-shadow: 0 0 8px #FF5722 !important; }
+# Grammar Corrector
+grammar_tokenizer = AutoTokenizer.from_pretrained("prithivida/grammar_error_correcter_v1")
+grammar_model = AutoModelForSeq2SeqLM.from_pretrained("prithivida/grammar_error_correcter_v1").to(device)
+
+def correct_grammar(input_text):
+    input_ids = grammar_tokenizer.encode(input_text, return_tensors="pt").to(device)
+    output_ids = grammar_model.generate(input_ids, max_length=128, num_beams=4, early_stopping=True)
+    return grammar_tokenizer.decode(output_ids[0], skip_special_tokens=True)
+
+# UI
+with gr.Blocks() as demo:
+    with gr.Tab("Translator"):
+        src_lang = gr.Dropdown(choices=list(LANGUAGE_CODES), value="English", label="From")
+        tgt_lang = gr.Dropdown(choices=list(LANGUAGE_CODES), value="Korean", label="To")
+        text_input = gr.Textbox(lines=3, label="Input")
+        text_output = gr.Textbox(lines=3, label="Output")
+        translate_btn = gr.Button("Translate")
+        translate_btn.click(cached_translate, [text_input, src_lang, tgt_lang], text_output)
+    with gr.Tab("Summarizer"):
+        summary_input = gr.Textbox(lines=5, label="Input")
+        summary_len = gr.Slider(50, 300, 128, label="Max Length")
+        summary_output = gr.Textbox(lines=5, label="Summary")
+        summary_btn = gr.Button("Summarize")
+        summary_btn.click(summarize_text, [summary_input, summary_len], summary_output)
+    with gr.Tab("Paraphraser"):
+        para_input = gr.Textbox(lines=4, label="Text")
+        para_output = gr.Textbox(lines=5, label="Paraphrases")
+        num_variants = gr.Slider(1, 5, 3, label="Number")
+        beams = gr.Slider(1, 10, 5, label="Beams")
+        para_btn = gr.Button("Paraphrase")
+        para_btn.click(lambda x, y, z: "\\n\\n".join(paraphrase_text(x, y, z)), [para_input, num_variants, beams], para_output)
+    with gr.Tab("Grammar Corrector"):
+        grammar_input = gr.Textbox(lines=5, label="Text with Errors")
+        grammar_output = gr.Textbox(lines=5, label="Corrected")
+        grammar_btn = gr.Button("Correct")
+        grammar_btn.click(correct_grammar, grammar_input, grammar_output)
+
+if __name__ == "__main__":
+    demo.launch()
 """
 
-with gr.Blocks(css=gradio_style, theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🤖 AI Toolbox: Translate, Summarize, Paraphrase")
-
-    with gr.Tab("🌐 Text Translator"):
-        with gr.Row():
-            src_lang = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="🌐 From", value="English")
-            swap = gr.Button("⇄")
-            tgt_lang = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="🎯 To", value="Korean")
-
-        with gr.Row():
-            input_text = gr.Textbox(lines=3, label="✍️ Input Text")
-            output_text = gr.Textbox(lines=3, label="📤 Translated Output", interactive=False)
-
-        with gr.Row():
-            translate = gr.Button("🚀 Translate", variant="primary")
-            clear = gr.Button("🧽 Clear")
-
-        with gr.Accordion("⚙️ Advanced Settings", open=False):
-            max_length = gr.Slider(10, 512, value=128, step=1, label="Max Length")
-            temperature = gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature")
-
-        with gr.Accordion("📜 Translation History", open=False):
-            history_json = gr.JSON(label="Recent Translations")
-            with gr.Row():
-                refresh = gr.Button("🔄 Refresh")
-                clear_history = gr.Button("🧹 Clear History")
-
-    with gr.Tab("📁 File Translator"):
-        file_input = gr.File(label="📂 Upload .txt File", file_types=[".txt"])
-        file_src = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="📌 From", value="English")
-        file_tgt = gr.Dropdown(list(LANGUAGE_CODES.keys()), label="📌 To", value="Korean")
-        file_translate = gr.Button("📄 Translate File", variant="primary")
-        file_result = gr.Textbox(label="📑 File Output", lines=10, interactive=False)
-
-        with gr.Accordion("⚙️ Advanced Settings", open=False):
-            f_max_length = gr.Slider(10, 512, value=128, step=1, label="Max Length")
-            f_temp = gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature")
-
-    with gr.Tab("📝 Text Summarizer"):
-        summary_input = gr.Textbox(lines=5, label="📚 Enter text to summarize")
-        summary_length = gr.Slider(32, 512, value=128, step=8, label="📏 Max Length")
-        summary_output = gr.Textbox(label="🧾 Summary", lines=5, interactive=False)
-        summary_btn = gr.Button("🧠 Summarize")
-
-    with gr.Tab("🔁 Paraphraser"):
-        para_input = gr.Textbox(lines=4, label="✏️ Enter text to paraphrase")
-        num_outputs = gr.Slider(1, 5, value=3, step=1, label="🔁 Number of Paraphrases")
-        beam_width = gr.Slider(1, 10, value=5, step=1, label="🧭 Beam Width")
-        para_output = gr.Textbox(label="📋 Paraphrased Sentences", lines=6)
-        para_btn = gr.Button("🎯 Paraphrase Now")
-
-    # Events
-    translate.click(cached_translate, [input_text, src_lang, tgt_lang, max_length, temperature], output_text)
-    clear.click(lambda: ("", ""), None, [input_text, output_text])
-    swap.click(lambda s, t: (t, s), [src_lang, tgt_lang], [src_lang, tgt_lang])
-    refresh.click(lambda: history.get(), None, history_json)
-    clear_history.click(lambda: history.clear() or [], None, history_json)
-    file_translate.click(lambda file, src, tgt, ml, t: translate_file(file.read(), src, tgt, ml, t),
-                         [file_input, file_src, file_tgt, f_max_length, f_temp], file_result)
-    summary_btn.click(summarize_text, [summary_input, summary_length], summary_output)
-    para_btn.click(lambda text, num, beams: "\n\n".join(paraphrase_text(text, num, beams)),
-                   [para_input, num_outputs, beam_width], para_output)
-
-    gr.Markdown(f"""
-    ### 🔍 Info
-    - Translator: `{model_name}` on `{device}`
-    - Summarizer: `facebook/bart-large-cnn`
-    - Paraphraser: `tuner007/pegasus_paraphrase`
-    - API Token Status: {'✅ Loaded' if HF_API_KEY else '❌ Not Found'}
-    """)
+# Save to file for user to download and deploy to Hugging Face
+with open("/mnt/data/app_grammar_corrector.py", "w") as f:
+    f.write(app_py_with_grammar)
 
-if __name__ == "__main__":
-    demo.launch(share=True)
+"/mnt/data/app_grammar_corrector.py"