Update app.py

app.py

@@ -1,113 +1,72 @@
 # app.py
-from flask import Flask, render_template, request, redirect, url_for, flash
-import sqlite3
-from datetime import datetime
-
-app = Flask(__name__)
-app.secret_key = 'supersecretkey'
-
-# Database connection helper
-def get_db_connection():
-    conn = sqlite3.connect('healthcare.db')
-    conn.row_factory = sqlite3.Row
-    return conn
-
-### 1. Patient Registration ###
-@app.route('/register', methods=['GET', 'POST'])
-def register():
-    if request.method == 'POST':
-        name = request.form['name']
-        age = request.form['age']
-        gender = request.form['gender']
-        mobile = request.form['mobile']
-
-        if len(mobile) != 10 or not mobile.isdigit():
-            flash("Mobile number must be exactly 10 digits.", "error")
-            return redirect(url_for('register'))
-
-        conn = get_db_connection()
-        conn.execute("INSERT INTO patients (name, age, gender, mobile) VALUES (?, ?, ?, ?)",
-                     (name, age, gender, mobile))
-        conn.commit()
-        patient_id = conn.execute("SELECT last_insert_rowid()").fetchone()[0]
-        conn.close()
-
-        flash(f'Registration successful! Patient ID: {patient_id}', 'success')
-        return redirect(url_for('register'))
-
-    return render_template('register.html')
-
-### 2. Book Appointment ###
-@app.route('/book', methods=['GET', 'POST'])
-def book():
-    conn = get_db_connection()
-    doctors = ["Dr. Pranay - Cardiology", "Dr. Priyanka - Gynecology", "Dr. Pavan - Physiotherapy",
-               "Dr. Pranav - Orthopedic", "Dr. Uday - General Surgery", "Dr. RajaSuri - Psychology"]
-    if request.method == 'POST':
-        patient_id = request.form['patient_id']
-        doctor = request.form['doctor']
-        date = request.form['date']
-        time_slot = request.form['time_slot']
-
-        # Check if slot is available
-        slot_check = conn.execute(
-            "SELECT * FROM appointments WHERE date = ? AND time_slot = ? AND doctor = ?",
-            (date, time_slot, doctor)).fetchone()
-        if slot_check:
-            flash("Slot is already booked. Please choose another.", "error")
-        else:
-            conn.execute("INSERT INTO appointments (patient_id, doctor, date, time_slot, is_booked) VALUES (?, ?, ?, ?, ?)",
-                         (patient_id, doctor, date, time_slot, 1))
-            conn.commit()
-            flash("Appointment booked successfully!", "success")
-
-        conn.close()
-        return redirect(url_for('book'))
-
-    return render_template('book.html', doctors=doctors)
-
-### 3. Laboratory Tests ###
-@app.route('/lab', methods=['GET', 'POST'])
-def lab():
-    tests = {"Urine Test": 200, "Blood Test": 500, "CBP": 300, "Saliva Test": 150}
-    if request.method == 'POST':
-        patient_id = request.form['patient_id']
-        selected_tests = request.form.getlist('tests')
-        conn = get_db_connection()
-
-        total_amount = 0
-        for test in selected_tests:
-            amount = tests[test]
-            total_amount += amount
-            conn.execute("INSERT INTO lab_tests (patient_id, test_name, amount) VALUES (?, ?, ?)",
-                         (patient_id, test, amount))
-
-        # Link total amount to billing
-        conn.execute("INSERT INTO billing (patient_id, total_amount) VALUES (?, ?)",
-                     (patient_id, total_amount))
-        conn.commit()
-        conn.close()
-        flash(f"Lab tests booked successfully! Total amount: {total_amount}", "success")
-
-        return redirect(url_for('lab'))
-
-    return render_template('lab.html', tests=tests)
-
-### 4. Reports ###
-@app.route('/reports/<int:patient_id>')
-def reports(patient_id):
-    conn = get_db_connection()
-    reports = conn.execute("SELECT * FROM lab_tests WHERE patient_id = ?", (patient_id,)).fetchall()
-    conn.close()
-    return render_template('reports.html', reports=reports)
-
-### 5. Billing ###
-@app.route('/billing/<int:patient_id>')
-def billing(patient_id):
-    conn = get_db_connection()
-    billing = conn.execute("SELECT total_amount FROM billing WHERE patient_id = ?", (patient_id,)).fetchone()
-    conn.close()
-    return render_template('billing.html', billing=billing, patient_id=patient_id)
-
-if __name__ == "__main__":
-    app.run(debug=True)
+import gradio as gr
+from patient_registration import register_patient
+from appointment_booking import book_appointment
+from laboratory import book_lab_tests
+from reports import get_reports
+from billing import get_billing
+
+def register_interface(name, age, gender, mobile):
+    return register_patient(name, age, gender, mobile)
+
+def book_interface(patient_id, doctor, date, time_slot):
+    return book_appointment(patient_id, doctor, date, time_slot)
+
+def lab_interface(patient_id, tests):
+    return book_lab_tests(patient_id, tests)
+
+def report_interface(patient_id):
+    return get_reports(patient_id)
+
+def billing_interface(patient_id):
+    return get_billing(patient_id)
+
+# Define Gradio app interfaces
+with gr.Blocks() as app:
+    gr.Markdown("# Healthcare App")
+
+    # Patient Registration
+    with gr.Tab("Register Patient"):
+        name = gr.Textbox(label="Name")
+        age = gr.Number(label="Age")
+        gender = gr.Dropdown(["Male", "Female", "Other"], label="Gender")
+        mobile = gr.Textbox(label="Mobile Number")
+        register_btn = gr.Button("Register")
+        register_output = gr.Textbox()
+        register_btn.click(register_interface, inputs=[name, age, gender, mobile], outputs=register_output)
+
+    # Book Appointment
+    with gr.Tab("Book Appointment"):
+        patient_id = gr.Number(label="Patient ID")
+        doctor = gr.Dropdown(["Dr. Pranay - Cardiology", "Dr. Priyanka - Gynecology", "Dr. Pavan - Physiotherapy",
+                              "Dr. Pranav - Orthopedic", "Dr. Uday - General Surgery", "Dr. RajaSuri - Psychology"], label="Doctor")
+        date = gr.Textbox(label="Date (YYYY-MM-DD)")
+        time_slot = gr.Textbox(label="Time Slot (e.g., 9 AM - 10 AM)")
+        book_btn = gr.Button("Book Appointment")
+        book_output = gr.Textbox()
+        book_btn.click(book_interface, inputs=[patient_id, doctor, date, time_slot], outputs=book_output)
+
+    # Laboratory Tests
+    with gr.Tab("Laboratory Tests"):
+        patient_id_lab = gr.Number(label="Patient ID")
+        tests = gr.CheckboxGroup(["Urine Test", "Blood Test", "CBP", "Saliva Test"], label="Select Tests")
+        lab_btn = gr.Button("Book Tests")
+        lab_output = gr.Textbox()
+        lab_btn.click(lab_interface, inputs=[patient_id_lab, tests], outputs=lab_output)
+
+    # Reports
+    with gr.Tab("Reports"):
+        patient_id_reports = gr.Number(label="Patient ID")
+        report_btn = gr.Button("Get Reports")
+        report_output = gr.Textbox()
+        report_btn.click(report_interface, inputs=[patient_id_reports], outputs=report_output)
+
+    # Billing
+    with gr.Tab("Billing"):
+        patient_id_billing = gr.Number(label="Patient ID")
+        billing_btn = gr.Button("Get Billing")
+        billing_output = gr.Textbox()
+        billing_btn.click(billing_interface, inputs=patient_id_billing, outputs=billing_output)
+
+# Launch the app
+app.launch()