Update app.py

app.py

@@ -1,10 +1,10 @@
+# app.py
 import gradio as gr
 import sqlite3
 import hashlib
-import os
 
 # ---------------- Database Setup ----------------
-DB_PATH = './family.db'  # relative path for HF Space
+DB_PATH = './family.db'
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 
@@ -44,6 +44,16 @@ CREATE TABLE IF NOT EXISTS expenses(
 ''')
 conn.commit()
 
+# ---------------- Default Admin ----------------
+DEFAULT_ADMIN = "admin"
+DEFAULT_PASSWORD = "1234"
+cursor.execute("SELECT * FROM users WHERE username=?", (DEFAULT_ADMIN,))
+if not cursor.fetchone():
+    cursor.execute("INSERT INTO users(username,password_hash) VALUES (?,?)",
+                   (DEFAULT_ADMIN, hashlib.sha256(DEFAULT_PASSWORD.encode()).hexdigest()))
+    conn.commit()
+    print(f"Default admin created! Username: {DEFAULT_ADMIN}, Password: {DEFAULT_PASSWORD}")
+
 # ---------------- Utility Functions ----------------
 def hash_password(password):
     return hashlib.sha256(password.encode()).hexdigest()
@@ -122,53 +132,60 @@ def delete_expense(id):
 
 # ---------------- Gradio Interface ----------------
 with gr.Blocks() as demo:
-    gr.Markdown("## Family Software Online")
-
+    gr.Markdown("## 🏠 Family Management App")
+    
+    # ---------------- Login Section ----------------
     with gr.Row():
         username = gr.Textbox(label="Username")
         password = gr.Textbox(label="Password", type="password")
         login_btn = gr.Button("Login")
         login_status = gr.Label("")
-
-    with gr.Column(visible=False) as main_app:
-        gr.Markdown("### Family Members")
-        id_input = gr.Number(label="ID (for Edit/Delete)")
-        name = gr.Textbox(label="Name")
-        dob = gr.Textbox(label="DOB (YYYY-MM-DD)")
-        gender = gr.Textbox(label="Gender")
-        relationship = gr.Textbox(label="Relationship")
-        phone = gr.Textbox(label="Phone")
-        email = gr.Textbox(label="Email")
-        add_member_btn = gr.Button("Add Member")
-        edit_member_btn = gr.Button("Edit Member")
-        delete_member_btn = gr.Button("Delete Member")
-        view_members_btn = gr.Button("View Members")
-        members_output = gr.Textbox(label="Members", lines=10)
-
-        gr.Markdown("### Expenses")
-        expense_id = gr.Number(label="Expense ID (Edit/Delete)")
-        member_id_input = gr.Number(label="Member ID")
-        amount_input = gr.Number(label="Amount")
-        category_input = gr.Textbox(label="Category")
-        date_input = gr.Textbox(label="Date (YYYY-MM-DD)")
-        desc_input = gr.Textbox(label="Description")
-        add_expense_btn = gr.Button("Add Expense")
-        edit_expense_btn = gr.Button("Edit Expense")
-        delete_expense_btn = gr.Button("Delete Expense")
-        view_expenses_btn = gr.Button("View Expenses")
-        expenses_output = gr.Textbox(label="Expenses", lines=10)
-
-    # Button actions
+        register_btn = gr.Button("Register Admin")
+    
+    main_app = gr.Column(visible=False)
+    
     login_btn.click(check_login, [username, password], [login_status, main_app])
-    add_member_btn.click(add_member, [name, dob, gender, relationship, phone, email], members_output)
-    edit_member_btn.click(edit_member, [id_input, name, dob, gender, relationship, phone, email], members_output)
-    delete_member_btn.click(delete_member, [id_input], members_output)
-    view_members_btn.click(view_members, [], members_output)
-
-    add_expense_btn.click(add_expense, [member_id_input, amount_input, category_input, date_input, desc_input], expenses_output)
-    edit_expense_btn.click(edit_expense, [expense_id, member_id_input, amount_input, category_input, date_input, desc_input], expenses_output)
-    delete_expense_btn.click(delete_expense, [expense_id], expenses_output)
-    view_expenses_btn.click(view_expenses, [], expenses_output)
-
-# ---------------- Launch without errors ----------------
+    register_btn.click(register_admin, [username, password], login_status)
+    
+    # ---------------- Tabs ----------------
+    with main_app:
+        with gr.Tabs():
+            with gr.Tab("Family Members"):
+                id_input = gr.Number(label="ID (for Edit/Delete)")
+                name = gr.Textbox(label="Name")
+                dob = gr.Textbox(label="DOB (YYYY-MM-DD)")
+                gender = gr.Textbox(label="Gender")
+                relationship = gr.Textbox(label="Relationship")
+                phone = gr.Textbox(label="Phone")
+                email = gr.Textbox(label="Email")
+                add_member_btn = gr.Button("Add Member")
+                edit_member_btn = gr.Button("Edit Member")
+                delete_member_btn = gr.Button("Delete Member")
+                view_members_btn = gr.Button("View Members")
+                members_output = gr.Textbox(label="Members", lines=10)
+                
+                add_member_btn.click(add_member, [name, dob, gender, relationship, phone, email], members_output)
+                edit_member_btn.click(edit_member, [id_input, name, dob, gender, relationship, phone, email], members_output)
+                delete_member_btn.click(delete_member, [id_input], members_output)
+                view_members_btn.click(view_members, [], members_output)
+            
+            with gr.Tab("Expenses"):
+                expense_id = gr.Number(label="Expense ID (Edit/Delete)")
+                member_id_input = gr.Number(label="Member ID")
+                amount_input = gr.Number(label="Amount")
+                category_input = gr.Textbox(label="Category")
+                date_input = gr.Textbox(label="Date (YYYY-MM-DD)")
+                desc_input = gr.Textbox(label="Description")
+                add_expense_btn = gr.Button("Add Expense")
+                edit_expense_btn = gr.Button("Edit Expense")
+                delete_expense_btn = gr.Button("Delete Expense")
+                view_expenses_btn = gr.Button("View Expenses")
+                expenses_output = gr.Textbox(label="Expenses", lines=10)
+                
+                add_expense_btn.click(add_expense, [member_id_input, amount_input, category_input, date_input, desc_input], expenses_output)
+                edit_expense_btn.click(edit_expense, [expense_id, member_id_input, amount_input, category_input, date_input, desc_input], expenses_output)
+                delete_expense_btn.click(delete_expense, [expense_id], expenses_output)
+                view_expenses_btn.click(view_expenses, [], expenses_output)
+
+# Launch the app
 demo.launch()