Adding pie chart

app.py

@@ -43,14 +43,26 @@ def analyze_sentiment(text):
     else:
         return "Unknown"  # O algún valor por defecto para datos no procesables
 
+def plot_pie_chart(sentiment_counts):
+    labels = sentiment_counts.keys()
+    sizes = sentiment_counts.values()
+    fig, ax = plt.subplots()
+    ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90)
+    ax.axis('equal')  # Equal aspect ratio ensures that pie is drawn as a circle.
+    return fig
+
 def sentiment_counts_by_category(category):
     df_category = load_reviews(category)
-    df_category = df_category.head(30)
-    # Aplica la función de análisis de sentimientos modificada
+    df_category = df_category.head(30)  # Limita el análisis a las primeras 30 reseñas
     df_category['sentiment'] = df_category['reviewText'].apply(analyze_sentiment)
-    # Llama a la función de trazado y devuelve el gráfico
-    fig = plot_sentiment_distribution(df_category)
-    return fig
+    # Conteo de sentimientos
+    sentiment_counts = df_category['sentiment'].value_counts(normalize=True)  # Obtiene porcentajes
+    # Genera la gráfica de barras para la distribución de sentimientos
+    bar_fig = plot_sentiment_distribution(df_category)
+    # Genera la gráfica de pastel para la distribución porcentual de sentimientos
+    pie_fig = plot_pie_chart(sentiment_counts)
+    # Retorna ambas figuras
+    return bar_fig, pie_fig
 
 def show_first_five(category):
     df_category = load_reviews(category)
@@ -71,10 +83,11 @@ with gr.Blocks() as demo:
             plot_button = gr.Button("Graficar Distribución de Sentimientos")
         with gr.Column():
             output_df = gr.Dataframe()
-            output_plot = gr.Plot()
+            output_bar_plot = gr.Plot()
+            output_pie_plot = gr.Plot()
 
     show_button.click(show_first_five, inputs=category, outputs=output_df)
-    plot_button.click(sentiment_counts_by_category, inputs=category, outputs=output_plot)
+    plot_button.click(sentiment_counts_by_category, inputs=category, outputs=[output_bar_plot, output_pie_plot])
 
 # Lanza la interfaz
 if __name__ == "__main__":