alexneakameni commited on
Commit
9c7849e
·
verified ·
1 Parent(s): d942552

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -107,21 +107,27 @@ class VoterCardVerifier:
107
  def save_card_number(self, ocr_data, card_number):
108
  return self.db.add(str(ocr_data), str(card_number))
109
 
 
 
 
 
 
110
 
111
  # Interface Gradio
112
  def create_interface():
 
 
113
  description = (
114
  "Card ID Counter is an application designed to count voter card numbers "
115
  "by analyzing uploaded images using OCR technology. It ensures accuracy by "
116
  "checking if the provided number appears on the card and confirming its from Cameroon.\n\n"
 
117
  "🌟 **Code Repository**: [Card ID Counter GitHub](https://github.com/Nganga-AI/card-counter)"
118
  )
119
 
120
- verifier = VoterCardVerifier()
121
-
122
  def process(image, card_number: str):
123
  if not card_number.strip():
124
- return "Veuillez entrer un numéro de carte"
125
  return verifier.verify_card(image, card_number)
126
 
127
  interface = gr.Interface(
 
107
  def save_card_number(self, ocr_data, card_number):
108
  return self.db.add(str(ocr_data), str(card_number))
109
 
110
+ def get_id_counted(self):
111
+ docs = self.db.db.collection(collecttion_name).stream()
112
+ count = sum(1 for _ in docs)
113
+ return count
114
+
115
 
116
  # Interface Gradio
117
  def create_interface():
118
+ verifier = VoterCardVerifier()
119
+ n_cards = verifier.get_id_counted()
120
  description = (
121
  "Card ID Counter is an application designed to count voter card numbers "
122
  "by analyzing uploaded images using OCR technology. It ensures accuracy by "
123
  "checking if the provided number appears on the card and confirming its from Cameroon.\n\n"
124
+ f"Card Count: {n_cards}\n\n"
125
  "🌟 **Code Repository**: [Card ID Counter GitHub](https://github.com/Nganga-AI/card-counter)"
126
  )
127
 
 
 
128
  def process(image, card_number: str):
129
  if not card_number.strip():
130
+ return "Enter your elect count number"
131
  return verifier.verify_card(image, card_number)
132
 
133
  interface = gr.Interface(