Update app.py

app.py

@@ -203,9 +203,10 @@ def main():
         st.subheader('Error Correction')
         formatter = PythonCodeFormatter(output_code)
         formatted_code = formatter.format()
-        st.write('Code after correction:')
-        st.write(formatted_code)
-
+        #st.write('Code after correction:')
+        #st.write(formatted_code)
+        st.subheader('Code after correction:')
+        st.code(formatted_code, language='python')