add key verification

.gitignore

@@ -0,0 +1,2 @@
+__pycache__
+.DS_Store

app.py

@@ -1,3 +1,4 @@
+import hashlib
 from types import SimpleNamespace
 import gradio as gr
 from PIL import Image
@@ -12,11 +13,15 @@ from utils import custom_plot
 
 model = YOLO("YOLOV8s_Barcode_Detection.pt")
 reader = easyocr.Reader(['en'])
-loading_img = "loading.gif"
+key_hash = "6734e0268423441620ada6c66bf76c255500139962741833ab33185274b8a010"
+hash = lambda x: hashlib.sha256(x.encode()).hexdigest()
 
-def process_image(input_img, progress=gr.Progress()):
+
+def process_image(input_img, activation_key, progress=gr.Progress()):
     if input_img is None:
-        return None, "No image provided."
+        return None, "請上傳圖片後再進行辨識。"
+    if hash(activation_key) != key_hash:
+        return None, "金鑰錯誤，請確認後再試一次。"
 
     # Perform object detection on an image
     result = model(input_img, imgsz=(1280))[0]
@@ -77,6 +82,7 @@ def process_image(input_img, progress=gr.Progress()):
                         pred_text += cur_box[1]
                     else:
                         break
+                # HACK
                 if len(pred_text) != 15:
                     res = []
             res = [SimpleNamespace(text=pred_text)] if res else []
@@ -84,35 +90,35 @@ def process_image(input_img, progress=gr.Progress()):
         texts.append(res[0].text if res else None)
 
     output_text = '\n'.join([t for t in texts if isinstance(t, str)])
-    yield loading_img, output_text
     results_img = custom_plot(
         result, font_size=40, pil=True,
         barcode_texts=texts,
     )
-    yield results_img, output_text
+    return results_img, output_text
 
 
 
 # Defining the Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Barcode")
-    gr.Markdown("_")
+    gr.Markdown("請用下方 example 圖片測試條碼辨識功能，或上傳您自己的圖片。")
     
     with gr.Row():
         with gr.Column():
-            input_view = gr.Image(type="pil", label="Input Image")
-            btn = gr.Button("Transform", variant="primary")
+            input_view = gr.Image(type="pil", label="輸入圖片")
+            activation_key = gr.Textbox(label="金鑰")
+            btn = gr.Button("辨識", variant="primary")
 
-            gr.Examples(examples=["test.jpeg"], inputs=input_view)
+            gr.Examples(examples=[["example1.jpeg", ""], ["example2.jpeg", ""]], inputs=input_view)
         with gr.Column():
-            output_view = gr.Image(type="pil", label="Output Image")
+            output_view = gr.Image(type="pil", label="辨識結果")
         with gr.Column():
-            text_output = gr.Textbox(label="Results")
+            text_output = gr.Textbox(label="條碼內容")
 
     # Wire up the button
     btn.click(
         fn=process_image, 
-        inputs=input_view, 
+        inputs=[input_view, activation_key],
         outputs=[output_view, text_output]
     )