Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import nltk
-import matplotlib.pyplot as plt
+import plotly.graph_objects as go
 from nltk.sentiment.vader import SentimentIntensityAnalyzer
 
 nltk.download("vader_lexicon")
@@ -14,14 +14,27 @@ def sentiment_analysis(text):
     values = list(scores.values())
     colors = ['red' if label == 'neg' else 'green' if label == 'pos' else 'white' for label in labels]
     
-    plt.figure(figsize=(5, 3))
-    plt.bar(labels, values, color=colors)
-    plt.ylim(0, 1)
-    plt.title("Sentiment Analysis")
-    plt.xlabel("Sentiment")
-    plt.ylabel("Score")
+    fig = go.Figure(go.Bar(
+        x=values,
+        y=labels,
+        orientation='h',
+        marker=dict(color=colors)
+    ))
     
-    plt.savefig('sentiment_chart.png')
+    fig.update_layout(
+        title="Sentiment Analysis",
+        xaxis_title="Score",
+        yaxis_title="Sentiment",
+        xaxis=dict(range=[0, 1]),
+        yaxis=dict(showgrid=True),
+        plot_bgcolor='rgba(0,0,0,0)',
+        paper_bgcolor='rgba(0,0,0,0)'
+    )
+    
+    fig.update_xaxes(showgrid=True)
+    fig.update_yaxes(showgrid=True)
+    
+    fig.write_image('sentiment_chart.png')
     return gr.Image('sentiment_chart.png')
 
 demo = gr.Interface(