Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import requests
-from gramformer import Gramformer
+from gingerit.gingerit import GingerIt  # Import GingerIt for grammar correction
 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
+# Initialize GingerIt parser
+ginger_parser = GingerIt()
 
 # Load Groq Cloud API key securely from environment variables
 groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")
@@ -52,11 +52,10 @@ 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 using Gramformer
+# Grammar and Tone Checker Function using GingerIt
 def check_grammar(email_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
+    corrected_text = ginger_parser.parse(email_text)["result"]
+    return corrected_text
 
 # Function to validate the DOB format (DD-MM-YYYY)
 def validate_dob(dob):
@@ -83,7 +82,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 using Gramformer
+    # Step 4: Run grammar and tone check using GingerIt
     polished_email = check_grammar(generated_email + signature)
 
     # Return the final polished email with the signature