mr2along commited on
Commit
5936f93
1 Parent(s): 8a7fd4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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()