Upload 11 files

app.py

@@ -0,0 +1,87 @@
+import torch
+from flask import Flask, render_template, request
+from difflib import HtmlDiff
+import pandas as pd
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, T5ForConditionalGeneration
+app = Flask(__name__)
+
+# Load Grammarly Coedit-Large model
+tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large")
+model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
+
+# Load custom dataset
+custom_dataset_path = "styleguide_words.csv"
+custom_dataset = pd.read_csv(custom_dataset_path)
+
+# Create a mapping between words to be replaced and their replacements
+replacement_mapping = dict(zip(custom_dataset["Not Allowed"], custom_dataset["Replacement"]))
+
+@app.route('/')
+def index():
+   return render_template('index.html')
+
+@app.route('/correct', methods=['POST'])
+def correct():
+   text = request.form['text']
+   corrected_text = grammar_correction(text)
+   return render_template('result.html', original_text=text, corrected_text=corrected_text)
+
+@app.route('/styleguide', methods=['POST'])
+def styleguide():
+   text = request.form['corrected_text']
+   highlighted_text, suggestions = apply_styleguide(text)
+   return render_template('styleguide.html', corrected_text=text, highlighted_text=highlighted_text, suggestions=suggestions)
+
+@app.route('/compare', methods=['POST'])
+def compare():
+   original_text = request.form['original_text']
+   final_text = request.form['final_text']
+   highlighted_changes = highlight_changes(original_text, final_text)
+   return render_template('compare.html', original_text=original_text, final_text=final_text, highlighted_changes=highlighted_changes)
+
+def grammar_correction(text):
+   # Split the text into sentences
+   sentences = text.split(". ")
+   corrected_sentences = []
+   for sentence in sentences:
+       # Check if the sentence is a command (starts with "-" or contains "_")
+       if sentence.startswith("-") or "_" in sentence:
+           corrected_sentences.append(sentence)  # Skip the command line
+           continue
+       # Tokenize input text
+       input_ids = tokenizer(sentence, return_tensors="pt").input_ids
+       outputs = model.generate(input_ids, max_length=256)
+       edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+       corrected_sentences.append(edited_text)
+       
+      #  inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True)
+      #  # Generate corrected text using the model
+      #  with torch.no_grad():
+      #      outputs = model.generate(**inputs)
+      #  # Decode and append corrected sentence to list
+      #  corrected_sentence = tokenizer.decode(outputs[0], skip_special_tokens=True)
+      #  corrected_sentences.append(corrected_sentence)
+   
+   # Join the corrected sentences into a single paragraph
+   corrected_text = ". ".join(corrected_sentences)
+   return corrected_text
+
+def apply_styleguide(text):
+   # Highlight words mentioned in the CSV file and suggest replacements
+   highlighted_text = text
+   suggestions = []
+   for not_allowed_word, replacement_word in replacement_mapping.items():
+       if not_allowed_word in highlighted_text:
+           highlighted_text = highlighted_text.replace(not_allowed_word, f'<span style="background-color: yellow">{not_allowed_word}</span> ({replacement_word})')
+           suggestions.append((not_allowed_word, replacement_word))
+   return highlighted_text, suggestions
+   
+def highlight_changes(original_text, final_text):
+   # Function to highlight changes between original and final text
+   # You can modify this function as needed
+   diff = HtmlDiff()
+   highlighted_changes = diff.make_table(original_text.splitlines(), final_text.splitlines(), context=True, numlines=2)
+   return highlighted_changes
+   
+if __name__ == '__main__':
+   app.run(debug=True)

requirements.txt

@@ -0,0 +1,8 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch
+torchvision
+torchaudio
+
+flask
+pandas
+transformers

static/styles.css

@@ -0,0 +1,73 @@
+/* styles.css */
+/* Common styles */
+body {
+    font-family: Arial, sans-serif;
+    margin: 0;
+    padding: 0;
+ }
+ header {
+    background-color: #f8f8f8;
+    padding: 20px;
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+ }
+ .logo img {
+    height: 40px;
+    margin-right: 10px;
+ }
+ nav ul {
+    list-style-type: none;
+    padding: 0;
+ }
+ nav ul li {
+    display: inline-block;
+    margin-right: 20px;
+ }
+ nav ul li a {
+    text-decoration: none;
+    color: #333;
+ }
+ main {
+    padding: 20px;
+ }
+ footer {
+    background-color: #333;
+    color: #fff;
+    text-align: center;
+    padding: 10px;
+ }
+ /* Page-specific styles */
+ .hero {
+    text-align: center;
+    margin-top: 50px;
+ }
+ .hero textarea {
+    width: 100%;
+    height: 200px;
+    padding: 10px;
+    margin-top: 20px;
+ }
+ .hero button {
+    background-color: #333;
+    color: #fff;
+    border: none;
+    padding: 10px 20px;
+    cursor: pointer;
+ }
+ .compare {
+    margin-top: 50px;
+ }
+ .compare h2 {
+    text-align: center;
+ }
+ .comparison {
+    display: flex;
+    justify-content: space-between;
+ }
+ .original, .final {
+    flex-basis: 45%;
+ }
+ .original h3, .final h3 {
+    margin-bottom: 10px;
+ }

styleguide_words.csv

@@ -0,0 +1,4 @@
+Not Allowed,Replacement
+guys,R&D team
+slave,child
+killing,canceling

