Update app.py

app.py

@@ -1,16 +1,18 @@
-# app.py
-
 import gradio as gr
 import speech_recognition as sr
 from Levenshtein import ratio
 import tempfile
 import numpy as np
 import soundfile as sf
+import language_tool_python
+
+# Initialize LanguageTool for English grammar checking
+tool = language_tool_python.LanguageTool('en-US')
 
 def transcribe_audio(file_info):
     r = sr.Recognizer()
     with tempfile.NamedTemporaryFile(delete=True, suffix=".wav") as tmpfile:
-        sf.write(tmpfile.name, data=file_info, samplerate=44100, format='WAV')
+        sf.write(tmpfile.name, data=file_info[1], samplerate=44100, format='WAV')
         tmpfile.seek(0)
         with sr.AudioFile(tmpfile.name) as source:
             audio_data = r.record(source)
@@ -38,19 +40,33 @@ def pronunciation_correction(expected_text, file_info):
 
     return feedback, description
 
+def grammar_check(text):
+    matches = tool.check(text)
+    corrected_text = tool.correct(text)
+    return corrected_text, matches
+
 with gr.Blocks() as app:
-    with gr.Column():
-        text_input = gr.Textbox(label="Type your sentence here", lines=5, placeholder="Type here...")
-        audio_input = gr.Audio(label="Upload Audio File", type="numpy")
-        check_pronunciation_button = gr.Button("Check Pronunciation")
-        
-        pronunciation_feedback = gr.Textbox(label="Pronunciation Feedback", interactive=False)
-        pronunciation_score = gr.Number(label="Pronunciation Accuracy Score: 0 (No Match) ~ 1 (Perfect)", interactive=False)
-
-        check_pronunciation_button.click(
-            pronunciation_correction,
-            inputs=[text_input, audio_input],
-            outputs=[pronunciation_feedback, pronunciation_score]
-        )
+    with gr.Row():
+        sentence_input = gr.Textbox(label="Enter a Sentence", lines=2)
+        check_grammar_button = gr.Button("Check Grammar")
+        corrected_sentence_output = gr.Textbox(label="Corrected Sentence", interactive=False)
+        grammar_feedback = gr.Textbox(label="Grammar Feedback", interactive=False)
+
+    audio_input = gr.Audio(label="Upload Audio File", type="numpy")
+    check_pronunciation_button = gr.Button("Check Pronunciation")
+    pronunciation_feedback = gr.Textbox(label="Pronunciation Feedback")
+    pronunciation_score = gr.Number(label="Pronunciation Accuracy Score: 0 (No Match) ~ 1 (Perfect)")
+
+    check_grammar_button.click(
+        grammar_check,
+        inputs=sentence_input,
+        outputs=[corrected_sentence_output, grammar_feedback]
+    )
+    
+    check_pronunciation_button.click(
+        pronunciation_correction,
+        inputs=[corrected_sentence_output, audio_input],
+        outputs=[pronunciation_feedback, pronunciation_score]
+    )
 
 app.launch(debug=True)