Update app.py

app.py

@@ -128,11 +128,9 @@ if st.button("Generate Code"):
             st.success("Code generated successfully!")
             
             st.markdown('<div class="output-container">', unsafe_allow_html=True)
-            code_blocks = completed_text.split("\n\n")
-            for block in code_blocks:
-                st.markdown('<div class="code-block">', unsafe_allow_html=True)
-                st.code(block)
-                st.markdown('</div>', unsafe_allow_html=True)
+            st.markdown('<div class="code-block">', unsafe_allow_html=True)
+            st.code(completed_text)
+            st.markdown('</div>', unsafe_allow_html=True)
             st.markdown('</div>', unsafe_allow_html=True)
 
 st.markdown("""