Update app.py

app.py

@@ -1,7 +1,143 @@
 import gradio as gr
+import sqlite3
+import os
+import pandas as pd
+from datetime import datetime
 
-def greet(name):
-    return "Hello " + name + "!!"
+DB_FILE = "cases.db"
+
+# ------------------ DB Setup ------------------
+def init_db():
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("""
+        CREATE TABLE IF NOT EXISTS cases (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            citizen_name TEXT,
+            fine_amount REAL,
+            description TEXT,
+            police_evidence TEXT,
+            citizen_appeal TEXT,
+            citizen_evidence TEXT,
+            status TEXT,
+            created_at TEXT
+        )
+    """)
+    conn.commit()
+    conn.close()
+
+init_db()
+
+# ------------------ Police Functions ------------------
+def police_create_case(citizen_name, fine_amount, description, police_evidence):
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("""
+        INSERT INTO cases (citizen_name, fine_amount, description, police_evidence, status, created_at)
+        VALUES (?, ?, ?, ?, ?, ?)
+    """, (citizen_name, fine_amount, description, police_evidence.name if police_evidence else None,
+          "Pending Payment", datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
+    conn.commit()
+    case_id = c.lastrowid
+    conn.close()
+    return f"✅ Case created with ID: {case_id}"
+
+# ------------------ Citizen Functions ------------------
+def citizen_view_case(case_id):
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("SELECT * FROM cases WHERE id=?", (case_id,))
+    row = c.fetchone()
+    conn.close()
+    if not row:
+        return "❌ Case not found"
+    return pd.DataFrame([row], columns=["ID", "Citizen", "Fine", "Description", "Police Evidence",
+                                        "Citizen Appeal", "Citizen Evidence", "Status", "Created At"])
+
+def citizen_pay_fine(case_id):
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("UPDATE cases SET status=? WHERE id=?", ("Fine Paid", case_id))
+    conn.commit()
+    conn.close()
+    return f"💰 Fine for case {case_id} marked as paid."
+
+def citizen_appeal(case_id, appeal_text, evidence_file):
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("UPDATE cases SET citizen_appeal=?, citizen_evidence=?, status=? WHERE id=?",
+              (appeal_text, evidence_file.name if evidence_file else None, "Appealed", case_id))
+    conn.commit()
+    conn.close()
+    return f"📤 Appeal submitted for case {case_id}"
+
+# ------------------ Magistrate Functions ------------------
+def magistrate_dashboard():
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("SELECT * FROM cases WHERE status='Appealed'")
+    rows = c.fetchall()
+    conn.close()
+    if not rows:
+        return "No appealed cases."
+    return pd.DataFrame(rows, columns=["ID", "Citizen", "Fine", "Description", "Police Evidence",
+                                       "Citizen Appeal", "Citizen Evidence", "Status", "Created At"])
+
+def magistrate_decide(case_id, decision):
+    new_status = {
+        "Approve Appeal": "Appeal Approved",
+        "Reject Appeal": "Appeal Rejected",
+        "Escalate Hearing": "Hearing Scheduled"
+    }[decision]
+
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("UPDATE cases SET status=? WHERE id=?", (new_status, case_id))
+    conn.commit()
+    conn.close()
+    return f"⚖️ Case {case_id} updated: {new_status}"
+
+# ------------------ Gradio UI ------------------
+with gr.Blocks() as demo:
+    gr.Markdown("# 🚦 Smart Traffic Fine & Case Management MVP")
+
+    with gr.Tab("Citizen Portal"):
+        case_id_input = gr.Number(label="Enter Case ID", precision=0)
+        view_btn = gr.Button("View Case")
+        case_output = gr.Dataframe()
+        view_btn.click(citizen_view_case, inputs=case_id_input, outputs=case_output)
+
+        with gr.Row():
+            pay_btn = gr.Button("Pay Fine")
+            appeal_text = gr.Textbox(label="Appeal Text")
+            appeal_file = gr.File(label="Upload Appeal Evidence", type="file")
+            appeal_btn = gr.Button("Submit Appeal")
+
+        pay_result = gr.Textbox(label="Result")
+        appeal_result = gr.Textbox(label="Appeal Result")
+
+        pay_btn.click(citizen_pay_fine, inputs=case_id_input, outputs=pay_result)
+        appeal_btn.click(citizen_appeal, inputs=[case_id_input, appeal_text, appeal_file], outputs=appeal_result)
+
+    with gr.Tab("Traffic Police Portal"):
+        citizen_name = gr.Textbox(label="Citizen Name")
+        fine_amount = gr.Number(label="Fine Amount")
+        description = gr.Textbox(label="Description")
+        police_file = gr.File(label="Upload Evidence", type="file")
+        create_case_btn = gr.Button("Create Case")
+        police_result = gr.Textbox(label="Result")
+        create_case_btn.click(police_create_case, inputs=[citizen_name, fine_amount, description, police_file], outputs=police_result)
+
+    with gr.Tab("Magistrate Portal"):
+        dashboard_btn = gr.Button("Refresh Appeals Dashboard")
+        dashboard_output = gr.Dataframe()
+        dashboard_btn.click(magistrate_dashboard, outputs=dashboard_output)
+
+        case_id_mag = gr.Number(label="Case ID", precision=0)
+        decision = gr.Dropdown(["Approve Appeal", "Reject Appeal", "Escalate Hearing"], label="Decision")
+        decide_btn = gr.Button("Submit Decision")
+        decision_result = gr.Textbox(label="Decision Result")
+        decide_btn.click(magistrate_decide, inputs=[case_id_mag, decision], outputs=decision_result)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()
+