Create app.py

app.py

@@ -0,0 +1,85 @@
+import gradio as gr
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+import difflib
+
+# -----------------------------
+# Load Grammar Correction Model
+# -----------------------------
+MODEL_NAME = "vennify/t5-base-grammar-correction"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
+
+# -----------------------------
+# Grammar correction function
+# -----------------------------
+def correct_grammar(text, language):
+    if not text.strip():
+        return "", ""
+    
+    # Prepend instruction
+    input_text = "grammar: " + text
+    input_ids = tokenizer.encode(input_text, return_tensors="pt")
+    
+    outputs = model.generate(input_ids, max_length=512)
+    corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    # Highlight differences
+    diff = difflib.ndiff(text.split(), corrected_text.split())
+    highlighted = []
+    for token in diff:
+        if token.startswith("+"):
+            highlighted.append(f"**{token[2:]}**")  # Added / corrected
+        elif token.startswith("-"):
+            highlighted.append(f"~~{token[2:]}~~")  # Removed
+        else:
+            highlighted.append(token[2:])
+    highlighted_text = " ".join(highlighted)
+    
+    return corrected_text, highlighted_text
+
+# -----------------------------
+# HF-ready Gradio UI
+# -----------------------------
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("<h1 style='color:#8a2be2'>✨ Grammar Corrector</h1>")
+    gr.Markdown("<p style='color:#4b0082'>Correct your English sentences instantly!</p>")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            language = gr.Dropdown(
+                choices=["English"],
+                value="English",
+                label="Language"
+            )
+            user_input = gr.Textbox(
+                label="Enter text (multi-line supported)",
+                placeholder="Type your text here...",
+                lines=5
+            )
+            correct_btn = gr.Button("Correct Grammar ✅")
+
+        with gr.Column(scale=1):
+            corrected_output = gr.Textbox(
+                label="Corrected Text",
+                lines=5
+            )
+            highlighted_output = gr.Markdown(
+                label="Highlighted Changes"
+            )
+            download_btn = gr.File(label="Download Corrected Text")
+
+    # Function to generate and save corrected text
+    def generate_and_save(text, language):
+        corrected, highlighted = correct_grammar(text, language)
+        with open("corrected_text.txt", "w", encoding="utf-8") as f:
+            f.write(corrected)
+        return corrected, highlighted, "corrected_text.txt"
+
+    correct_btn.click(
+        fn=generate_and_save,
+        inputs=[user_input, language],
+        outputs=[corrected_output, highlighted_output, download_btn]
+    )
+
+# Launch app (HF Spaces automatically runs this)
+demo.launch()