Spaces:
Runtime error
Runtime error
macrdel
commited on
Commit
•
3a69ea2
1
Parent(s):
84bcdcd
update pipeline sentiment src.py
Browse files- app/src/src.py +5 -4
app/src/src.py
CHANGED
@@ -42,18 +42,19 @@ def get_comments(api_key, video_id):
|
|
42 |
for num, x in enumerate(res['items'])
|
43 |
}
|
44 |
|
45 |
-
def get_sentim(data,
|
46 |
"""Get result of sentimental analysis"""
|
47 |
-
res =
|
|
|
48 |
return res['label'], res['score']
|
49 |
|
50 |
-
def pipeline_sentiment(url_video, api_key,
|
51 |
"""Pipeline of sentimental analysis"""
|
52 |
video_id = get_video_id(url_video)
|
53 |
comments = get_comments(api_key, video_id)
|
54 |
comments_df = pd.DataFrame(comments).T
|
55 |
|
56 |
-
text_tuple = [get_sentim(i,
|
57 |
comments_df[["sentiment", "score"]] = pd.DataFrame(list(text_tuple))
|
58 |
return comments_df
|
59 |
|
|
|
42 |
for num, x in enumerate(res['items'])
|
43 |
}
|
44 |
|
45 |
+
def get_sentim(data, headers, url):
|
46 |
"""Get result of sentimental analysis"""
|
47 |
+
res = requests.post(url, headers=headers, json=data)
|
48 |
+
res = res.json()[0][0]
|
49 |
return res['label'], res['score']
|
50 |
|
51 |
+
def pipeline_sentiment(url_video, api_key, headers, url):
|
52 |
"""Pipeline of sentimental analysis"""
|
53 |
video_id = get_video_id(url_video)
|
54 |
comments = get_comments(api_key, video_id)
|
55 |
comments_df = pd.DataFrame(comments).T
|
56 |
|
57 |
+
text_tuple = [get_sentim(i, headers, url) for i in comments_df["text_comment"]]
|
58 |
comments_df[["sentiment", "score"]] = pd.DataFrame(list(text_tuple))
|
59 |
return comments_df
|
60 |
|