Spaces:
Build error
Build error
| import hashlib, shutil, uuid, os | |
| from data.dataset_manager import load_hf_dataset, export_full_dataset, upload_admin_audio, export_admin_audio_annotations | |
| from config.settings import * | |
| from config.app_state import load_state, save_state, ui_state_for_mode | |
| import gradio as gr | |
| UPLOADS_DIR = "uploads" | |
| os.makedirs(UPLOADS_DIR, exist_ok = True) | |
| def hash_password(password: str) -> str: | |
| return hashlib.sha256(password.encode()).hexdigest() | |
| def admin_login(input_password): | |
| if hash_password(input_password) == ADMIN_PASSWORD_HASH: | |
| #if input_password == ADMIN_PASSWORD_HASH: | |
| #ds = load_hf_dataset() | |
| #df = ds.to_pandas() | |
| # For download: write temp CSV | |
| #temp_csv = "annotations_export.csv" | |
| #df.to_csv(temp_csv, index=False) | |
| state = load_state() | |
| mode_label = f"Mode actuel : {'Audio fichier' if state['mode'] == 'audio_file' else 'TTS'}" | |
| if state["mode"] == "audio_file": | |
| mode_label += f" - {state.get('audio_name')} (id: {state.get('audio_id')})" | |
| return ( | |
| gr.update(visible=True), | |
| #df, | |
| #temp_csv, | |
| gr.update(value="Admin authentication successful."), | |
| gr.update(value = mode_label) | |
| ) | |
| else: | |
| return ( | |
| gr.update(visible=False), | |
| #None, | |
| #None, | |
| gr.update(value="Admin authentication failed."), | |
| gr.update(value = "") | |
| ) | |
| def set_audio_file_mode(audio_files): | |
| """Reçoit une liste de chemins de fichiers""" | |
| if not audio_files: | |
| return (gr.update(value = 'Aucun dossier sélectionné.'),) + ui_state_for_mode("tts") | |
| entries = [] | |
| for path in audio_files: | |
| audio_id = str(uuid.uuid4())[:8] | |
| audio_name = os.path.splitext(os.path.basename(path))[0] | |
| dest_path = os.path.abspath(os.path.join(UPLOADS_DIR, f"{audio_id}_{audio_name}.wav")) | |
| shutil.copy(path, dest_path) | |
| #Upload sur HF | |
| upload_admin_audio(dest_path, f"{audio_name}.wav", audio_id) | |
| entries.append({ | |
| "audio_id": audio_id, | |
| "audio_name": audio_name, | |
| "audio_path": dest_path, | |
| "audio_folder": f"annotations_{audio_name}_{audio_id}", | |
| "count": 0 | |
| }) | |
| state = { | |
| "mode": "audio_file", | |
| "audio_files": entries | |
| } | |
| save_state(state) | |
| return ( | |
| gr.update(value = f"Mode basculé : {len(entries)} audios chargés pour annotation"), | |
| ) + ui_state_for_mode("audio_file") | |
| def switch_to_tts_mode(): | |
| state = load_state() | |
| state["mode"] = "tts" | |
| save_state(state) | |
| return (gr.update(value = "Mode basculé : TTS (Text-to-Speech)"),) + ui_state_for_mode("tts") | |
| def _audio_file_ui_state(): | |
| hidden = gr.update(visible=False) | |
| return ( | |
| hidden, hidden, | |
| hidden, | |
| hidden, | |
| hidden, | |
| hidden, | |
| hidden, | |
| gr.update(value = "Display Audio") | |
| ) | |
| def _tts_ui_state(): | |
| visible = gr.update(visible=True) | |
| return ( | |
| visible, visible, | |
| visible, | |
| visible, | |
| visible, | |
| visible, | |
| visible, | |
| gr.update(value = "Generate Audio") | |
| ) | |
| def export_dataset(): | |
| try: | |
| zip_path = export_full_dataset() | |
| return zip_path | |
| except Exception as e: | |
| print(f"Export error: {e}") | |
| return None | |
| def export_admin_audio_dataset(): | |
| try: | |
| return export_admin_audio_annotations() | |
| except Exception as e: | |
| print(f"Export admin audio annotations error: {e}") | |
| return None | |