Upload folder using huggingface_hub

home.py

@@ -1,5 +1,6 @@
 import os
 from datetime import datetime
+from random import choices
 from typing import Dict, Any, List, Tuple
 
 import gradio as gr
@@ -220,7 +221,11 @@ def add_patient_ui(user_email, name, age, gender):
     if not user:
         return "User not found"
     pid = db.add_patient(user["_id"], name, age, gender)
-    return f"✅ Patient {name} added (ID: {pid})"
+    if pid:
+        (user_dict,patients) = load_user(user_email)
+        return f"✅ Patient {name} added (ID: {pid})", gr.update(choices=patients)
+    else:
+        return f"❌ Error adding patient", gr.update()
 
 
 def edit_patient_ui(patient_id, name, age, gender):
@@ -231,36 +236,38 @@ def edit_patient_ui(patient_id, name, age, gender):
     return "✅ Updated" if success else "❌ Patient not found"
 
 
-def delete_patient_ui(patient_id):
+def delete_patient_ui(user_email, patient_id):
     db = get_db()
     success = db.delete_patient(patient_id)
-    return "✅ Deleted" if success else "❌ Patient not found"
+    if success:
+        (user,patients) = load_user(user_email)
+        return "✅ Deleted", gr.update(choices=patients)
+    else:
+        return "❌ Patient not found", gr.update()
 
 
-# ----- Gradio UI -----
 # ----- Gradio UI (Sidebar Layout) -----
 with gr.Blocks(
     title=get_app_title(), theme=get_app_theme(), css=get_css(), fill_height=True
 ) as demo:
-    with gr.Row():
-        with gr.Column(scale=1):  # Sidebar
-            gr.Markdown("### Sidebar")
-
-            email_in = gr.Textbox(label="User Email", placeholder="doctor1@sheami.com")
-            load_btn = gr.Button("🔍 Load Patients")
+    with gr.Sidebar():  # Sidebar
+        # gr.Markdown("### Sidebar")
 
-            patient_list = gr.Radio(label="Patients", choices=[], interactive=True)
+        email_in = gr.Textbox(label="User Email", placeholder="doctor1@sheami.com")
+        load_btn = gr.Button("🔍 Load Patients")
 
-            with gr.Accordion("➕ Add Patient", open=False):
-                new_name = gr.Textbox(label="Name")
-                new_age = gr.Number(label="Age")
-                new_gender = gr.Dropdown(["M", "F"], label="Gender")
-                add_btn = gr.Button("Add")
-                add_out = gr.Textbox(label="Status")
+        patient_list = gr.Radio(label="Patients", choices=[], interactive=True)
 
-            delete_btn = gr.Button("🗑️ Delete Selected")
-            delete_out = gr.Textbox(label="Status")
+        with gr.Accordion("➕ Add Patient", open=False):
+            new_name = gr.Textbox(label="Name")
+            new_age = gr.Number(label="Age")
+            new_gender = gr.Dropdown(["M", "F"], label="Gender")
+            add_btn = gr.Button("Add")
+            add_out = gr.Textbox(label="Status")
 
+        delete_btn = gr.Button("🗑️ Delete Selected")
+        delete_out = gr.Textbox(label="Status")
+    with gr.Row():
         with gr.Column(scale=3):  # Main area
             with gr.Row():
                 edit_name = gr.Textbox(label="Edit Name")
@@ -335,11 +342,11 @@ with gr.Blocks(
     add_btn.click(
         add_patient_ui,
         inputs=[email_in, new_name, new_age, new_gender],
-        outputs=add_out,
+        outputs=[add_out,patient_list],
     )
 
     delete_btn.click(
-        delete_patient_ui, inputs=[patient_list], outputs=delete_out
+        delete_patient_ui, inputs=[email_in, patient_list], outputs=[delete_out,patient_list]
     )
 
     edit_btn.click(

modules/db.py

@@ -7,6 +7,9 @@ from dotenv import load_dotenv
 class SheamiDB:
     def __init__(self, uri: str, db_name: str = "sheami"):
         """Initialize connection to MongoDB Atlas (or local Mongo)."""
+        if not uri:
+            load_dotenv(override=True)
+            uri = os.getenv("MONGODB_URI")
         self.client = MongoClient(uri)
         self.db = self.client[db_name]
 
@@ -140,6 +143,10 @@ class SheamiDB:
         result = self.patients.update_one({"_id": patient_id}, {"$set": fields})
         return result.modified_count > 0
 
-    def delete_patient(self, patient_id):
-        result = self.patients.delete_one({"_id": patient_id})
-        return result.deleted_count > 0
+    def delete_patient(self, patient_id: str) -> bool:
+        try:
+            result = self.patients.delete_one({"_id": ObjectId(patient_id)})
+            return result.deleted_count > 0
+        except Exception as e:
+            print(f"Error deleting patient: {e}")
+            return False