polish

app.py

@@ -14,7 +14,12 @@ BODY_ORDER = ['Very light-bodied', 'Light-bodied', 'Medium-bodied', 'Full-bodied
 ACIDITY_ORDER = ['Low', 'Medium', 'High']
 BODY_MAPPING = {'Very light-bodied': 1, 'Light-bodied': 2, 'Medium-bodied': 3, 'Full-bodied': 4, 'Very full-bodied': 5}
 WINE_TYPE_ORDER = {'Red': 2, 'Rosé': 1, 'White': 0}
-SAMPLE_THRESHOLDS = {'Common (100+)': 100, 'Uncommon (50+)': 50, 'Rare (20+)': 20}
+SAMPLE_THRESHOLDS = {
+    'Very Common (250+)': 250,
+    'Common (100+)': 100, 
+    'Uncommon (50+)': 50, 
+    'Rare (20+)': 20
+}
 COUNTRY_FLAGS = {
     'United States': '🇺🇸', 'France': '🇫🇷', 'Italy': '🇮🇹', 'Spain': '🇪🇸', 'Germany': '🇩🇪', 'Australia': '🇦🇺',
     'Chile': '🇨🇱', 'Argentina': '🇦🇷', 'Portugal': '🇵🇹', 'South Africa': '🇿🇦', 'New Zealand': '🇳🇿',
@@ -270,8 +275,8 @@ def create_wine_chart(chart_data, regional_grouping):
     fig = make_subplots(
         rows=1, cols=5,
         specs=[[{}, {"type": "bar"}, {"type": "bar"}, {"type": "box"}, {}]],
-        column_widths=[0.30, 0.25, 0.25, 0.15, 0.05],
-        horizontal_spacing=0.02,
+        column_widths=[0.25, 0.22, 0.22, 0.16, 0.05],
+        horizontal_spacing=0.025,
         shared_yaxes=True
     )
 
@@ -293,7 +298,7 @@ def create_wine_chart(chart_data, regional_grouping):
     fig.add_trace(go.Scatter(
         y=y_labels, x=[0.03] * num_rows, mode='text',
         text=y_labels, textposition='middle right',
-        textfont={'size': 14, 'color': '#2F2F2F'},
+        textfont={'size': 22, 'color': '#1A1A1A'},
         hoverinfo='none', showlegend=False
     ), row=1, col=1)
 
@@ -350,7 +355,7 @@ def create_wine_chart(chart_data, regional_grouping):
     fig.add_trace(go.Scatter(
         y=y_labels, x=[0.5] * num_rows, mode='text',
         text=chart_data['pairing_emoji'], textposition='middle center',
-        textfont={'size': 22}, showlegend=False,
+        textfont={'size': 32}, showlegend=False,
         hoverinfo='text', hovertext=chart_data['pairing_names']
     ), row=1, col=5)
 
@@ -358,22 +363,22 @@ def create_wine_chart(chart_data, regional_grouping):
         title={
             'text': "Wine Characteristics by Grape Variety",
             'x': 0.5,
-            'font': {'size': 24, 'color': '#2F2F2F'}
+            'font': {'size': 26, 'color': '#1A1A1A'}
         },
-        height=max(500, num_rows * 40),
-        barmode='stack', showlegend=False, plot_bgcolor='#FFFFFF', paper_bgcolor='#F5F5F5',
-        margin=dict(l=10, r=10, t=120, b=20), boxgap=0.5, bargap=0.4
+        height=max(600, num_rows * 55),
+        barmode='stack', showlegend=False, plot_bgcolor='#FAFAFA', paper_bgcolor='#E8E9EA',
+        margin=dict(l=30, r=30, t=100, b=60), boxgap=0.3, bargap=0.2
     )
 
-    column_titles = ["Wine / Hover for Info", "Body Profile (%)", "Acidity Profile (%)", "Alcohol (ABV %)", "Food Pairing"]
+    column_titles = ["Grape Variety", "Body Profile (%)", "Acidity Profile (%)", "Alcohol (ABV %)", "Food Pairing"]
     # Add column titles
     title_start = time.time()
     for i, title in enumerate(column_titles, 1):
         domain = fig.layout[f'xaxis{i if i > 1 else ""}'].domain
         fig.add_annotation(
-            x=(domain[0] + domain[1]) / 2, y=1.05,
+            x=(domain[0] + domain[1]) / 2, y=1.02,
             xref="paper", yref="paper", text=f"<b>{title}</b>",
-            xanchor='center', showarrow=False, font={'size': 14, 'color': '#2F2F2F'}
+            xanchor='center', showarrow=False, font={'size': 20, 'color': '#1A1A1A'}
         )
     print(f"[TIMING] Column titles completed in {time.time() - title_start:.2f}s")
 
@@ -381,7 +386,12 @@ def create_wine_chart(chart_data, regional_grouping):
     axes_start = time.time()
     for i in range(1, 6):
         fig.update_yaxes(showticklabels=False, showgrid=False, zeroline=False, row=1, col=i)
-        fig.update_xaxes(showticklabels=False, showgrid=False, zeroline=False, title_text="", row=1, col=i)
+        # Show x-axis labels only for body, acidity, and ABV columns
+        if i in [2, 3, 4]:
+            fig.update_xaxes(showticklabels=True, showgrid=True, gridcolor='rgba(0,0,0,0.1)', 
+                           zeroline=False, title_text="", row=1, col=i)
+        else:
+            fig.update_xaxes(showticklabels=False, showgrid=False, zeroline=False, title_text="", row=1, col=i)
     print(f"[TIMING] Axes formatting completed in {time.time() - axes_start:.2f}s")
 
     # Final axis configuration
@@ -390,14 +400,12 @@ def create_wine_chart(chart_data, regional_grouping):
                      row=1, col=1)
     print(f"[TIMING] Final axis configuration completed in {time.time() - final_axes_start:.2f}s")
 
-    # Skip alternating row backgrounds for better performance
-    # The rectangles were causing major slowdown (3+ seconds)
-    # bg_start = time.time()
-    # for i in range(num_rows):
-    #     if i % 2 == 1:
-    #         fig.add_hrect(y0=i - 0.5, y1=i + 0.5, fillcolor="#F0F0F0", layer="below", line_width=0, row=1, col="all")
-    # print(f"[TIMING] Background rectangles completed in {time.time() - bg_start:.2f}s")
-    print("[TIMING] Skipped background rectangles for performance")
+    # Add alternating row backgrounds for better readability
+    bg_start = time.time()
+    for i in range(num_rows):
+        if i % 2 == 1:
+            fig.add_hrect(y0=i - 0.5, y1=i + 0.5, fillcolor="#F0F2F3", layer="below", line_width=0, row=1, col="all")
+    print(f"[TIMING] Background rectangles completed in {time.time() - bg_start:.2f}s")
 
     chart_time = time.time() - chart_start
     print(f"[TIMING] Chart creation completed in {chart_time:.2f}s")
@@ -446,8 +454,8 @@ def create_interface():
                 info="1: Varietals, 5: All Blends"
             )
             min_samples_choice = gr.Radio(
-                choices=list(SAMPLE_THRESHOLDS.keys()), value='Common (100+)',
-                label="Minimum Sample Size"
+                choices=list(SAMPLE_THRESHOLDS.keys()), value='Very Common (250+)',
+                label="Minimum Sample Size (Wines per Variety)",
             )
             regional_grouping = gr.Checkbox(
                 value=True, label="Split By Country"