Fix button layout and spacing for professional UI

LAYOUT IMPROVEMENTS:
- Create compact single-row header with logo and text side-by-side
- Fix cramped feedback/retry button layout with proper spacing
- Add generous margins and padding for better visual hierarchy
- Use balanced column ratios [2, 2, 1, 1] for optimal button sizing

BUTTON DESIGN:
- Single row layout: [✨ Excellent] [🔧 Needs work] [space] [🔄 Retry]
- Full-width buttons with use_container_width=True for consistency
- Beautiful green gradient feedback confirmation
- Clean alignment and professional appearance

SPACE OPTIMIZATION:
- Compact header saves vertical space while maintaining branding
- Proper spacer above action buttons (1.5rem margin)
- Eliminated awkward multi-line button layouts
- Much better proportions and visual balance

Perfect for screenshots and professional demos\! 📸✨

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

app.py

@@ -573,35 +573,34 @@ if "gpt-oss-120b" in available_models:
 elif "deepseek-R1" in available_models:
     default_index = available_models.index("deepseek-R1")
 
-# Beautiful header with logo and branding
-col1, col2, col3 = st.columns([1, 2, 1])
-with col2:
-    st.markdown("""
-    <div style='text-align: center; padding: 1rem 0;'>
-        <img src='https://sustainability-lab.github.io/images/logo_light.svg' 
-             style='height: 60px; margin-bottom: 10px;' />
+# Compact header - everything in one row
+st.markdown("""
+<div style='
+    display: flex; 
+    align-items: center; 
+    justify-content: center; 
+    padding: 0.5rem 0; 
+    gap: 15px;
+    border-bottom: 1px solid #e5e7eb;
+    margin-bottom: 1rem;
+'>
+    <img src='https://sustainability-lab.github.io/images/logo_light.svg' 
+         style='height: 40px;' />
+    <div>
         <h1 style='
             margin: 0; 
-            font-size: 2.5rem; 
+            font-size: 1.8rem; 
             font-weight: 700; 
-            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-            -webkit-background-clip: text;
-            -webkit-text-fill-color: transparent;
-            background-clip: text;
+            color: #1f2937;
         '>VayuChat</h1>
         <p style='
-            margin: 5px 0 0 0; 
-            font-size: 1.1rem; 
-            color: #6b7280; 
-            font-weight: 500;
-        '>AI-Powered Air Quality Data Analysis</p>
-        <p style='
-            margin: 2px 0 0 0; 
-            font-size: 0.9rem; 
-            color: #9ca3af;
-        '>Sustainability Lab • IIT Gandhinagar</p>
+            margin: 0; 
+            font-size: 0.8rem; 
+            color: #6b7280;
+        '>AI Air Quality Analysis • Sustainability Lab, IIT Gandhinagar</p>
     </div>
-    """, unsafe_allow_html=True)
+</div>
+""", unsafe_allow_html=True)
 
 
 # Load data with caching for better performance
@@ -862,66 +861,70 @@ for response_id, response in enumerate(st.session_state.responses):
         code = response.get("gen_code", "")
 
 
-        # Awesome feedback system
+        # Beautiful action bar with feedback and retry
+        st.markdown('<div style="margin: 1.5rem 0 0.5rem 0;"></div>', unsafe_allow_html=True)  # Spacer
+        
         if "feedback" in st.session_state.responses[response_id]:
+            # Show submitted feedback nicely
             feedback_data = st.session_state.responses[response_id]["feedback"]
