youssef1214
commited on
Commit
·
c5a320e
1
Parent(s):
5aeccd7
Update Database.py
Browse files- Database.py +7 -2
Database.py
CHANGED
@@ -30,8 +30,8 @@ all_reviews = db.collection("complaints")
|
|
30 |
# Create a query against the collection
|
31 |
documents_to_summarize = all_reviews.get()
|
32 |
feedbacks = db.collection("feedbacks").where("feedbacks","!=",'').get()
|
33 |
-
|
34 |
-
|
35 |
documents=[]
|
36 |
documnetskey=set()
|
37 |
#get all documents for today that have no summary
|
@@ -151,11 +151,16 @@ def on_snapshot(doc_snapshot, changes, read_time):
|
|
151 |
def on_snapshotfeedback(doc_snapshot, changes, read_time):
|
152 |
global queuedUnLabeledFeedbacks
|
153 |
global semphoreFeedback
|
|
|
154 |
for doc in doc_snapshot:
|
155 |
feedback=Classes.Feedback.from_dict(source=doc.to_dict())
|
|
|
156 |
if(feedback.label==None):
|
157 |
queuedUnLabeledFeedbacks.append(feedback)
|
158 |
semphoreFeedback.release()
|
|
|
|
|
|
|
159 |
|
160 |
callback_donefeedback.set()
|
161 |
docwatch= db.collection("complaints").on_snapshot(on_snapshot,)
|
|
|
30 |
# Create a query against the collection
|
31 |
documents_to_summarize = all_reviews.get()
|
32 |
feedbacks = db.collection("feedbacks").where("feedbacks","!=",'').get()
|
33 |
+
allfeedbacks=[]
|
34 |
+
emotioncount={"anger":0,"sadness":0,"joy":0,"surprise":0,"love":0,"sympathy":0,"fear":0}
|
35 |
documents=[]
|
36 |
documnetskey=set()
|
37 |
#get all documents for today that have no summary
|
|
|
151 |
def on_snapshotfeedback(doc_snapshot, changes, read_time):
|
152 |
global queuedUnLabeledFeedbacks
|
153 |
global semphoreFeedback
|
154 |
+
global allfeedbacks
|
155 |
for doc in doc_snapshot:
|
156 |
feedback=Classes.Feedback.from_dict(source=doc.to_dict())
|
157 |
+
|
158 |
if(feedback.label==None):
|
159 |
queuedUnLabeledFeedbacks.append(feedback)
|
160 |
semphoreFeedback.release()
|
161 |
+
elif feedback not in allfeedbacks:
|
162 |
+
allfeedbacks.append(feedback)
|
163 |
+
emotioncount[feedback.label]+=1
|
164 |
|
165 |
callback_donefeedback.set()
|
166 |
docwatch= db.collection("complaints").on_snapshot(on_snapshot,)
|