Add application file

app.py

@@ -9,13 +9,13 @@ from pathlib import Path
 DATASET_REPO_ID = "MeysamSh/ENSIMSoundDataCollection"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# Admin Credentials (Keep these secure!)
+# Admin Credentials
 ADMIN_USERNAME = "admin"
 ADMIN_PASSWORD = "30c8663d3ca10ededd17ac1b55f3d533ab29cf1b8470b1729af09afda3f0a516" 
 
 AUTHORIZED_USERS = [
     "5e884898da28047151d0e56f8dc6292773603d0d6aabbdd62a11ef721d1542d8", 
-    "test@univ-lemans.fr"
+    "test"
 ]
 
 api = HfApi()
@@ -33,8 +33,8 @@ def verify_user(email):
 def upload_data(email, label, audio_path):
     if not audio_path: return "⚠️ No audio recorded."
     try:
-        # email_hash = hashlib.sha256(email.strip().lower().encode()).hexdigest()
-        email_index = AUTHORIZED_USERS.index(email.strip().lower()) if email.strip().lower() in AUTHORIZED_USERS else -1
+        clean_email = email.strip().lower()
+        email_index = AUTHORIZED_USERS.index(clean_email) if clean_email in AUTHORIZED_USERS else "unknown"
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         unique_filename = f"{email_index}_{timestamp}.wav"
         
@@ -42,7 +42,7 @@ def upload_data(email, label, audio_path):
         api.upload_file(path_or_fileobj=audio_path, path_in_repo=f"data/{unique_filename}",
                         repo_id=DATASET_REPO_ID, repo_type="dataset", token=HF_TOKEN)
         # Upload Metadata
-        meta_content = f"user_id,label,file_name,timestamp\n{email.strip().lower()},{label},{unique_filename},{timestamp}"
+        meta_content = f"user_id,label,file_name,timestamp\n{clean_email},{label},{unique_filename},{timestamp}"
         api.upload_file(path_or_fileobj=meta_content.encode(), path_in_repo=f"metadata/meta_{email_index}_{timestamp}.csv",
                         repo_id=DATASET_REPO_ID, repo_type="dataset", token=HF_TOKEN)
         return f"🎉 Success! Audio saved as {label}."
@@ -50,30 +50,42 @@ def upload_data(email, label, audio_path):
 
 # --- ADMIN LOGIC ---
 
+def get_stats():
+    """Helper to calculate stats from repository files"""
+    try:
+        files = api.list_repo_files(repo_id=DATASET_REPO_ID, repo_type="dataset")
+        audio_files = [f for f in files if f.startswith("data/") and f.endswith(".wav")]
+        
+        # Extract user indices from filenames like 'data/0_20260511.wav'
+        user_indices = set()
+        for f in audio_files:
+            filename = f.split("/")[-1] # get '0_20260511.wav'
+            user_id = filename.split("_")[0]
+            user_indices.add(user_id)
+            
+        stats_text = f"📊 **Stats:** {len(audio_files)} recordings from {len(user_indices)} unique contributors."
+        return audio_files, stats_text
+    except:
+        return [], "⚠️ Could not retrieve stats."
+
 def admin_login(user, pwd):
     pwd_hash = hashlib.sha256(pwd.encode()).hexdigest()
     if user == ADMIN_USERNAME and pwd_hash == ADMIN_PASSWORD:
-        files = api.list_repo_files(repo_id=DATASET_REPO_ID, repo_type="dataset")
-        audio_files = [f for f in files if f.startswith("data/")]
-        return gr.update(visible=True), gr.update(choices=audio_files), "🔓 Admin Authenticated"
-    return gr.update(visible=False), gr.update(choices=[]), "❌ Invalid Credentials"
+        audio_files, stats_text = get_stats()
+        return gr.update(visible=True), gr.update(choices=audio_files), "🔓 Admin Authenticated", stats_text
+    return gr.update(visible=False), gr.update(choices=[]), "❌ Invalid Credentials", ""
 
 def delete_selected_file(file_path):
-    if not file_path: return "⚠️ Select a file."
+    if not file_path: return "⚠️ Select a file.", gr.update()
     try:
-        # Delete the audio file
         api.delete_file(path_in_repo=file_path, repo_id=DATASET_REPO_ID, repo_type="dataset", token=HF_TOKEN)
-        
-        # Attempt to delete corresponding metadata
         meta_path = file_path.replace("data/", "metadata/meta_").replace(".wav", ".csv")
         try:
             api.delete_file(path_in_repo=meta_path, repo_id=DATASET_REPO_ID, repo_type="dataset", token=HF_TOKEN)
         except: pass 
         
-        # Refresh file list
-        files = api.list_repo_files(repo_id=DATASET_REPO_ID, repo_type="dataset")
-        audio_files = [f for f in files if f.startswith("data/")]
-        return f"🗑️ Deleted {file_path}", gr.update(choices=audio_files)
+        audio_files, stats_text = get_stats()
+        return f"🗑️ Deleted {file_path}. {stats_text}", gr.update(choices=audio_files, value=None)
     except Exception as e: return f"❌ Error: {str(e)}", gr.update()
 
 # --- UI ---
@@ -85,12 +97,11 @@ with gr.Blocks() as demo:
         # STUDENT TAB
         with gr.TabItem("Student Recording"):
             with gr.Row():
-                email_input = gr.Textbox(label="Email", placeholder="test@univ-lemans.fr")
+                email_input = gr.Textbox(label="Email", placeholder="test")
                 login_btn = gr.Button("Verify", variant="primary")
             login_status = gr.Markdown("Waiting for login...")
 
             with gr.Column(visible=False) as recording_zone:
-                # gr.Separator()
                 label_input = gr.Radio(choices=["Engine", "Environmental", "Mechanical"], label="Category")
                 audio_input = gr.Audio(label="Record (40s)", sources=["microphone"], type="filepath")
                 submit_btn = gr.Button("🚀 Submit", variant="primary")
@@ -104,9 +115,11 @@ with gr.Blocks() as demo:
                 admin_login_btn = gr.Button("Login Admin")
             
             admin_msg = gr.Markdown("Log in to manage files.")
+            # This will show the statistics
+            admin_stats_display = gr.Markdown("") 
             
             with gr.Column(visible=False) as admin_panel:
-                # gr.Separator()
+                gr.Separator()
                 file_dropdown = gr.Dropdown(label="Select File to Remove", choices=[])
                 delete_btn = gr.Button("🗑️ Delete Selected File", variant="stop")
                 delete_status = gr.Textbox(label="Delete Progress")
@@ -115,8 +128,17 @@ with gr.Blocks() as demo:
     login_btn.click(verify_user, [email_input], [recording_zone, login_status])
     submit_btn.click(upload_data, [email_input, label_input, audio_input], final_status)
     
-    admin_login_btn.click(admin_login, [admin_user, admin_pass], [admin_panel, file_dropdown, admin_msg])
-    delete_btn.click(delete_selected_file, [file_dropdown], [delete_status, file_dropdown])
+    admin_login_btn.click(
+        admin_login, 
+        [admin_user, admin_pass], 
+        [admin_panel, file_dropdown, admin_msg, admin_stats_display]
+    )
+    
+    delete_btn.click(
+        delete_selected_file, 
+        [file_dropdown], 
+        [delete_status, file_dropdown]
+    )
 
 if __name__ == "__main__":
     demo.launch(theme=gr.themes.Soft())