fix: got_ocr

app.py

@@ -75,7 +75,7 @@ def perform_ocr(selected_index, image_gallery, task, fine_grained_type, color, b
     if selected_index is None or len(image_gallery) == 0:
         return "请先选择一张图片"
 
-    selected_image = image_gallery[selected_index]
+    selected_image = image_gallery[selected_index][0]
 
     # 根据选择的任务和参数调用GOT OCR
     got_mode = task
@@ -86,19 +86,28 @@ def perform_ocr(selected_index, image_gallery, task, fine_grained_type, color, b
     return result
 
 
+def task_update(task):
+    if "fine-grained" in task:
+        return [gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)]
+    else:
+        return [gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)]
+
+
+def fine_grained_update(fine_grained_type):
+    if fine_grained_type == "color":
+        return [gr.update(visible=True), gr.update(visible=False)]
+    elif fine_grained_type == "box":
+        return [gr.update(visible=False), gr.update(visible=True)]
+    else:
+        return [gr.update(visible=False), gr.update(visible=False)]
+
+
 with gr.Blocks() as demo:
     pdf_input = gr.File(label="上传PDF文件")
     image_gallery = gr.Gallery(label="PDF页面预览", columns=3, height="auto")
     selected_index = gr.State(None)
     task_dropdown = gr.Dropdown(
-        choices=[
-            "plain texts OCR",
-            "format texts OCR",
-            "plain multi-crop OCR",
-            "format multi-crop OCR",
-            "plain fine-grained OCR",
-            "format fine-grained OCR",
-        ],
+        choices=["plain texts OCR", "format texts OCR", "plain multi-crop OCR", "format multi-crop OCR", "plain fine-grained OCR", "format fine-grained OCR"],
         label="选择GOT模式",
         value="plain texts OCR",
     )
@@ -106,43 +115,12 @@ with gr.Blocks() as demo:
     color_dropdown = gr.Dropdown(choices=["red", "green", "blue"], label="颜色列表", visible=False)
     box_input = gr.Textbox(label="输入框: [x1,y1,x2,y2]", placeholder="例如: [0,0,100,100]", visible=False)
 
-    def task_update(task):
-        if "fine-grained" in task:
-            return [
-                gr.update(visible=True),
-                gr.update(visible=False),
-                gr.update(visible=False),
-            ]
-        else:
-            return [
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False),
-            ]
-
-    def fine_grained_update(fine_grained_type):
-        if fine_grained_type == "color":
-            return [
-                gr.update(visible=True),
-                gr.update(visible=False),
-            ]
-        elif fine_grained_type == "box":
-            return [
-                gr.update(visible=False),
-                gr.update(visible=True),
-            ]
-        else:
-            return [
-                gr.update(visible=False),
-                gr.update(visible=False),
-            ]
+    ocr_button = gr.Button("开始OCR识别")
+    ocr_result = gr.Textbox(label="OCR结果")
 
     task_dropdown.change(task_update, inputs=[task_dropdown], outputs=[fine_grained_dropdown, color_dropdown, box_input])
     fine_grained_dropdown.change(fine_grained_update, inputs=[fine_grained_dropdown], outputs=[color_dropdown, box_input])
 
-    ocr_button = gr.Button("开始OCR识别")
-    ocr_result = gr.Textbox(label="OCR结果")
-
     # 更新ocr_button的click事件，传递所有必要的参数
     ocr_button.click(
         fn=perform_ocr, inputs=[selected_index, image_gallery, task_dropdown, fine_grained_dropdown, color_dropdown, box_input], outputs=ocr_result