diff --git a/assets/test_case.zip b/assets/test_case.zip new file mode 100644 index 0000000..c765e54 Binary files /dev/null and b/assets/test_case.zip differ diff --git a/utils/actions/fundus_diagnosis.py b/utils/actions/fundus_diagnosis.py index e5d5f1d..6790ab5 100644 --- a/utils/actions/fundus_diagnosis.py +++ b/utils/actions/fundus_diagnosis.py @@ -47,7 +47,13 @@ def __call__(self, query: str) -> ActionReturn: logger.info("query: " + query) if query.startswith("{"): query = query.replace("'", "\"") # 为了解决如下错误:{'image_path':'static/lwh017-20180821-OD-1.jpg'} - query = json.loads(query) + try: + query = json.loads(query) + except: + t = ActionReturn(url=None, args=None, type=self.name, ) + t.result = "输入参数格式参数,输入需要为str:image_path" + t.state = ActionStatusCode.API_ERROR + return t if not (isinstance(query, dict) and ("image_path" in query or "value" in query)): response = "输入参数错误,请确定是否需要调用该工具" tool_return = ActionReturn(url=None, args=None, type=self.name) diff --git a/web_demo.py b/web_demo.py index 525e49b..9ad9d2d 100644 --- a/web_demo.py +++ b/web_demo.py @@ -117,6 +117,7 @@ def setup_sidebar(self): st.session_state.file_upload_key = 0 uploaded_file = st.sidebar.file_uploader( '眼底图文件', type=['png', 'jpg', 'jpeg'], key=st.session_state.file_upload_key) + st.sidebar.download_button(label="下载眼底图测试用例", data=open("assets/test_case.zip", "rb").read(), file_name="test_case.zip", mime="application/zip") return model_name, model, plugin_action, uploaded_file def init_model(self, option):