extract_text_from_image

app.py

@@ -2269,6 +2269,16 @@ def download_content(content):
     word_path = create_word(content)
     return word_path
 
+# OCR
+def extract_text_from_image(image):
+    """從上傳的圖片中提取文字"""
+    return "test"
+
+    # 使用 OCR 工具
+def extract_text_from_file(file):
+    """從上傳的文件中提取文字"""    
+    return "test"
+
 
 # === INIT PARAMS ===
 def init_params(request: gr.Request):
@@ -3936,6 +3946,15 @@ with gr.Blocks(theme=THEME, css=CSS) as demo:
                 with gr.Row():
                     with gr.Column():
                         chinese_full_paragraph_input = gr.Textbox(label="輸入段落全文", lines=5)
+                        with gr.Row():
+                            with gr.Column(scale=1):
+                                # 修正文件類型設置
+                                chinese_file_upload = gr.File(
+                                    label="上傳文件", 
+                                    file_count="single"  # 確保只能上傳一個文件
+                                )
+                            with gr.Column(scale=1):
+                                chinese_image_upload = gr.Image(label="上傳圖片或拍照", type="pil", sources=["upload", "webcam", "clipboard"])
                     with gr.Column():
                         with gr.Row():
                             chinese_full_paragraph_evaluate_button = gr.Button("段落全文分析", variant="primary")
@@ -3943,6 +3962,19 @@ with gr.Blocks(theme=THEME, css=CSS) as demo:
                             chinese_full_paragraph_evaluate_output_text = gr.Markdown(label="段落全文分析")
                         with gr.Row():
                             chinese_full_paragraph_evaluate_output_table = gr.Dataframe(label="段落全文分析表格", wrap=True, column_widths=[20, 15, 65], interactive=False)
+                
+                # 處理上傳文件和圖片的事件
+                chinese_file_upload.change(  # 使用 change 而不是 upload
+                    fn=extract_text_from_file,
+                    inputs=[chinese_file_upload],
+                    outputs=[chinese_full_paragraph_input]
+                )
+                
+                chinese_image_upload.change(  # 使用 change 而不是 upload
+                    fn=extract_text_from_image,
+                    inputs=[chinese_image_upload],
+                    outputs=[chinese_full_paragraph_input]
+                )
 
                 # 修改文章
                 with gr.Row():