youssef1214 commited on
Commit
c5a320e
·
1 Parent(s): 5aeccd7

Update Database.py

Browse files
Files changed (1) hide show
  1. 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,)