Spaces:
Runtime error
Runtime error
File size: 2,596 Bytes
2116489 ec163c5 2116489 ec163c5 2116489 ec163c5 2116489 ec163c5 2116489 ec163c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
import transformers
from flask import Flask, request, jsonify
from transformers import RobertaTokenizerFast, TFRobertaForSequenceClassification, pipeline
import gradio as gr
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import io
from io import BytesIO # Import BytesIO for image generation
# Load model and tokenizer
tokenizer = RobertaTokenizerFast.from_pretrained("arpanghoshal/EmoRoBERTa")
model = TFRobertaForSequenceClassification.from_pretrained("arpanghoshal/EmoROBERTa")
emotion = pipeline("sentiment-analysis", model="arpanghoshal/EmoROBERTa")
def analyze_csv(file):
try:
# Print file content for debugging
file_content = file.read()
print("File content:", file_content)
# Reset file position to the beginning
file.seek(0)
# Read the CSV file into a DataFrame
df = pd.read_csv(io.BytesIO(file_content))``
print("DataFrame shape:", df.shape) # Print DataFrame shape for debugging
print("DataFrame columns:", df.columns) # Print DataFrame columns for debugging
# Check if the DataFrame is empty
if df.empty:
return "Empty file. Please upload a CSV file with data.", None
# Check if the expected column "phrase" is present in the DataFrame
if "phrase" not in df.columns:
return "Column 'phrase' not found in the CSV file. Please check the file format.", None
phrases = df["phrase"]
# Analyze sentiment for each phrase
emotion_labels = emotion(phrases)
# Create summary statistics
summary_df = pd.DataFrame(emotion_labels).describe()
# Create a bar chart of emotion distribution
plt.figure()
emotion_counts = emotion_labels.get("labels").value_counts()
emotion_counts.plot(kind="bar")
plt.title("Emotion Distribution")
plt.xlabel("Emotion")
plt.ylabel("Count")
# Generate PNG image of the chart
chart_img = BytesIO()
plt.savefig(chart_img, format="png")
chart_img.seek(0)
return summary_df.to_json(), chart_img.read()
except Exception as e:
error_message = f"Error processing the CSV file: {str(e)}"
print(error_message) # Print the error message for debugging
return error_message, None
iface = gr.Interface(
fn=analyze_csv,
inputs=[gr.File(label="Upload CSV File")],
outputs=["dataframe", "image"],
title="Emotion Analyzer with CSV",
description="Analyzes sentiment and creates charts/tables from a CSV file.",
)
iface.launch() |