Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import pandas as pd
 from transformers import pipeline
+import matplotlib.pyplot as plt
 
 # Configurar el clasificador de sentimientos multilingüe
 classifier = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
@@ -8,7 +9,7 @@ classifier = pipeline(task="zero-shot-classification", model="facebook/bart-larg
 # Función para analizar los sentimientos de una lista de textos
 def analyze_sentiments(texts):
     if not texts:
-        return "0.0%", "0.0%", "0.0%"  # Manejar el caso donde no hay textos para analizar
+        return "0.0%", "0.0%", "0.0%", None  # Manejar el caso donde no hay textos para analizar
 
     positive, negative, neutral = 0, 0, 0
     for text in texts:
@@ -26,7 +27,15 @@ def analyze_sentiments(texts):
     positive_percent = round((positive / total) * 100, 1)
     negative_percent = round((negative / total) * 100, 1)
     neutral_percent = round((neutral / total) * 100, 1)
-    return f"{positive_percent}%", f"{negative_percent}%", f"{neutral_percent}%"
+    
+    # Crear el gráfico circular
+    fig, ax = plt.subplots()
+    ax.pie([positive_percent, negative_percent, neutral_percent], labels=["Positivo", "Negativo", "Neutro"], autopct='%1.1f%%', colors=['green', 'red', 'blue'])
+    plt.title("Distribución de Sentimientos")
+    plt.savefig("sentiment_pie_chart.png")
+    plt.close(fig)
+
+    return f"{positive_percent}%", f"{negative_percent}%", f"{neutral_percent}%", "sentiment_pie_chart.png"
 
 # Función para cargar el archivo CSV y analizar los primeros 100 comentarios
 def analyze_sentiment_from_csv(file):
@@ -37,9 +46,9 @@ def analyze_sentiment_from_csv(file):
         texts = df['content'].head(100).tolist()  # Tomar solo los primeros 100 comentarios
         return analyze_sentiments(texts)
     except pd.errors.ParserError as e:
-        return f"Error al analizar el archivo CSV: {e}", "", ""
+        return f"Error al analizar el archivo CSV: {e}", "", "", None
     except Exception as e:
-        return f"Error inesperado: {e}", "", ""
+        return f"Error inesperado: {e}", "", "", None
 
 # Configurar la interfaz de Gradio
 demo = gr.Interface(
@@ -48,12 +57,12 @@ demo = gr.Interface(
     outputs=[
         gr.Textbox(label="Porcentaje Positivo"),
         gr.Textbox(label="Porcentaje Negativo"),
-        gr.Textbox(label="Porcentaje Neutro")
+        gr.Textbox(label="Porcentaje Neutro"),
+        gr.Image(type="filepath", label="Gráfico de Sentimientos")
     ],
-    title="Analizador de Sentimientos V.2",
-   	description="Porcentaje de comentarios positivos, negativos y neutrales"
+    title="Analizador de Sentimientos V.3",
+    description="Porcentaje de comentarios positivos, negativos y neutrales. Y GRAFICO CIRCULAR"
 )
 
 demo.launch(share=True)
 
-