Update app.py

app.py

@@ -152,7 +152,145 @@ class TTSDatasetCollector:
             logger.error(traceback.format_exc())
             return False, error_msg
     
-    # Remaining methods go here ...
+        def get_styled_text(self, text: str) -> str:
+        """Get text with current font styling"""
+        font_css = FONT_STYLES[self.current_font]['css']
+        return f"<div style='{font_css}'>{text}</div>"
+    
+    def generate_filenames(self, dataset_name: str, speaker_id: str) -> Tuple[str, str]:
+        """Generate unique filenames for audio and text files"""
+        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+        sentence_id = f"{self.current_index+1:04d}"
+        base_name = f"{dataset_name}_{speaker_id}_{sentence_id}_{timestamp}"
+        return f"{base_name}.wav", f"{base_name}.txt"
+    
+    def save_recording(self, audio_file, speaker_id: str, dataset_name: str) -> Tuple[bool, str]:
+        """Save recording with enhanced error handling and logging"""
+        if not all([audio_file, speaker_id, dataset_name]):
+            missing = []
+            if not audio_file: missing.append("audio recording")
+            if not speaker_id: missing.append("speaker ID")
+            if not dataset_name: missing.append("dataset name")
+            return False, f"Missing required information: {', '.join(missing)}"
+        
+        try:
+            # Validate inputs
+            if not speaker_id.strip().isalnum():
+                return False, "Speaker ID must contain only letters and numbers"
+            
+            if not dataset_name.strip().isalnum():
+                return False, "Dataset name must contain only letters and numbers"
+            
+            # Generate filenames
+            audio_name, text_name = self.generate_filenames(dataset_name, speaker_id)
+            
+            # Create speaker directories
+            audio_dir = self.root_path / 'audio' / speaker_id
+            text_dir = self.root_path / 'transcriptions' / speaker_id
+            audio_dir.mkdir(exist_ok=True)
+            text_dir.mkdir(exist_ok=True)
+            
+            # Save audio file
+            audio_path = audio_dir / audio_name
+            shutil.copy2(audio_file, audio_path)
+            
+            # Save transcription
+            text_path = text_dir / text_name
+            self.save_transcription(
+                text_path,
+                self.sentences[self.current_index],
+                {
+                    'speaker_id': speaker_id,
+                    'dataset_name': dataset_name,
+                    'timestamp': datetime.now().isoformat(),
+                    'audio_file': audio_name,
+                    'font_style': self.current_font
+                }
+            )
+            
+            # Update metadata
+            self.update_metadata(speaker_id, dataset_name)
+            
+            # Log success
+            self.log_operation(
+                f"Saved recording: Speaker={speaker_id}, Dataset={dataset_name}, "
+                f"Audio={audio_name}, Text={text_name}"
+            )
+            
+            return True, f"Recording saved successfully as {audio_name}"
+            
+        except Exception as e:
+            error_msg = f"Error saving recording: {str(e)}"
+            self.log_operation(error_msg, "error")
+            logger.error(traceback.format_exc())
+            return False, error_msg
+    
+    def save_transcription(self, file_path: Path, text: str, metadata: Dict) -> None:
+        """Save transcription with metadata"""
+        content = f"""[METADATA]
+Recording_ID: {metadata['audio_file']}
+Speaker_ID: {metadata['speaker_id']}
+Dataset_Name: {metadata['dataset_name']}
+Timestamp: {metadata['timestamp']}
+Font_Style: {metadata['font_style']}
+
+[TEXT]
+{text}
+"""
+        with open(file_path, 'w', encoding='utf-8') as f:
+            f.write(content)
+    
+    def update_metadata(self, speaker_id: str, dataset_name: str) -> None:
+        """Update dataset metadata with error handling"""
+        metadata_file = self.root_path / 'metadata' / 'dataset_info.json'
+        
+        try:
+            if metadata_file.exists():
+                with open(metadata_file, 'r') as f:
+                    metadata = json.load(f)
+            else:
+                metadata = {'speakers': {}, 'last_updated': None}
+            
+            # Update speaker data
+            if speaker_id not in metadata['speakers']:
+                metadata['speakers'][speaker_id] = {
+                    'total_recordings': 0,
+                    'datasets': {}
+                }
+            
+            if dataset_name not in metadata['speakers'][speaker_id]['datasets']:
+                metadata['speakers'][speaker_id]['datasets'][dataset_name] = {
+                    'recordings': 0,
+                    'sentences': len(self.sentences),
+                    'first_recording': datetime.now().isoformat(),
+                    'last_recording': None,
+                    'font_styles_used': []
+                }
+            
+            # Update counts and timestamps
+            metadata['speakers'][speaker_id]['total_recordings'] += 1
+            metadata['speakers'][speaker_id]['datasets'][dataset_name]['recordings'] += 1
+            metadata['speakers'][speaker_id]['datasets'][dataset_name]['last_recording'] = \
+                datetime.now().isoformat()
+            
+            # Update font styles
+            if self.current_font not in metadata['speakers'][speaker_id]['datasets'][dataset_name]['font_styles_used']:
+                metadata['speakers'][speaker_id]['datasets'][dataset_name]['font_styles_used'].append(
+                    self.current_font
+                )
+            
+            metadata['last_updated'] = datetime.now().isoformat()
+            
+            # Save updated metadata
+            with open(metadata_file, 'w') as f:
+                json.dump(metadata, f, indent=2)
+            
+            self.log_operation(f"Updated metadata for {speaker_id} in {dataset_name}")
+            
+        except Exception as e:
+            error_msg = f"Error updating metadata: {str(e)}"
+            self.log_operation(error_msg, "error")
+            logger.error(traceback.format_exc())
 
 def create_interface():
     """Create Gradio interface with enhanced features"""
@@ -288,11 +426,9 @@ if __name__ == "__main__":
         interface.launch(
             server_name="0.0.0.0",
             server_port=7860,
-            share=True,
-            enable_queue=True
+            share=True
         )
     except Exception as e:
         logger.error(f"Failed to launch interface: {str(e)}")
         logger.error(traceback.format_exc())
         raise
-