Update app.py

app.py

@@ -55,7 +55,11 @@ if st.button("Generate Code"):
         with st.spinner("Generating code..."):
             completed_text = generate_response(prompt)
             st.success("Code generated successfully!")
-            st.code(completed_text, language="python")
+            
+            # Split the response into separate code blocks based on a custom delimiter
+            code_blocks = completed_text.split("\n\n")
+            for block in code_blocks:
+                st.code(block, language="python")
 
 st.markdown("""
 <style>