Upload app.py

app.py

@@ -0,0 +1,121 @@
+# coding: utf-8
+# Copyright (C) 2022, [Breezedeus](https://github.com/breezedeus).
+
+from PIL import Image, ImageFilter
+import streamlit as st
+
+from cnocr import CnOcr
+from cnocr.utils import set_logger
+
+from antiocr.anti_ocr import AntiOcr
+from antiocr.consts import RESOURCE_PATH
+
+logger = set_logger()
+st.set_page_config(layout="wide")
+
+FONT_FP = '/System/Library/Fonts/PingFang.ttc'
+
+
+@st.cache(allow_output_mutation=True)
+def get_ocr_model():
+    return CnOcr()
+
+
+def main():
+    st.sidebar.header('输出设置')
+    char_reverse_ratio = st.sidebar.slider(
+        '文字倒转概率', min_value=0.0, max_value=1.0, value=0.1
+    )
+    char_to_pinyin_ratio = st.sidebar.slider(
+        '文字转拼音概率', min_value=0.0, max_value=1.0, value=0.1
+    )
+    cols = st.sidebar.columns(2)
+    min_font_size = int(cols[0].number_input('最小文字大小', 2, 80, value=15))
+    max_font_size = int(
+        cols[1].number_input(
+            '最大文字大小', min_font_size + 1, 120, value=max(40, min_font_size + 1)
+        )
+    )
+    text_color = st.sidebar.color_picker('文字颜色', value='#5087DC')
+
+    st.sidebar.markdown('----')
+    use_random_bg = st.sidebar.checkbox('随机生成背景图片')
+    if use_random_bg:
+        bg_text_density = st.sidebar.slider(
+            '背景图片文字密度', min_value=0.0, max_value=3.0, value=1.0
+        )
+        cols = st.sidebar.columns(2)
+        bg_min_size = int(
+            cols[0].number_input('背景图片最小文字', 2, 80, key='bg_min', value=15)
+        )
+        bg_max_size = int(
+            cols[1].number_input(
+                '背景图片最大文字',
+                bg_min_size + 1,
+                120,
+                key='bg_max',
+                value=max(70, bg_min_size + 1),
+            )
+        )
+        bg_text_color = st.sidebar.color_picker('背景图片文字颜色', value='#07BCE0')
+        bg_gen_config = dict(
+            text_density=bg_text_density,
+            text_color=bg_text_color,
+            min_font_size=bg_min_size,
+            max_font_size=bg_max_size,
+        )
+        bg_image = None
+    else:
+        bg_gen_config = None
+        bg_image = Image.open(RESOURCE_PATH / 'bg.jpeg')
+        bg_image = bg_image.filter(ImageFilter.MaxFilter(3))
+
+    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)
+    st.markdown('')
+    st.markdown('')
+    desc = '<strong>antiOCR</strong> 对指定的文字（来自输入或者图片）进行处理，输出图片，此图片无法通过OCR技术识别出有意义的文字。'
+    st.markdown(f"<div style='text-align: left;'>{desc}</div>", unsafe_allow_html=True)
+    st.markdown('')
+    st.subheader('选择待转换文字图片，或者直接输入待转换文字')
+    default_texts = '真的猛士，敢于直面惨淡的人生，敢于正视淋漓的鲜血。这是怎样的哀痛者和幸福者？然而造化又常常为庸人设计，以时间的流逝，来洗涤旧迹，仅是留下淡红的血色和微漠的悲哀。在这淡红的血色和微漠的悲哀中，又给人暂得偷生，维持着这似人非人的世界。 ——鲁迅'
+    content_file = st.file_uploader('待转换文字图片', type=["png", "jpg", "jpeg", "webp"])
+    ocr = get_ocr_model()
+    anti = AntiOcr()
+    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])
+        except Exception as e:
+            st.error(e)
+
+    texts = st.text_area('待转换文字图片', value=texts or default_texts, height=120)
+
+    if st.button("生成图片"):
+        if texts:
+            with st.spinner('图片生成中…'):
+                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,
+                )
+            st.subheader('输出图片')
+            st.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))
+
+
+if __name__ == '__main__':
+    main()