Update app.py

app.py

@@ -55,7 +55,7 @@ if st.button("Generate Code"):
         with st.spinner("Generating code..."):
             completed_text = generate_response(prompt)
             st.success("Code generated successfully!")
-            st.text_area("Ath's Code:", completed_text, height=400)
+            st.code(completed_text, language="python")
 
 st.markdown("""
 <style>
@@ -79,15 +79,11 @@ st.markdown("""
     .stButton button:hover {
         background-color: #5a52d1;
     }
-    .sidebar .sidebar-content {
-        background-color: #4b4c5c;
-    }
-    .sidebar .sidebar-content select {
-        color: white;
-        background-color: #4b4c5c;
-    }
-    .sidebar .sidebar-content label {
-        color: white;
+    .stCodeBlock {
+        background-color: #272822;
+        color: #f8f8f2;
+        font-size: 1rem;
+        border-radius: 10px;
     }
 </style>
 """, unsafe_allow_html=True)