Update app.py

app.py

@@ -1,7 +1,49 @@
 import gradio as gr
+import easyocr
 
-def greet(name):
-    return "Hello " + name + "!!"
+# 创建一个 EasyOCR 阅读器实例
+reader = easyocr.Reader(['en'])  # 选择语言，这里选择英文
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def toggle_visibility(input_type):
+    """根据输入类型控制文本框和文件上传控件的可见性"""
+    user_input_visible = input_type == "Text Input"
+    file_upload_visible = input_type == "File Upload"
+    return gr.update(visible=user_input_visible), gr.update(visible=file_upload_visible)
+
+def process_input(input_type, user_input, uploaded_file):
+    print('ooooocr')
+    if input_type == "File Upload" and uploaded_file is not None:
+        # 读取上传的文件
+        with open(uploaded_file.name, "rb") as f:
+            image = f.read()
+        results = reader.readtext(image)
+        # 提取识别的文本
+        extracted_text = ' '.join([text[1] for text in results])
+        print("提取的文本:")
+        print(extracted_text)
+        return extracted_text
+    elif input_type == "Text Input":
+        return user_input
+
+demo = gr.Blocks()
+
+with demo:
+    gr.Markdown("# Step 1: Generate the keys")
+    b_gen_key_and_install = gr.Button("Generate the keys and send public part to server")
+    evaluation_key = gr.Textbox(label="Evaluation key (truncated):", max_lines=4, interactive=False)
+    user_id = gr.Textbox(label="", max_lines=4, interactive=False, visible=False)
+
+    gr.Markdown("# Step 2: Choose Input Method")
+    input_type = gr.Radio(choices=["Text Input", "File Upload"], label="Select Input Method")
+    user_input = gr.Textbox(label="Enter Text", placeholder="Type here...", visible=False)  # Initially hidden
+    file_upload = gr.File(label="Upload File", file_types=[".jpg", ".png"], visible=False)  # Initially hidden
+
+    # 使用change事件来调用toggle_visibility函数
+    input_type.change(toggle_visibility, inputs=input_type, outputs=[user_input, file_upload])
+
+    submit_button = gr.Button("Submit")
+    output_text = gr.Textbox(label="Extracted Text")
+
+    submit_button.click(process_input, inputs=[input_type, user_input, file_upload], outputs=output_text)
+
+demo.launch()