UPDATE

app.py

@@ -30,31 +30,49 @@ def main():
     
     # Check button
     if st.button("Check"):
-        # Spelling correction
-        corrected_spelling = fix_spelling(text_input)[0]['generated_text']
+        # Split the text into manageable chunks
+        text_chunks = split_text(text_input)
         
-        # Grammar correction
-        result = happy_tt.generate_text(f"grammar: {text_input}", args=args)
-        corrected_grammar = result.text
+        corrected_spelling_chunks = []
+        corrected_grammar_chunks = []
         
-        # Increment counters if corrections were made
-        if corrected_spelling != text_input:
-            spelling_counter += 1
-        if corrected_grammar != text_input:
-            grammar_counter += 1
+        # Process each chunk
+        for chunk in text_chunks:
+            try:
+                # Spelling correction
+                corrected_spelling = fix_spelling(chunk)[0]['generated_text']
+                corrected_spelling_chunks.append(corrected_spelling)
+                
+                # Grammar correction
+                result = happy_tt.generate_text(f"grammar: {chunk}", args=args)
+                corrected_grammar = result.text
+                corrected_grammar_chunks.append(corrected_grammar)
+                
+                # Increment counters if corrections were made
+                if corrected_spelling != chunk:
+                    spelling_counter += 1
+                if corrected_grammar != chunk:
+                    grammar_counter += 1
+            except Exception as e:
+                st.error(f"Error processing chunk: {chunk}\n{e}")
+        
+        # Combine chunks back into full text
+        corrected_spelling_text = ' '.join(corrected_spelling_chunks)
+        corrected_grammar_text = ' '.join(corrected_grammar_chunks)
         
         # Display corrected text
         st.subheader("Corrected Text:")
-        st.write(corrected_grammar)  # Display grammar-corrected text
+        st.write(corrected_grammar_text)  # Display grammar-corrected text
         
         # Display spelling correction if there's a difference
-        if corrected_spelling != corrected_grammar:
-            st.write(f"Spelling was also corrected to: {corrected_spelling}")
+        if corrected_spelling_text != corrected_grammar_text:
+            st.write("Spelling was also corrected to:")
+            st.write(corrected_spelling_text)
     
     # Display counters
-    st.sidebar.subheader("Corrections Summary")
-    st.sidebar.write(f"Spelling Corrections: {spelling_counter}")
-    st.sidebar.write(f"Grammar Corrections: {grammar_counter}")
+    st.subheader("Corrections Summary")
+    st.write(f"Spelling Corrections: {spelling_counter}")
+    st.write(f"Grammar Corrections: {grammar_counter}")
 
 if __name__ == "__main__":
     main()