Pro-Coder commited on
Commit
a35916e
·
verified ·
1 Parent(s): a4a72fe

Create utils.py

Browse files
Files changed (1) hide show
  1. modules/utils.py +40 -0
modules/utils.py ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # modules/utils.py
2
+
3
+ import pandas as pd
4
+
5
+ def save_results_to_csv(sentiment_results: list, summaries: list, filepath: str = "results.csv"):
6
+ """
7
+ Save sentiment analysis and summaries into a CSV file.
8
+ """
9
+ data = []
10
+
11
+ for i, sent in enumerate(sentiment_results):
12
+ summary = summaries[i]["summary"] if i < len(summaries) else ""
13
+ data.append({
14
+ "Text": sent.get("text", ""),
15
+ "Sentiment": sent.get("label", ""),
16
+ "Score": sent.get("score", 0.0),
17
+ "Summary": summary
18
+ })
19
+
20
+ df = pd.DataFrame(data)
21
+ df.to_csv(filepath, index=False, encoding="utf-8")
22
+
23
+ return filepath
24
+
25
+
26
+ def display_dataframe(sentiment_results: list, summaries: list):
27
+ """
28
+ Create a pandas DataFrame for displaying results inside UI.
29
+ """
30
+ data = []
31
+ for i, sent in enumerate(sentiment_results):
32
+ summary = summaries[i]["summary"] if i < len(summaries) else ""
33
+ data.append({
34
+ "Text": sent.get("text", ""),
35
+ "Sentiment": sent.get("label", ""),
36
+ "Score": sent.get("score", 0.0),
37
+ "Summary": summary
38
+ })
39
+
40
+ return pd.DataFrame(data)