Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pandas as pd
 import numpy as np
 import plotly.express as px
+import plotly.graph_objects as go
 
 # Pricing data
 aws_instances = {
@@ -154,12 +155,41 @@ def generate_cost_comparison(
 
     # Recommendation and Breakeven omitted for brevity
 
-    # Chart
+    # Chart with annotations
     df = pd.DataFrame(results)
     colors = {r['provider']: c for r,c in zip(results, ['#FF9900','#4285F4','#D62828'])}
-    fig = px.bar(df, x='provider', y='cost', color='provider', color_discrete_map=colors)
-    fig.update_yaxes(tickprefix='$')
-    fig.update_layout(showlegend=False, height=500)
+    
+    # Create figure using plotly graph objects for more control
+    fig = go.Figure()
+    
+    # Add bars
+    for i, row in df.iterrows():
+        fig.add_trace(go.Bar(
+            x=[row['provider']], 
+            y=[row['cost']], 
+            name=row['provider'],
+            marker_color=colors[row['provider']]
+        ))
+    
+    # Add annotations on top of each bar
+    for i, row in df.iterrows():
+        fig.add_annotation(
+            x=row['provider'],
+            y=row['cost'],
+            text=f"${row['cost']:.2f}",
+            showarrow=False,
+            yshift=10,  # Position above the bar
+            font=dict(size=14)
+        )
+    
+    # Update layout
+    fig.update_layout(
+        showlegend=False,
+        height=500,
+        yaxis=dict(title='Monthly Cost ($)', tickprefix='$'),
+        xaxis=dict(title=''),
+        title='Cost Comparison'
+    )
 
     html = f"""
     <div style='padding:20px;font-family:Arial;'>
@@ -193,4 +223,4 @@ with gr.Blocks(title="Cloud Cost Estimator", theme=gr.themes.Soft(primary_hue="i
                          model_size, storage_gb, reserved_instances, spot_instances, multi_year_commitment],
                  outputs=[out_html, out_plot])
 
-demo.launch()
+demo.launch()