Fix calendar and numpy imports in code generation

- Add calendar and numpy imports to code template
- Include these modules in execution environment globals
- Add numpy to requirements.txt
- Improve code styling with gradients and better visual design
- Replace basic feedback with 4-point rating system (Excellent/Good/Okay/Poor)

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

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

app.py

@@ -230,7 +230,7 @@ st.markdown("""
     color: #475569;
     border: 1px solid #e2e8f0;
     padding: 0.375rem 0.75rem;
-    font-size: 0.7rem;
+    font-size: 0.65rem;
     font-weight: normal;
     text-align: left;
     white-space: normal;
@@ -258,45 +258,57 @@ st.markdown("""
 /* Code container styling */
 .code-container {
     margin: 1rem 0;
-    border: 1px solid #e2e8f0;
-    border-radius: 8px;
-    background: #f8fafc;
+    border: 1px solid #d1d5db;
+    border-radius: 12px;
+    background: white;
+    box-shadow: 0 1px 3px rgba(0, 0, 0, 0.1);
 }
 
 .code-header {
     display: flex;
     justify-content: space-between;
     align-items: center;
-    padding: 0.75rem 1rem;
-    background: #f1f5f9;
+    padding: 0.875rem 1.25rem;
+    background: linear-gradient(135deg, #f8fafc 0%, #f1f5f9 100%);
     border-bottom: 1px solid #e2e8f0;
     cursor: pointer;
-    transition: background-color 0.2s;
+    transition: all 0.2s ease;
+    border-radius: 12px 12px 0 0;
 }
 
 .code-header:hover {
-    background: #e2e8f0;
+    background: linear-gradient(135deg, #e2e8f0 0%, #cbd5e1 100%);
 }
 
 .code-title {
-    font-size: 0.875rem;
-    font-weight: 500;
-    color: #374151;
+    font-size: 0.9rem;
+    font-weight: 600;
+    color: #1e293b;
+    display: flex;
+    align-items: center;
+    gap: 0.5rem;
+}
+
+.code-title:before {
+    content: "⚡";
+    font-size: 0.8rem;
 }
 
 .toggle-text {
     font-size: 0.75rem;
-    color: #6b7280;
+    color: #64748b;
+    font-weight: 500;
 }
 
 .code-block {
-    background: #1e293b;
+    background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%);
     color: #e2e8f0;
-    padding: 1rem;
-    font-family: 'Monaco', 'Menlo', monospace;
+    padding: 1.5rem;
+    font-family: 'SF Mono', 'Monaco', 'Menlo', 'Consolas', monospace;
     font-size: 0.875rem;
     overflow-x: auto;
-    line-height: 1.5;
+    line-height: 1.6;
+    border-radius: 0 0 12px 12px;
 }
 
 .answer-container {
@@ -665,25 +677,6 @@ if not st.session_state.responses:
     </div>
     """, unsafe_allow_html=True)
 
-# Workflow steps section (like in mockup)
-st.markdown("""
-<div style='background: #f8fafc; border-bottom: 1px solid #e2e8f0; padding: 1rem 2rem; display: flex; gap: 1rem; margin-bottom: 1rem;'>
-    <div style='display: flex; align-items: center; gap: 0.5rem; padding: 0.5rem 1rem; border-radius: 6px; font-size: 0.875rem; font-weight: 500; background: #dcfce7; color: #166534;'>
-        <div style='width: 20px; height: 20px; border-radius: 50%; background: #22c55e; color: white; display: flex; align-items: center; justify-content: center; font-size: 0.75rem; font-weight: 600;'>1</div>
-        <span>Natural Language Query</span>
-    </div>
-    <div style='margin: 0 1rem; color: #9ca3af;'>→</div>
-    <div style='display: flex; align-items: center; gap: 0.5rem; padding: 0.5rem 1rem; border-radius: 6px; font-size: 0.875rem; font-weight: 500; background: #dcfce7; color: #166534;'>
-        <div style='width: 20px; height: 20px; border-radius: 50%; background: #22c55e; color: white; display: flex; align-items: center; justify-content: center; font-size: 0.75rem; font-weight: 600;'>2</div>
-        <span>Code Generation</span>
-    </div>
-    <div style='margin: 0 1rem; color: #9ca3af;'>→</div>
-    <div style='display: flex; align-items: center; gap: 0.5rem; padding: 0.5rem 1rem; border-radius: 6px; font-size: 0.875rem; font-weight: 500; background: #dbeafe; color: #1d4ed8;'>
-        <div style='width: 20px; height: 20px; border-radius: 50%; background: #3b82f6; color: white; display: flex; align-items: center; justify-content: center; font-size: 0.75rem; font-weight: 600;'>3</div>
-        <span>Visualization</span>
-    </div>
-</div>
-""", unsafe_allow_html=True)
 
 
 def show_custom_response(response):
@@ -797,16 +790,28 @@ with chat_container:
             else:
                 # Beautiful feedback section
                 st.markdown("---")
-                st.markdown("**How was this response?**")
+                st.markdown("**Rate this response:**")
                 
-                col1, col2 = st.columns(2)
+                # More detailed feedback options
+                col1, col2, col3, col4 = st.columns(4)
                 with col1:
-                    thumbs_up = st.button("👍 Helpful", key=f"{feedback_key}_up", use_container_width=True)
+                    excellent = st.button("🎯 Excellent", key=f"{feedback_key}_excellent", use_container_width=True)
                 with col2:
-                    thumbs_down = st.button("👎 Not Helpful", key=f"{feedback_key}_down", use_container_width=True)
+                    good = st.button("✅ Good", key=f"{feedback_key}_good", use_container_width=True)
+                with col3:
+                    okay = st.button("⚠️ Okay", key=f"{feedback_key}_okay", use_container_width=True)
+                with col4:
+                    poor = st.button("❌ Poor", key=f"{feedback_key}_poor", use_container_width=True)
                 
-                if thumbs_up or thumbs_down:
-                    thumbs = "👍 Helpful" if thumbs_up else "👎 Not Helpful"
+                if excellent or good or okay or poor:
+                    if excellent:
+                        thumbs = "🎯 Excellent"
+                    elif good:
+                        thumbs = "✅ Good"
+                    elif okay:
+                        thumbs = "⚠️ Okay"
+                    else:
+                        thumbs = "❌ Poor"
                     comments = st.text_area(
                         "Tell us more (optional):", 
                         key=f"{feedback_key}_comments",

src.py

@@ -244,7 +244,7 @@ def ask_question(model_name, question):
         # Professional matplotlib styling
         parameters = {
             "font.size": 11,
-            "figure.dpi": 150,
+            "figure.dpi": 300,
             "figure.facecolor": "white",
             "axes.facecolor": "white",
             "axes.edgecolor": "#e2e8f0",
@@ -269,8 +269,31 @@ def ask_question(model_name, question):
 import pandas as pd
 import matplotlib.pyplot as plt
 import uuid
+import calendar
+import numpy as np
 
-plt.rcParams.update({parameters})
+# Set professional matplotlib styling
+plt.rcParams.update({{
+    'font.size': 11,
+    'figure.dpi': 300,
+    'figure.facecolor': 'white',
+    'axes.facecolor': 'white',
+    'axes.edgecolor': '#e2e8f0',
+    'axes.linewidth': 1.2,
+    'axes.labelcolor': '#374151',
+    'axes.spines.top': False,
+    'axes.spines.right': False,
+    'axes.spines.left': True,
+    'axes.spines.bottom': True,
+    'axes.grid': True,
+    'grid.color': '#f1f5f9',
+    'grid.linewidth': 0.8,
+    'grid.alpha': 0.7,
+    'xtick.color': '#6b7280',
+    'ytick.color': '#6b7280',
+    'text.color': '#374151',
+    'figure.figsize': [12, 6]
+}})
 
 df = pd.read_csv("Data.csv")
 df["Timestamp"] = pd.to_datetime(df["Timestamp"])
@@ -332,7 +355,9 @@ Complete the following code to answer the user's question:
                 'pd': pd,
                 'plt': plt,
                 'os': os,
-                'uuid': __import__('uuid')
+                'uuid': __import__('uuid'),
+                'calendar': __import__('calendar'),
+                'np': __import__('numpy')
             }
             
             exec(full_code, global_vars, local_vars)