Update app.py

app.py

@@ -1,6 +1,3 @@
-# Enhanced Hospital Queue Management System - Gradio App for Hugging Face
-# Features: Multiple Doctors, Patient Queue, Phone Number Status Check, Appointment Scheduling, Estimated Wait Times
-
 import sqlite3
 import gradio as gr
 import pandas as pd
@@ -31,7 +28,7 @@ cursor.execute('''CREATE TABLE IF NOT EXISTS doctors (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
     name TEXT NOT NULL,
     specialty TEXT,
-    avg_consultation_time INTEGER DEFAULT 15,  # in minutes
+    avg_consultation_time INTEGER DEFAULT 15,  -- in minutes
     available BOOLEAN DEFAULT 1
 )''')
 
@@ -41,7 +38,7 @@ cursor.execute('''CREATE TABLE IF NOT EXISTS patients (
     phone TEXT NOT NULL,
     email TEXT,
     symptoms TEXT,
-    priority INTEGER DEFAULT 3,  # 1=emergency, 2=urgent, 3=normal
+    priority INTEGER DEFAULT 3,  -- 1=emergency, 2=urgent, 3=normal
     doctor_id INTEGER,
     queue_number INTEGER,
     token TEXT UNIQUE,
@@ -826,141 +823,67 @@ with gr.Blocks(theme=theme, title="Hospital Queue Management System") as demo:
         
     # Admin Panel Tab    
     with gr.Tab("⚙️ Admin Panel"):
-        gr.Markdown("### Manage Doctors")
-        
-        # Doctor availability management
-        doctors_table = gr.Dataframe(
-            headers=["ID", "Name", "Specialty", "Status"], 
-            interactive=False,
-            label="Doctors List"
-        )
-        
-        with gr.Row():
-            toggle_doctor_id = gr.Number(label="Doctor ID", precision=0)
-            toggle_btn = gr.Button("Toggle Availability")
-            toggle_output = gr.Textbox(label="Result")
+        gr.Markdown("### Doctor Availability Management")
+    
+    doctors_table = gr.Dataframe(
+        headers=["ID", "Name", "Specialty", "Status"],
+        interactive=False,
+        label="Doctors"
+    )
+    
+    with gr.Row():
+        doctor_id = gr.Number(label="Doctor ID", precision=0)
+        toggle_btn = gr.Button("Toggle Availability")
+        toggle_output = gr.Textbox(label="Result")
+    
+    refresh_doctors_btn = gr.Button("Refresh Doctor List")
+    
+    # Set up event handlers
+    toggle_btn.click(
+        fn=toggle_doctor_availability, 
+        inputs=doctor_id, 
+        outputs=toggle_output
+    )
+    
+    refresh_doctors_btn.click(
+        fn=get_doctor_availability, 
+        inputs=[], 
+        outputs=doctors_table
+    )
+    
+    # Initialize doctors table
+    demo.load(
+        fn=get_doctor_availability, 
+        inputs=[], 
+        outputs=doctors_table
+    )
+
+    # System Maintenance Section
+    gr.Markdown("### System Maintenance")
+    
+    with gr.Row():
+        backup_btn = gr.Button("Backup Database")
+        backup_output = gr.Textbox(label="Backup Status")
+    
+    def backup_database():
+        """Create a backup of the database"""
+        try:
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            backup_file = f"hospital_backup_{timestamp}.db"
             
-        refresh_doctors_btn = gr.Button("Refresh Doctors List")
-        
-        # Set up event handlers
-        toggle_btn.click(
-            fn=toggle_doctor_availability, 
-            inputs=toggle_doctor_id, 
-            outputs=toggle_output
-        )
-        
-        refresh_doctors_btn.click(
-            fn=get_doctor_availability,
-            inputs=[],
-            outputs=doctors_table
-        )
-        
-        # Initialize doctors table
-        demo.load(fn=get_doctor_availability, inputs=[], outputs=doctors_table)
-        
-        # System monitoring section
-        gr.Markdown("### System Monitoring")
-        
-        with gr.Row():
-            db_size = gr.Textbox(label="Database Size")
-            system_status = gr.Textbox(label="System Status")
+            # Create connection to new file
+            backup_conn = sqlite3.connect(backup_file)
             
-        def check_system_status():
-            """Check system status and database size"""
-            try:
-                # Get database file size
-                if os.path.exists("hospital.db"):
-                    size_bytes = os.path.getsize("hospital.db")
-                    size_kb = size_bytes / 1024
-                    if size_kb < 1024:
-                        size_str = f"{size_kb:.2f} KB"
-                    else:
-                        size_mb = size_kb / 1024
-                        size_str = f"{size_mb:.2f} MB"
-                else:
-                    size_str = "Database file not found"
-                
-                # Check database connection
-                if conn:
-                    try:
-                        cursor.execute("SELECT COUNT(*) FROM doctors")
-                        cursor.fetchone()
-                        status = "System operational - Database connection OK"
-                    except:
-                        status = "Database connection error"
-                else:
-                    status = "Database connection not established"
-                    
-                return size_str, status
-                
-            except Exception as e:
-                return "Error getting size", f"Error: {str(e)}"
-        
-        check_system_btn = gr.Button("Check System Status")
-        check_system_btn.click(
-            fn=check_system_status,
-            inputs=[],
-            outputs=[db_size, system_status]
-        )
-        
-        # Backup & Maintenance
-        gr.Markdown("### Backup & Maintenance")
-        
-        def backup_database():
-            """Create a backup of the database"""
-            try:
-                timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-                backup_file = f"hospital_backup_{timestamp}.db"
-                
-                # Create connection to source and destination databases
-                source_conn = sqlite3.connect("hospital.db")
-                backup_conn = sqlite3.connect(backup_file)
-                
-                # Backup
-                source_conn.backup(backup_conn)
-                
-                # Close connections
-                source_conn.close()
-                backup_conn.close()
-                
-                return f"Backup created successfully: {backup_file}"
-                
-            except Exception as e:
-                return f"Backup failed: {str(e)}"
-        
-        backup_btn = gr.Button("Create Database Backup")
-        backup_output = gr.Textbox(label="Backup Result")
-        
-        backup_btn.click(fn=backup_database, inputs=[], outputs=backup_output)
-        
-        # Emergency queue reset
-        def reset_queue():
-            """Reset the entire patient queue for emergency situations"""
-            try:
-                cursor.execute("UPDATE patients SET status='Completed' WHERE status IN ('Waiting', 'In Consultation')")
-                conn.commit()
-                affected = cursor.rowcount
-                return f"Queue reset successfully. {affected} patients marked as completed."
-            except Exception as e:
-                conn.rollback()
-                return f"Queue reset failed: {str(e)}"
-        
-        with gr.Row():
-            gr.Markdown("### ⚠️ Emergency Queue Reset ⚠️")
-        with gr.Row():
-            reset_confirm = gr.Checkbox(label="I confirm I want to reset all queues")
-            reset_btn = gr.Button("Reset All Queues", variant="stop")
-            reset_output = gr.Textbox(label="Reset Result")
-        
-        def safe_reset_queue(confirmed):
-            if confirmed:
-                return reset_queue()
-            else:
-                return "Please check the confirmation box to reset queues"
-                
-        reset_btn.click(fn=safe_reset_queue, inputs=reset_confirm, outputs=reset_output)
-        
-# ---------------------------
-# LAUNCH APP
-# ---------------------------
-demo.launch()
+            # Copy data
+            conn.backup(backup_conn)
+            backup_conn.close()
+            
+            return f"Backup created successfully: {backup_file}"
+        except Exception as e:
+            return f"Backup failed: {str(e)}"
+    
+    backup_btn.click(fn=backup_database, inputs=[], outputs=backup_output)
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(share=False)