biosn2 commited on
Commit
1071433
·
verified ·
1 Parent(s): 5750788

Upload indextts/utils/webui_utils.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. indextts/utils/webui_utils.py +42 -0
indextts/utils/webui_utils.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+
4
+ def html_center(text, label='p'):
5
+ return f"""<div style="text-align: center; margin: 100; padding: 50;">
6
+ <{label} style="margin: 0; padding: 0;">{text}</{label}>
7
+ </div>"""
8
+
9
+
10
+ def html_left(text, label='p'):
11
+ return f"""<div style="text-align: left; margin: 0; padding: 0;">
12
+ <{label} style="margin: 0; padding: 0;">{text}</{label}>
13
+ </div>"""
14
+
15
+
16
+ def next_page(page_number,sentences):
17
+ new_page_number = int(page_number) + 1
18
+ update_page_number = gr.update(value=str(new_page_number))
19
+ update_prev_page = gr.update(visible=True, interactive=True)
20
+ if len(sentences.values) <= new_page_number * 20:
21
+ update_next_page = gr.update(visible=False, interactive=False)
22
+ else:
23
+ update_next_page = gr.update(visible=True, interactive=True)
24
+ return update_page_number, update_next_page, update_prev_page
25
+
26
+
27
+ def prev_page(page_number):
28
+ new_page_number = int(page_number) - 1
29
+ update_page_number = gr.update(value=str(new_page_number))
30
+ if new_page_number == 1:
31
+ update_prev_page = gr.update(visible=False, interactive=False)
32
+ else:
33
+ update_prev_page = gr.update(visible=True, interactive=True)
34
+ update_next_page = gr.update(visible=True, interactive=True)
35
+ return update_page_number, update_next_page, update_prev_page
36
+
37
+
38
+ def update_current_texts(page_number,sentences):
39
+ start_index = (int(page_number) - 1) * 20
40
+ end_index = int(page_number) * 20
41
+ current_texts = sentences.values[start_index:end_index if end_index < len(sentences.values) else len(sentences.values)]
42
+ return gr.update(values=current_texts)