Upload 2 files

add error message & example0

app.py

@@ -4,6 +4,7 @@ import os
 from ultralytics import YOLO
 
 from google.cloud import vision
+
 _api_key = os.environ["API_KEY"]
 _project_id = os.environ["PROJECT_ID"]
 client = vision.ImageAnnotatorClient(client_options={"quota_project_id": _project_id, "api_key": _api_key})
@@ -29,7 +30,7 @@ keypointModel = r'keypoints-best.pt'
 minSz = 1280
 
 
-_examples = [["example1.jpg",True], ["example2.jpg",False], ["example3.jpg",True]]
+_examples = [["example0.jpg", True],["example1.jpg",True], ["example2.jpg",False], ["example3.jpg",True]]
 
 
 def unwarp_image(warped_image, src_points, dst_points, output_width, output_height):
@@ -64,7 +65,7 @@ def get_corners(results:list, img):
     planars = []
     kps = []
     for kpco in r.keypoints.xy.cpu():#.squeeze()
-        assert len(kpco)>0, "not found"
+        # assert len(kpco)>0, "not found"
         keypoints = {k:v.numpy() for v,k in zip(kpco,KP)}
         sz = model1DIM
         dstCorners = np.array([(0,0),(sz,0),(sz,sz),(0,sz)])
@@ -102,14 +103,17 @@ def preprocessImg(planar):
 
 
 def get_keypoints(results:list):
-    assert len(results) ==1, "found multiple dials. expected only 1"
+    if len(results) !=1: 
+        raise gr.Error("found multiple dials. expected only 1")
     r = results[0]
     # ordering
     kp = "start_kp center end_kp tip".split()
     kpco = r.keypoints.xy.cpu().squeeze()
     keypoints = {k:v.numpy() for v,k in zip(kpco,kp)}
-    assert len(keypoints["center"])==2, "center keypoint not found"
-    assert len(keypoints["tip"])==2, "tip keypoint not found"
+    if len(keypoints["center"])!=2:
+        raise gr.Error("center keypoint not found")
+    elif len(keypoints["tip"])!=2:
+        raise gr.Error("tip keypoint not found")
     return keypoints
 
 def cosangle(a,b, ignoreRot=False):
@@ -188,7 +192,7 @@ def result_as_validvalue(contents:list[dict])->tuple[list[dict], list[str]]:
         valid.append({"text":f["text"], "value": value, "mid": m, "apchar":a, "box":b})
 
     valid.sort(key=lambda e: e["value"])
-    return valid, other
+    return valid, list(set(other))
 
 
 distance = lambda a,b : np.sqrt(np.square(np.array(a)-np.array(b)).sum())
@@ -344,10 +348,11 @@ def get_needle_value(img, keypoints):
     contents = get_text_from_image(client, img)
     toc = time()
     print(f"ocr took: {toc-tic2:.1g}")
-
-    assert len(contents)
+    if 0==len(contents):
+        raise gr.Error("failed to get any text/number")
     valid,other = result_as_validvalue(contents)
-    assert len(valid)
+    if 0==len(valid):
+        raise gr.Error("failed to get any number")
 
     valid.append({"text":"tip", "mid":keypoints["tip"]})
     ix,an = sort_clockwise_with_start([e["mid"] for e in valid],*keypoints["center"], 0)
@@ -376,7 +381,8 @@ def get_needle_value(img, keypoints):
 
     center = np.array(keypoints["center"])
     values, rate = determine_ocr_neighbors(keypoints, valid, nearestIx)
-    assert len(values)>=2, "failed to find at least 2 OCR values"
+    if len(values)<2:
+        raise gr.Error("failed to find at least 2 OCR number values")
 
     # import pandas as pd
     # print(pd.DataFrame.from_dict(values))
@@ -484,13 +490,15 @@ def predict(img, detect_gauge_first):
 def test(img, detect_gauge_first):
     return {"msg":str(img.size), "other": detect_gauge_first}
 
+
 description = r"""
 <b>Official 🤗 Gradio demo</b> for <a href='https://synanthropic.com/reading-analog-gauge' target='_blank'><b>Reading Analog Gauges: Automate Gauge Readings with AI in Days, Not Months
 </b></a>.<br>
 <br>
 This model reads analog dial gauge by detecting, applying perspective correction, and gauge reading. 
 <br>
-The model was build <i><strong>only</strong></i> with synthetic data.<br>
+The model was build <i><strong>only</strong></i> with synthetic data (e.g. examples).<br>
+Hence, it <i>probably</i> will not work on significantly different images - give it a try. Let us know, so we can keep improving.<br>
 <br>
 You can read more about it [here](https://synanthropic.com/reading-analog-gauge).
 <br>
@@ -511,5 +519,6 @@ gr.Interface(title="Reading Analog Gauges",
                  ],
              outputs="json",
              examples=_examples,
+             allow_flagging="never",
              cache_examples=True)\
     .launch()