update

app.py

@@ -36,17 +36,14 @@ def get_ocr_model():
     return CnOcr()
 
 
-def save_image(img):
+def download_image_button(img):
     from io import BytesIO
 
     buf = BytesIO()
     img.save(buf, format="JPEG")
     byte_im = buf.getvalue()
     st.download_button(
-        label="下载图片",
-        data=byte_im,
-        file_name="antiOCR.jpeg",
-        mime="image/jpeg",
+        label="下载图片", data=byte_im, file_name="antiOCR.jpeg", mime="image/jpeg",
     )
 
 
@@ -108,9 +105,13 @@ def main():
 
     title = '让文字自由传播：<a href="https://github.com/breezedeus/antiOCR">antiOCR</a>'
     st.markdown(f"<h1 style='text-align: center;'>{title}</h1>", unsafe_allow_html=True)
-    subtitle = '作者：<a href="https://github.com/breezedeus">breezedeus</a>； ' \
-               '欢迎加入 <a href="https://cnocr.readthedocs.io/zh/latest/contact/">交流群</a>'
-    st.markdown(f"<div style='text-align: center;'>{subtitle}</div>", unsafe_allow_html=True)
+    subtitle = (
+        '作者：<a href="https://github.com/breezedeus">breezedeus</a>； '
+        '欢迎加入 <a href="https://cnocr.readthedocs.io/zh/latest/contact/">交流群</a>'
+    )
+    st.markdown(
+        f"<div style='text-align: center;'>{subtitle}</div>", unsafe_allow_html=True
+    )
     st.markdown('')
     st.markdown('')
     desc = '<strong>antiOCR</strong> 对指定的文字（来自输入或者图片）进行处理，输出图片，此图片无法通过OCR技术识别出有意义的文字。'
@@ -121,40 +122,45 @@ def main():
     content_file = st.file_uploader('输入待转换的文字图片：', type=["png", "jpg", "jpeg", "webp"])
     ocr = get_ocr_model()
     anti = AntiOcr()
-    texts = None
+    ocr_texts = None
     if content_file is not None:
         try:
             img = Image.open(content_file).convert('RGB')
             ocr_out = ocr.ocr(img)
-            texts = '\n'.join([out['text'] for out in ocr_out])
+            ocr_texts = '\n'.join([out['text'] for out in ocr_out])
         except Exception as e:
             st.error(e)
+        if not ocr_texts:
+            st.warning(f'抱歉，图片中未识别出任何文字。')
 
-    texts = st.text_area('或者，直接输入待转换的文字：', value=texts or default_texts, height=120)
-
-    if st.button("生成图片"):
-        if texts:
-            with st.spinner('图片生成中…'):
-                logger.info('\ngenerating an image for texts:\n %s', texts)
-                out_img = anti(
-                    texts,
-                    char_reverse_ratio=char_reverse_ratio,
-                    char_to_pinyin_ratio=char_to_pinyin_ratio,
-                    text_color=text_color,
-                    min_font_size=min_font_size,
-                    max_font_size=max_font_size,
-                    bg_image=bg_image,
-                    bg_gen_config=bg_gen_config,
-                    font_fp=font_fp,
-                )
-            st.subheader('输出图片')
-            st.image(out_img)
-            save_image(out_img)
-
-            st.markdown('**对输出图片进行OCR，结果如下（如果依旧出现敏感词，尝试重新生成图片）：**')
-            ocr_out = ocr.ocr(out_img)
-            new_texts = [out['text'] for out in ocr_out]
-            st.text('\n'.join(new_texts))
+    texts = st.text_area(
+        '或者，直接输入待转换的文字：', value=ocr_texts or '', height=120, placeholder=default_texts
+    )
+    texts = texts or default_texts
+
+    enter = st.button("生成图片")
+    if (content_file is not None and ocr_texts) or enter:
+        with st.spinner('图片生成中…'):
+            logger.info('\ngenerating an image for texts:\n %s', texts)
+            out_img = anti(
+                texts,
+                char_reverse_ratio=char_reverse_ratio,
+                char_to_pinyin_ratio=char_to_pinyin_ratio,
+                text_color=text_color,
+                min_font_size=min_font_size,
+                max_font_size=max_font_size,
+                bg_image=bg_image,
+                bg_gen_config=bg_gen_config,
+                font_fp=font_fp,
+            )
+        st.subheader('输出图片')
+        st.image(out_img)
+        download_image_button(out_img)
+
+        st.markdown('**对输出图片进行OCR，结果如下（如果依旧出现敏感词，尝试重新生成图片）：**')
+        ocr_out = ocr.ocr(out_img)
+        new_texts = [out['text'] for out in ocr_out]
+        st.text('\n'.join(new_texts))
 
 
 if __name__ == '__main__':