Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 import requests
-import language_tool_python
-from bs4 import BeautifulSoup
+from gramformer import Gramformer
 import os
 from datetime import datetime  # Import datetime for date validation
 
+# Initialize Gramformer (set corrector mode to True)
+gf = Gramformer(models=1, use_gpu=False)  # We set 'use_gpu=False' for Hugging Face Spaces
+
 # Load Groq Cloud API key securely from environment variables
 groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")
 
@@ -50,12 +52,11 @@ def generate_email_from_groq(bio, company_name, role):
         print(f"Error: {response.status_code}, {response.text}")
         return "Error generating email. Please check your API key or try again later."
 
-# Grammar and Tone Checker Function
+# Grammar and Tone Checker Function using Gramformer
 def check_grammar(email_text):
-    tool = language_tool_python.LanguageTool('en-US')
-    matches = tool.check(email_text)
-    corrected_text = language_tool_python.utils.correct(email_text, matches)
-    return corrected_text
+    corrected_sentences = list(gf.correct(email_text))
+    # Return the first corrected sentence (gramformer may return multiple suggestions)
+    return corrected_sentences[0] if corrected_sentences else email_text
 
 # Function to validate the DOB format (DD-MM-YYYY)
 def validate_dob(dob):
@@ -82,7 +83,7 @@ def create_email(name, dob, city, company_name, role, email, phone):
     # Step 3: Add the user's email, phone number, and LinkedIn profile to the signature
     signature = f"\n\nBest regards,\n{name}\nEmail: {email}\nPhone: {phone}\nLinkedIn: {linkedin_profile}"
 
-    # Step 4: Run grammar and tone check
+    # Step 4: Run grammar and tone check using Gramformer
     polished_email = check_grammar(generated_email + signature)
 
     # Return the final polished email with the signature