Update app.py

app.py

@@ -1,13 +1,31 @@
 import gradio as gr
 from transformers import pipeline
+import difflib
 
-# Swap in a grammar-correction model
-pipe = pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
+# Use a solid grammar correction + improvement model
+pipe = pipeline("text2text-generation", model="pszemraj/flan-t5-large-grammar-synthesis")
 
 def respond(prompt):
-    result = pipe(prompt, max_new_tokens=100)[0]['generated_text']
-    return f"Revised Essay:\n{result}\n\nExplanation:\nThe model corrected grammar, punctuation, and style based on typical English writing conventions."
+# Generate revised essay
+revised = pipe(prompt, max_new_tokens=512)[0]['generated_text']
 
-gr.Interface(fn=respond, inputs="text", outputs="text", title="Free AI Essay Bot").launch()
+# Generate a word-level diff
+diff = difflib.ndiff(prompt.split(), revised.split())
+changes = [line for line in diff if line.startswith("- ") or line.startswith("+ ")]
 
+# Format explanation
+if changes:
+explanation = "\n".join(changes)
+else:
+explanation = "No significant changes were made. Minor improvements only."
 
+return f"📝 Revised Essay:\n\n{revised}\n\n🛠️ Explanation of Changes:\n\n{explanation}"
+
+# Set up Gradio interface
+gr.Interface(
+fn=respond,
+inputs="text",
+outputs="text",
+title="Free AI Essay Bot",
+description="Paste an essay below. The AI will revise it and explain the changes."
+).launch()