Tonic commited on
Commit
f35ea73
Β·
unverified Β·
1 Parent(s): fee0756

improve interface logic

Browse files
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -101,28 +101,20 @@ def update_image_input(task):
101
 
102
  def update_inputs(task):
103
  if task in ["Plain Text OCR", "Format Text OCR", "Multi-crop OCR", "Render Formatted OCR"]:
104
- return [gr.update(visible=False)] * 5 + [gr.update(visible=True), gr.update(visible=False)]
105
  elif task == "Fine-grained OCR (Box)":
106
  return [
107
  gr.update(visible=True, choices=["ocr", "format"]),
108
  gr.update(visible=True),
109
  gr.update(visible=False),
110
- gr.update(visible=False),
111
- gr.update(visible=False),
112
- gr.update(visible=True),
113
- gr.update(visible=False)
114
  ]
115
  elif task == "Fine-grained OCR (Color)":
116
  return [
117
  gr.update(visible=True, choices=["ocr", "format"]),
118
  gr.update(visible=False),
119
  gr.update(visible=True, choices=["red", "green", "blue"]),
120
- gr.update(visible=False),
121
- gr.update(visible=True),
122
- gr.update(visible=False),
123
- gr.update(visible=True)
124
  ]
125
-
126
  def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
127
  res, html_content, unique_id = process_image(image, task, ocr_type, ocr_box, ocr_color)
128
 
 
101
 
102
  def update_inputs(task):
103
  if task in ["Plain Text OCR", "Format Text OCR", "Multi-crop OCR", "Render Formatted OCR"]:
104
+ return [gr.update(visible=False)] * 3
105
  elif task == "Fine-grained OCR (Box)":
106
  return [
107
  gr.update(visible=True, choices=["ocr", "format"]),
108
  gr.update(visible=True),
109
  gr.update(visible=False),
 
 
 
 
110
  ]
111
  elif task == "Fine-grained OCR (Color)":
112
  return [
113
  gr.update(visible=True, choices=["ocr", "format"]),
114
  gr.update(visible=False),
115
  gr.update(visible=True, choices=["red", "green", "blue"]),
 
 
 
 
116
  ]
117
+
118
  def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
119
  res, html_content, unique_id = process_image(image, task, ocr_type, ocr_box, ocr_color)
120