Update app.py

app.py

@@ -52,7 +52,7 @@ with col1:
         try:
             output = interpreter.run_code(code_input)
             st.subheader("✨ Code Output ✨")
-            st.write(output)
+            st.text_area("Execution Output", output, height=600, max_chars=None)
         except Exception as e:
             st.error(f"Error: {e}")
             logger.error(f"Code execution error: {e}")
@@ -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.text_area("corrected_code",corrected_code, height=400)
+        gemini_output.code(corrected_code, language="python")
 
 # Feedback form (outside of the columns, after all content)
 st.subheader("Feedback")