TDN-M commited on
Commit
9f66a8c
·
verified ·
1 Parent(s): ed006b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -11
app.py CHANGED
@@ -69,17 +69,20 @@ def convert_to_txt(file):
69
  temp_dir = tempfile.gettempdir()
70
  flagging_dir = os.path.join(temp_dir, "flagged")
71
 
72
- # Tạo giao diện Gradio với flagging_dir được chỉ định
73
- iface = gr.Interface(
74
- fn=convert_to_txt,
75
- inputs=gr.File(label="Tải lên file PDF, DOC hoặc DOCX"),
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)
 
69
  temp_dir = tempfile.gettempdir()
70
  flagging_dir = os.path.join(temp_dir, "flagged")
71
 
72
+ # Tạo giao diện Gradio với cấu hình đơn giản
73
+ with gr.Blocks() as iface:
74
+ gr.Markdown("# Chuyển đổi PDF/DOC/DOCX sang TXT (Hỗ trợ OCR)")
75
+ gr.Markdown("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.")
76
+ file_input = gr.File(label="Tải lên file PDF, DOC hoặc DOCX", file_types=[".pdf", ".doc", ".docx"])
77
+ file_output = gr.File(label="Tải xuống file TXT")
78
+ submit_button = gr.Button("Chuyển đổi")
79
+ submit_button.click(
80
+ fn=convert_to_txt,
81
+ inputs=file_input,
82
+ outputs=file_output,
83
+ api_name="convert_to_txt"
84
+ )
85
 
86
  # Khởi chạy ứng dụng
87
  if __name__ == "__main__":
88
+ iface.launch()