Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -351,28 +351,29 @@ Font_Style: {metadata['font_style']}
|
|
| 351 |
self.log_operation(error_msg, "error")
|
| 352 |
logger.error(traceback.format_exc())
|
| 353 |
|
| 354 |
-
|
| 355 |
-
|
| 356 |
-
|
| 357 |
-
|
| 358 |
-
'current': None,
|
| 359 |
-
'next': None,
|
| 360 |
-
'progress': "No text loaded"
|
| 361 |
-
}
|
| 362 |
-
|
| 363 |
-
current = self.get_styled_text(self.sentences[self.current_index])
|
| 364 |
-
next_text = None
|
| 365 |
-
|
| 366 |
-
if self.current_index < len(self.sentences) - 1:
|
| 367 |
-
next_text = self.get_styled_text(self.sentences[self.current_index + 1])
|
| 368 |
-
|
| 369 |
-
progress = f"Sentence {self.current_index + 1} of {len(self.sentences)}"
|
| 370 |
-
|
| 371 |
return {
|
| 372 |
-
'current':
|
| 373 |
-
'next':
|
| 374 |
-
'progress':
|
| 375 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 376 |
|
| 377 |
def navigate(self, direction: str) -> Dict[str, Optional[str]]:
|
| 378 |
"""Navigate through sentences"""
|
|
@@ -407,6 +408,8 @@ Font_Style: {metadata['font_style']}
|
|
| 407 |
logger.error(f"Error reading dataset statistics: {str(e)}")
|
| 408 |
return {}
|
| 409 |
|
|
|
|
|
|
|
| 410 |
def create_interface():
|
| 411 |
"""Create Gradio interface with enhanced features"""
|
| 412 |
|
|
@@ -441,7 +444,6 @@ def create_interface():
|
|
| 441 |
|
| 442 |
collector = TTSDatasetCollector()
|
| 443 |
|
| 444 |
-
|
| 445 |
with gr.Blocks(title="TTS Dataset Collection Tool", css=custom_css) as interface:
|
| 446 |
gr.Markdown("# TTS Dataset Collection Tool")
|
| 447 |
|
|
|
|
| 351 |
self.log_operation(error_msg, "error")
|
| 352 |
logger.error(traceback.format_exc())
|
| 353 |
|
| 354 |
+
# Add these methods to the TTSDatasetCollector class
|
| 355 |
+
def get_navigation_info(self) -> Dict[str, Optional[str]]:
|
| 356 |
+
"""Get current and next sentence information"""
|
| 357 |
+
if not self.sentences:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 358 |
return {
|
| 359 |
+
'current': None,
|
| 360 |
+
'next': None,
|
| 361 |
+
'progress': "No text loaded"
|
| 362 |
}
|
| 363 |
+
|
| 364 |
+
current = self.get_styled_text(self.sentences[self.current_index])
|
| 365 |
+
next_text = None
|
| 366 |
+
|
| 367 |
+
if self.current_index < len(self.sentences) - 1:
|
| 368 |
+
next_text = self.get_styled_text(self.sentences[self.current_index + 1])
|
| 369 |
+
|
| 370 |
+
progress = f"Sentence {self.current_index + 1} of {len(self.sentences)}"
|
| 371 |
+
|
| 372 |
+
return {
|
| 373 |
+
'current': current,
|
| 374 |
+
'next': next_text,
|
| 375 |
+
'progress': progress
|
| 376 |
+
}
|
| 377 |
|
| 378 |
def navigate(self, direction: str) -> Dict[str, Optional[str]]:
|
| 379 |
"""Navigate through sentences"""
|
|
|
|
| 408 |
logger.error(f"Error reading dataset statistics: {str(e)}")
|
| 409 |
return {}
|
| 410 |
|
| 411 |
+
|
| 412 |
+
# Then create the interface function outside the class
|
| 413 |
def create_interface():
|
| 414 |
"""Create Gradio interface with enhanced features"""
|
| 415 |
|
|
|
|
| 444 |
|
| 445 |
collector = TTSDatasetCollector()
|
| 446 |
|
|
|
|
| 447 |
with gr.Blocks(title="TTS Dataset Collection Tool", css=custom_css) as interface:
|
| 448 |
gr.Markdown("# TTS Dataset Collection Tool")
|
| 449 |
|