TDN-M commited on
Commit
238054e
·
verified ·
1 Parent(s): e31796c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -9,9 +9,9 @@ from PIL import Image
9
  import tempfile
10
 
11
  def convert_to_txt(file):
12
- # Kiểm tra xem file có được tải lên không
13
- if not file:
14
- return "Vui lòng tải lên một file PDF, DOC hoặc DOCX."
15
 
16
  # Lấy phần mở rộng file
17
  file_ext = Path(file.name).suffix.lower()
@@ -76,10 +76,10 @@ iface = gr.Interface(
76
  outputs=gr.File(label="Tải xuống file TXT"),
77
  title="Chuyển đổi PDF/DOC/DOCX sang TXT (Hỗ trợ OCR)",
78
  description="Tải lên file PDF, DOC hoặc DOCX để chuyển đổi nội dung thành file TXT. Hỗ trợ OCR cho PDF dạng ảnh.",
79
- flagging_dir=flagging_dir, # Chỉ định thư mục flagging
80
- allow_flagging="manual" # Hoặc đặt thành "never" để tắt flagging hoàn toàn
81
  )
82
 
83
  # Khởi chạy ứng dụng
84
  if __name__ == "__main__":
85
- iface.launch(share=True)
 
9
  import tempfile
10
 
11
  def convert_to_txt(file):
12
+ # Kiểm tra xem file có được tải lên và là file hợp lệ không
13
+ if file is None or not hasattr(file, 'name'):
14
+ return "Vui lòng tải lên một file PDF, DOC hoặc DOCX hợp lệ."
15
 
16
  # Lấy phần mở rộng file
17
  file_ext = Path(file.name).suffix.lower()
 
76
  outputs=gr.File(label="Tải xuống file TXT"),
77
  title="Chuyển đổi PDF/DOC/DOCX sang TXT (Hỗ trợ OCR)",
78
  description="Tải lên file PDF, DOC hoặc DOCX để chuyển đổi nội dung thành file TXT. Hỗ trợ OCR cho PDF dạng ảnh.",
79
+ flagging_dir=flagging_dir,
80
+ allow_flagging="manual"
81
  )
82
 
83
  # Khởi chạy ứng dụng
84
  if __name__ == "__main__":
85
+ iface.launch(share=True)