Update app.py
Browse files
app.py
CHANGED
@@ -86,10 +86,10 @@ def get_all_chapters_content(story_url, start_chapter, max_chapters):
|
|
86 |
f.write("-" * 50 + "\n")
|
87 |
|
88 |
# Tạo file EPUB từ nội dung đã lưu
|
89 |
-
create_epub_from_chapters(chapter_contents, story_name)
|
90 |
|
91 |
# Trả về kết quả
|
92 |
-
return f"Đã tải thành công {max_chapters} chương. Tổng thời gian: {total_time:.2f} giây, Thời gian trung bình: {avg_time_per_chapter:.2f} giây. File TXT: {output_file}"
|
93 |
|
94 |
# Hàm để tạo file EPUB từ nội dung các chương
|
95 |
def create_epub_from_chapters(chapter_contents, story_name):
|
@@ -105,12 +105,12 @@ def create_epub_from_chapters(chapter_contents, story_name):
|
|
105 |
|
106 |
# Lưu file EPUB
|
107 |
output_directory = f"./{story_name}.epub"
|
108 |
-
my_epub.create(output_directory) # Lưu file EPUB
|
109 |
print(f"Đã tạo file EPUB: {output_directory}")
|
110 |
|
111 |
except Exception as e:
|
112 |
print(f"Lỗi khi tạo file EPUB: {e}")
|
113 |
-
|
114 |
# Giao diện Gradio
|
115 |
def gradio_interface(story_url, start_chapter, max_chapters):
|
116 |
# Bắt đầu đo thời gian cho toàn bộ quá trình
|
@@ -135,7 +135,7 @@ gr.Interface(
|
|
135 |
gr.Textbox(label="Số chương bắt đầu", placeholder="Nhập số chương bắt đầu"),
|
136 |
gr.Textbox(label="Số chương muốn tải", placeholder="Nhập số chương muốn tải")
|
137 |
],
|
138 |
-
outputs="text",
|
139 |
title="Truyện Full Downloader",
|
140 |
description="Công cụ tải truyện từ truyenfull.tv và tạo file EPUB."
|
141 |
).launch()
|
|
|
86 |
f.write("-" * 50 + "\n")
|
87 |
|
88 |
# Tạo file EPUB từ nội dung đã lưu
|
89 |
+
epubfile=create_epub_from_chapters(chapter_contents, story_name)
|
90 |
|
91 |
# Trả về kết quả
|
92 |
+
return [f"Đã tải thành công {max_chapters} chương. Tổng thời gian: {total_time:.2f} giây, Thời gian trung bình: {avg_time_per_chapter:.2f} giây. File TXT: {output_file}",epubfile]
|
93 |
|
94 |
# Hàm để tạo file EPUB từ nội dung các chương
|
95 |
def create_epub_from_chapters(chapter_contents, story_name):
|
|
|
105 |
|
106 |
# Lưu file EPUB
|
107 |
output_directory = f"./{story_name}.epub"
|
108 |
+
epubfile=my_epub.create(output_directory) # Lưu file EPUB
|
109 |
print(f"Đã tạo file EPUB: {output_directory}")
|
110 |
|
111 |
except Exception as e:
|
112 |
print(f"Lỗi khi tạo file EPUB: {e}")
|
113 |
+
return epubfile
|
114 |
# Giao diện Gradio
|
115 |
def gradio_interface(story_url, start_chapter, max_chapters):
|
116 |
# Bắt đầu đo thời gian cho toàn bộ quá trình
|
|
|
135 |
gr.Textbox(label="Số chương bắt đầu", placeholder="Nhập số chương bắt đầu"),
|
136 |
gr.Textbox(label="Số chương muốn tải", placeholder="Nhập số chương muốn tải")
|
137 |
],
|
138 |
+
outputs=["text","file"],
|
139 |
title="Truyện Full Downloader",
|
140 |
description="Công cụ tải truyện từ truyenfull.tv và tạo file EPUB."
|
141 |
).launch()
|