Update app.py

app.py

@@ -163,8 +163,7 @@ def main():
     if st.session_state.generated_code:
         with st.container():
             st.subheader(f"Generated {selected_language} Code")
-            st.markdown(f"""<div class="code-block"><pre><code class="language-{selected_language.lower()}">{st.session_state.generated_code}</code></pre></div>""", 
-                       unsafe_allow_html=True)
+            st.code(st.session_state.generated_code, language=selected_language.lower())
             
             # Explanation section
             with st.expander("📝 Code Explanation", expanded=True):
@@ -186,8 +185,7 @@ def main():
         if st.session_state.alternative_code:
             with st.container():
                 st.subheader(f"Alternative {selected_language} Solution")
-                st.markdown(f"""<div class="code-block"><pre><code class="language-{selected_language.lower()}">{st.session_state.alternative_code}</code></pre></div>""", 
-                           unsafe_allow_html=True)
+                st.code(st.session_state.alternative_code, language=selected_language.lower())
 
 if __name__ == "__main__":
     main()