Update app.py

app.py

@@ -14,14 +14,15 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Custom CSS for Neon Theme
+# Custom CSS for Enhanced UI
 def inject_custom_css():
     st.markdown("""
     <style>
-        .main { background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%); }
+        .main { background: linear-gradient(135deg, #0f0f0f 0%, #1a1a2e 100%); color: #fff; }
         .stTextInput input { color: #4fffb0 !important; background: rgba(0,0,0,0.3) !important; }
-        .stButton>button { background: rgba(79,255,176,0.2) !important; border: 2px solid #4fffb0 !important; }
-        .code-editor { border: 2px solid #4fffb0; border-radius: 10px; padding: 1rem; background: rgba(0,0,0,0.5); }
+        .stButton>button { background: #4fffb0 !important; border: 2px solid #00ffcc !important; color: black; }
+        .code-box { border: 2px solid #4fffb0; border-radius: 10px; padding: 1rem; background: rgba(0,0,0,0.5); color: #fff; }
+        .stSelectbox div { background: rgba(0,0,0,0.3) !important; color: #4fffb0 !important; }
     </style>
     """, unsafe_allow_html=True)
 
@@ -36,13 +37,13 @@ def get_groq_client():
     return Groq(api_key=api_key)
 
 # Code Generation Functions
-def generate_code(query, model, client):
+def generate_code(query, language, model, client):
     try:
         completion = client.chat.completions.create(
             model=model,
             messages=[{
                 "role": "user",
-                "content": f"Generate Python code for: {query}. Include comments and error handling."
+                "content": f"Generate {language} code for: {query}. Include comments and error handling."
             }],
             temperature=0.6,
             max_tokens=4096,
@@ -59,33 +60,42 @@ def main():
     if not client:
         return
 
-    st.title("AI Code Generator 🚀")
+    st.title("🔥 AI Code Generator 🚀")
+    st.subheader("Generate, Edit & Save Your Code Instantly!")
     
     col1, col2 = st.columns([3, 1])
+    
     with col1:
         query = st.text_input("Describe your coding requirement:")
+        language = st.selectbox("Select Programming Language:", ["Python", "JavaScript", "Java", "C++", "Go"])
     
     if 'generated_code' not in st.session_state:
         st.session_state.generated_code = None
     
     if col2.button("🔄 New Code"):
         st.session_state.generated_code = None
-
+    
     if query and not st.session_state.generated_code:
         with st.spinner("Generating code..."):
-            code = generate_code(query, PRIMARY_MODEL, client)
+            code = generate_code(query, language, PRIMARY_MODEL, client)
             if not code:
                 st.warning("Trying backup model...")
-                code = generate_code(query, BACKUP_MODEL, client)
+                code = generate_code(query, language, BACKUP_MODEL, client)
             
             if code:
                 st.session_state.generated_code = code
             else:
                 st.error("Failed to generate code. Please try again.")
-
+    
     if st.session_state.generated_code:
-        with st.expander("Generated Code", expanded=True):
-            st.markdown(f"```python\n{st.session_state.generated_code}\n```")
+        with st.expander("📜 Generated Code", expanded=True):
+            st.code(st.session_state.generated_code, language.lower())
+            
+            col3, col4 = st.columns([1, 1])
+            with col3:
+                st.download_button("💾 Download Code", st.session_state.generated_code, file_name=f"generated.{language.lower()}")
+            with col4:
+                st.button("📋 Copy Code", on_click=lambda: st.session_state.update({'clipboard': st.session_state.generated_code}))
 
 if __name__ == "__main__":
-    main()
+    main()