Add application file

app.py

@@ -1,7 +1,122 @@
+import os
 import gradio as gr
+import hashlib
+from datetime import datetime
+from huggingface_hub import HfApi
+from pathlib import Path
 
-def greet(name):
-    return "Hello " + name + "!!"
+# --- CONFIGURATION ---
+DATASET_REPO_ID = "MeysamSh/ENSIMSoundDataCollection"
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Admin Credentials (Keep these secure!)
+ADMIN_USERNAME = "admin"
+ADMIN_PASSWORD = "30c8663d3ca10ededd17ac1b55f3d533ab29cf1b8470b1729af09afda3f0a516" 
+
+AUTHORIZED_USERS = [
+    "5e884898da28047151d0e56f8dc6292773603d0d6aabbdd62a11ef721d1542d8", 
+    "test@univ-lemans.fr"
+]
+
+api = HfApi()
+
+# --- LOGIC FUNCTIONS ---
+
+def verify_user(email):
+    if not email: return gr.update(visible=False), "⚠️ Enter email."
+    clean_email = email.strip().lower()
+    email_hash = hashlib.sha256(clean_email.encode()).hexdigest()
+    if clean_email in AUTHORIZED_USERS or email_hash in AUTHORIZED_USERS:
+        return gr.update(visible=True), f"✅ Access Granted: {clean_email}"
+    return gr.update(visible=False), "🚫 Not authorized."
+
+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
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        unique_filename = f"{email_index}_{timestamp}.wav"
+        
+        # Upload Audio
+        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}"
+        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}."
+    except Exception as e: return f"❌ Error: {str(e)}"
+
+# --- ADMIN LOGIC ---
+
+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"
+
+def delete_selected_file(file_path):
+    if not file_path: return "⚠️ Select a file."
+    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)
+    except Exception as e: return f"❌ Error: {str(e)}", gr.update()
+
+# --- UI ---
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎙️ ENSIM Sound Data Platform")
+    
+    with gr.Tabs():
+        # STUDENT TAB
+        with gr.TabItem("Student Recording"):
+            with gr.Row():
+                email_input = gr.Textbox(label="Email", placeholder="test@univ-lemans.fr")
+                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")
+                final_status = gr.Textbox(label="Status", interactive=False)
+
+        # ADMIN TAB
+        with gr.TabItem("Administration"):
+            with gr.Row():
+                admin_user = gr.Textbox(label="Admin Username")
+                admin_pass = gr.Textbox(label="Admin Password", type="password")
+                admin_login_btn = gr.Button("Login Admin")
+            
+            admin_msg = gr.Markdown("Log in to manage files.")
+            
+            with gr.Column(visible=False) as admin_panel:
+                # 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")
+
+    # --- EVENT HANDLERS ---
+    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])
+
+if __name__ == "__main__":
+    demo.launch(theme=gr.themes.Soft())