templates/base.html

@@ -0,0 +1,30 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+   <meta charset="UTF-8">
+   <meta name="viewport" content="width=device-width, initial-scale=1.0">
+   <title>{% block title %}Your App Name{% endblock %}</title>
+   <link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}">
+</head>
+<body>
+   <header>
+       <div class="logo">
+           <img src="{{ url_for('static', filename='logo.png') }}" alt="Company Logo">
+           <h1>Editor Assistant</h1>
+       </div>
+       <nav>
+           <ul>
+               <li><a href="/">Home</a></li>
+               <li><a href="/compare">Compare</a></li>
+               <li><a href="/styleguide">Style Guide</a></li>
+           </ul>
+       </nav>
+   </header>
+   <main>
+       {% block content %}{% endblock %}
+   </main>
+   <footer>
+       <p>CIEG, India</p>
+   </footer>
+</body>
+</html>

templates/compare.html

@@ -0,0 +1,21 @@
+{% extends 'base.html' %}
+{% block title %}Compare - Editor Assistant{% endblock %}
+{% block content %}
+<section class="compare">
+   <h2>Comparison between Original and Final Text</h2>
+   <div class="comparison">
+       <div class="original">
+           <h3>Original Text</h3>
+           <p>{{ original_text }}</p>
+       </div>
+       <div class="final">
+           <h3>Final Text</h3>
+           <p>{{ final_text }}</p>
+       </div>
+   </div>
+   <div class="highlighted-changes">
+       {{ highlighted_changes|safe }}
+   </div>
+   <a href="/" class="button">Go Back</a>
+</section>
+{% endblock %}

templates/index.html

@@ -0,0 +1,35 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+   <meta charset="UTF-8">
+   <meta name="viewport" content="width=device-width, initial-scale=1.0">
+   <title>Home - Your App Name</title>
+   <link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}">
+</head>
+<body>
+   <header>
+       <div class="logo">
+           <img src="{{ url_for('static', filename='logo.png') }}" alt="Your App Logo">
+       </div>
+       <nav>
+           <ul>
+               <li><a href="/">Home</a></li>
+           </ul>
+       </nav>
+   </header>
+   <main>
+       <section class="hero">
+           <h2>Editor Assistant</h2>
+           <p>Start by entering your text for grammar correction.</p>
+           <form action="/correct" method="post">
+               <textarea name="text" placeholder="Enter your text here"></textarea>
+               <button type="submit">Submit</button>
+           </form>
+           <a href="https://wwwin.synopsys.com/~docnet/deliverables/help/synopsys_style_guide/#page/style_guide_0/writing_guidelines.htm#ww1000358364" class="button">Style Guide</a>
+       </section>
+   </main>
+   <footer>
+       <p>&copy; CIEG, India</p>
+   </footer>
+</body>
+</html>

templates/result.html

@@ -0,0 +1,41 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+   <meta charset="UTF-8">
+   <meta name="viewport" content="width=device-width, initial-scale=1.0">
+   <title>Result - Your App Name</title>
+   <link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}">
+</head>
+<body>
+   <header>
+       <div class="logo">
+           <img src="{{ url_for('static', filename='logo.png') }}" alt="Your App Logo">
+       </div>
+       <nav>
+           <ul>
+               <li><a href="/">Home</a></li>
+           </ul>
+       </nav>
+   </header>
+   <main>
+       <section class="result">
+           <h2>Grammar Correction Result</h2>
+           <div class="original">
+               <h3>Original Text</h3>
+               <p>{{ original_text }}</p>
+           </div>
+           <div class="corrected">
+               <h3>Corrected Text</h3>
+               <p>{{ corrected_text }}</p>
+           </div>
+           <form action="/styleguide" method="post">
+               <input type="hidden" name="corrected_text" value="{{ corrected_text }}">
+               <button type="submit">Style Guide</button>
+           </form>
+       </section>
+   </main>
+   <footer>
+       <p>&copy; CIEG, India</p>
+   </footer>
+</body>
+</html>

templates/styleguide.html

@@ -0,0 +1,46 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+   <meta charset="UTF-8">
+   <meta name="viewport" content="width=device-width, initial-scale=1.0">
+   <title>Style Guide - Your App Name</title>
+   <link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}">
+</head>
+<body>
+   <header>
+       <div class="logo">
+           <img src="{{ url_for('static', filename='logo.png') }}" alt="Your App Logo">
+       </div>
+       <nav>
+           <ul>
+               <li><a href="/">Home</a></li>
+           </ul>
+       </nav>
+   </header>
+   <main>
+       <section class="style-guide">
+           <h2>Style Guide</h2>
+           <div class="corrected">
+               <h3>Corrected Text</h3>
+               <p>{{ corrected_text }}</p>
+           </div>
+           <div class="highlighted">
+               <h3>Highlighted Text</h3>
+               <p>{{ highlighted_text|safe }}</p>
+           </div>
+           <div class="suggestions">
+               <h3>Suggestions</h3>
+               <ul>
+                   {% for suggestion in suggestions %}
+                   <li>{{ suggestion[0] }}: {{ suggestion[1] }}</li>
+                   {% endfor %}
+               </ul>
+           </div>
+           <a href="/" class="button">Go Back</a>
+       </section>
+   </main>
+   <footer>
+       <p>&copy; CIEG, India</p>
+   </footer>
+</body>
+</html>