updated

app.py

@@ -8,6 +8,19 @@ st.set_page_config(
     layout="wide"
 )
 
+# Custom CSS
+st.markdown("""
+    <style>
+    .stMultiSelect {
+        max-width: 800px;
+    }
+    .main > div {
+        padding-left: 1rem;
+        padding-right: 1rem;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
 # Load data
 @st.cache_data
 def load_data():
@@ -20,14 +33,16 @@ df = load_data()
 # Title
 st.title("LLM Max Output Tokens Analysis")
 
-# Company selection
-companies = sorted(df['company'].unique())
-selected_companies = st.multiselect(
-    "Select companies to display:",
-    options=companies,
-    default=companies,
-    key='company_filter'
-)
+# Company selection in a more compact layout
+col1, col2 = st.columns([2, 1])
+with col1:
+    companies = sorted(df['company'].unique())
+    selected_companies = st.multiselect(
+        "Select companies to display:",
+        options=companies,
+        default=companies,
+        key='company_filter'
+    )
 
 # Filter data based on selection
 filtered_df = df[df['company'].isin(selected_companies)]
@@ -44,27 +59,47 @@ fig = px.line(filtered_df,
                   'max_output_tokens': 'Max Output Tokens',
                   'company': 'Company'
               },
-              markers=True)  # Add markers to make trends clearer
+              markers=True)
 
+# Improved chart layout
 fig.update_layout(
     hovermode='x unified',
     xaxis_title="Launch Date",
     yaxis_title="Max Output Tokens",
-    yaxis_type="log",  # Using log scale for better visualization
-    height=600,  # Make chart taller
+    yaxis_type="log",
+    height=500,
     showlegend=True,
     legend=dict(
-        yanchor="top",
-        y=0.99,
-        xanchor="left",
-        x=0.01
+        orientation="h",
+        yanchor="bottom",
+        y=1.02,
+        xanchor="right",
+        x=1
+    ),
+    margin=dict(l=40, r=40, t=60, b=40),
+    yaxis=dict(
+        tickformat=",",
+        dtick=0.30102999566,  # log10(2) for better log scale ticks
     ),
-    margin=dict(l=20, r=20, t=40, b=20)
+    plot_bgcolor='white',
+    paper_bgcolor='white',
+)
+
+# Improved grid and traces
+fig.update_xaxes(
+    gridcolor='lightgray',
+    gridwidth=0.5,
+    showgrid=True
+)
+fig.update_yaxes(
+    gridcolor='lightgray',
+    gridwidth=0.5,
+    showgrid=True
 )
 
 fig.update_traces(
-    line=dict(width=2),  # Make lines thicker
-    marker=dict(size=8)  # Make markers more visible
+    line=dict(width=2),
+    marker=dict(size=8)
 )
 
 # Display the chart
@@ -96,9 +131,12 @@ st.dataframe(
     hide_index=True
 )
 
-# Attribution
-st.markdown("---")
+# Attribution with better spacing
+st.markdown("<br>", unsafe_allow_html=True)
 st.markdown(
-    "By: [Daniel Rosehill](https://danielrosehill.com) | "
+    "<div style='border-top: 1px solid #ccc; padding-top: 1rem; color: #666;'>"
+    "By: <a href='https://danielrosehill.com' style='color: #666;'>Daniel Rosehill</a> | "
     "Data sourced from public sources on February 8, 2025"
+    "</div>",
+    unsafe_allow_html=True
 )