-            st.markdown(f"""
-            <div style='
-                background: linear-gradient(135deg, #f3f4f6 0%, #e5e7eb 100%);
-                border: 1px solid #d1d5db;
-                border-radius: 12px;
-                padding: 1rem;
-                margin: 0.5rem 0;
-                display: flex;
-                align-items: center;
-                gap: 12px;
-            '>
-                <span style='font-size: 1.2rem;'>{feedback_data.get('score', '')}</span>
-                <span style='color: #6b7280; font-weight: 500;'>
-                    {feedback_data.get('text', 'Thanks for your feedback!')}
-                </span>
-            </div>
-            """, unsafe_allow_html=True)
+            col1, col2 = st.columns([3, 1])
+            with col1:
+                st.markdown(f"""
+                <div style='
+                    background: linear-gradient(135deg, #ecfdf5 0%, #d1fae5 100%);
+                    border: 1px solid #a7f3d0;
+                    border-radius: 8px;
+                    padding: 0.75rem 1rem;
+                    display: flex;
+                    align-items: center;
+                    gap: 8px;
+                '>
+                    <span style='font-size: 1.1rem;'>{feedback_data.get('score', '')}</span>
+                    <span style='color: #059669; font-weight: 500; font-size: 0.9rem;'>
+                        Thanks for your feedback!
+                    </span>
+                </div>
+                """, unsafe_allow_html=True)
+            with col2:
+                if st.button("🔄 Retry", key=f"retry_{response_id}", use_container_width=True):
+                    user_prompt = ""
+                    if response_id > 0:
+                        user_prompt = st.session_state.responses[response_id-1].get("content", "")
+                    if user_prompt:
+                        if response_id > 0:
+                            retry_prompt = st.session_state.responses[response_id-1].get("content", "")
+                            del st.session_state.responses[response_id]
+                            del st.session_state.responses[response_id-1]
+                            st.session_state.follow_up_prompt = retry_prompt
+                            st.rerun()
         else:
-            # Beautiful feedback buttons
-            st.markdown("""
-            <div style='margin: 0.5rem 0; color: #6b7280; font-size: 0.9rem; font-weight: 500;'>
-                How was this response?
-            </div>
-            """, unsafe_allow_html=True)
+            # Clean feedback and retry layout
+            col1, col2, col3, col4 = st.columns([2, 2, 1, 1])
             
-            col1, col2, col3 = st.columns([1, 1, 3])
             with col1:
-                if st.button("✨ Excellent", key=f"{feedback_key}_excellent", help="This response was very helpful!"):
+                if st.button("✨ Excellent", key=f"{feedback_key}_excellent", use_container_width=True):
                     feedback = {"score": "✨ Excellent", "text": ""}
                     st.session_state.responses[response_id]["feedback"] = feedback
                     st.rerun()
+            
             with col2:
-                if st.button("🔧 Needs work", key=f"{feedback_key}_poor", help="This response could be improved"):
+                if st.button("🔧 Needs work", key=f"{feedback_key}_poor", use_container_width=True):
                     feedback = {"score": "🔧 Needs work", "text": ""}
                     st.session_state.responses[response_id]["feedback"] = feedback
                     st.rerun()
-        
-        # Clean action buttons
-        col1, col2, col3 = st.columns([1, 4, 1])
-        with col1:
-            if st.button("🔄 Retry", key=f"retry_{response_id}", help="Regenerate with current model"):
-                # Get the last user prompt that led to this response
-                user_prompt = ""
-                if response_id > 0:
-                    user_prompt = st.session_state.responses[response_id-1].get("content", "")
-                
-                if user_prompt:
-                    # Remove this response and the user message before it, then re-add the user message
+            
+            with col4:
+                if st.button("🔄 Retry", key=f"retry_{response_id}", use_container_width=True):
+                    user_prompt = ""
                     if response_id > 0:
-                        # Store the user prompt
-                        retry_prompt = st.session_state.responses[response_id-1].get("content", "")
-                        # Remove both user message and assistant response
-                        del st.session_state.responses[response_id]
-                        del st.session_state.responses[response_id-1]
-                        # Re-add user message and trigger new response
-                        st.session_state.follow_up_prompt = retry_prompt
-                        st.rerun()
+                        user_prompt = st.session_state.responses[response_id-1].get("content", "")
+                    if user_prompt:
+                        if response_id > 0:
+                            retry_prompt = st.session_state.responses[response_id-1].get("content", "")
+                            del st.session_state.responses[response_id]
+                            del st.session_state.responses[response_id-1]
+                            st.session_state.follow_up_prompt = retry_prompt
+                            st.rerun()
 
 # Chat input with better guidance
 prompt = st.chat_input("💬 Ask about air quality trends, pollution analysis, or city comparisons...", key="main_chat")