Update app.py

app.py

@@ -73,7 +73,7 @@ with col3:
     if st.button("Corrected Code"):
         logger.info("User requested code correction.")
         corrected_code = get_correction_and_comments(code_input)
-        gemini_output.write(corrected_code, language="python")
+        gemini_output.text_area("corrected_code",corrected_code, height=400)
 
 # Feedback form (outside of the columns, after all content)
 st.subheader("Feedback")