diff options
| author | Raghuram Subramani <raghus2247@gmail.com> | 2025-04-07 12:15:39 +0530 |
|---|---|---|
| committer | Raghuram Subramani <raghus2247@gmail.com> | 2025-04-07 12:15:39 +0530 |
| commit | 24ea17819efdd4d3ab19271c7a1f277cc7490277 (patch) | |
| tree | f777915845f70ee0938c35d0260cd2cb8a25177c | |
| parent | 06392f8588ffac7022654a5f7b2e1e8c3d0d2dd9 (diff) | |
update app
| -rw-r--r-- | flake.nix | 3 | ||||
| -rw-r--r-- | heinous_offences/app.py | 143 | ||||
| -rw-r--r-- | heinous_offences/import_csv.py | 24 |
3 files changed, 134 insertions, 36 deletions
@@ -17,6 +17,7 @@ p.uvicorn p.jinja2 p.streamlit + p.gradio # p.pdf2image # p.openai-whisper @@ -29,6 +30,8 @@ geckodriver tesseract + + nodejs_22 ]; }; }; diff --git a/heinous_offences/app.py b/heinous_offences/app.py index d527aa8..7cec2bd 100644 --- a/heinous_offences/app.py +++ b/heinous_offences/app.py @@ -1,19 +1,138 @@ -import streamlit as st +from fastapi import FastAPI, Form, UploadFile +from fastapi.responses import HTMLResponse, RedirectResponse +import gradio as gr +import csv from tinydb import TinyDB, Query +import shutil -db = TinyDB('db.json') +DB_FILE = "db.json" +ADMIN_PASSWORD = "shivermetimbers" -st.title('Heinous Crime Lookup') +db = TinyDB(DB_FILE) -section = st.number_input('Section', value=0) +name_from_key = { + 'ipc': 'Indian Penal Code', + 'bns': 'Bhartiya Nyay Sanhita', + 'pocso': 'POCSO, 2012', + 'scst': 'SCST Act, 1989', + 'ndps': 'NDPS Act, 1985', + 'arms': 'Arms Act, 1959', + 'motor': 'Motor Vehicle Act, 1988', + 'it': 'IT Act, 2000' +} -if section > 0: - offence = db.search(Query().section == str(section))[0] - st.subheader(offence['severity']) - st.write(offence['section_text']) +key_from_name = { value: key for key, value in name_from_key.items() } - st.subheader('Minimum Punishment') - st.write(offence['minimum_punishment']) +crime_query = Query() - st.subheader('Comments') - st.write(offence['comment']) +# ---------- Gradio Logic ---------- +def lookup_crime(section, act_key): + if section <= 0: + return "Please enter a valid section number.", "", "", "" + + table = db.table(key_from_name[act_key]) + results = table.search(crime_query.section == str(section)) + + if not results: + return f"No record found for section {section} under {name_from_key.get(act_key)}.", "", "", "" + + offence = results[0] + return [ + f"## Severity: {offence.get('severity', 'N/A')}", + f"{offence.get('section_text', 'N/A')}", + f"### Minimum Punishment: {offence.get('minimum_punishment', 'N/A')}", + f"Comments: {offence.get('comment', 'N/A')}" + ] + +gradio_ui = gr.Blocks() +with gradio_ui: + gr.Markdown("## Heinous Crime Lookup Tool") + + with gr.Row(): + section_input = gr.Number(label="Enter Section Number", value=0) + act_dropdown = gr.Dropdown(choices=list(name_from_key.values()), label="Select Act") + + submit_btn = gr.Button("Lookup") + severity = gr.Markdown() + section_text = gr.Markdown() + punishment = gr.Markdown() + comment = gr.Markdown() + + submit_btn.click( + fn=lookup_crime, + inputs=[section_input, act_dropdown], + outputs=[severity, section_text, punishment, comment] + ) + +# ---------- FastAPI Logic ---------- +app = FastAPI() + +@app.get("/admin", response_class=HTMLResponse) +async def admin_form(): + options_html = "".join( + f"<option value='{key}'>{label}</option>" for key, label in name_from_key.items() + ) + return f""" + <html> + <head><title>Admin Upload</title></head> + <link + rel="stylesheet" + href="https://cdn.jsdelivr.net/npm/@picocss/pico@2/css/pico.min.css" + > + <body> + <main class='container'> + <h2>Admin Panel - Upload CSV</h2> + <form action="/admin" method="post" enctype="multipart/form-data"> + <label>Password:</label> + <input type="password" name="password" required><br><br> + + <label>Select Act:</label> + <select name="act" required> + {options_html} + </select><br><br> + + <label>CSV File:</label> + <input type="file" name="file" accept=".csv" required><br><br> + <input type="submit" value="Upload & Replace DB"> + </form> + </main> + </body> + </html> + """ + +@app.post("/admin") +async def handle_admin_upload( + password: str = Form(...), + act: str = Form(...), + file: UploadFile = Form(...) +): + if password != ADMIN_PASSWORD: + return HTMLResponse("<h3>Incorrect password.</h3>", status_code=401) + + if act not in name_from_key: + return HTMLResponse("<h3>Invalid Act selected.</h3>", status_code=400) + + with open('tmp.csv', "wb") as buffer: + shutil.copyfileobj(file.file, buffer) + + db.drop_table(act) + table = db.table(act) + + with open('tmp.csv', "r", encoding="utf-8") as f: + reader = csv.reader(f) + next(reader) + for row in reader: + if len(row) < 5: + continue + table.insert({ + "section": row[0], + "section_text": row[1], + "minimum_punishment": row[2], + "severity": row[3], + "comment": row[4], + }) + + return RedirectResponse("/admin", status_code=303) + +# Mount Gradio on "/" +gr.mount_gradio_app(app, gradio_ui, path="/") diff --git a/heinous_offences/import_csv.py b/heinous_offences/import_csv.py deleted file mode 100644 index bff906f..0000000 --- a/heinous_offences/import_csv.py +++ /dev/null @@ -1,24 +0,0 @@ -import tinydb -import csv - -db = tinydb.TinyDB('db.json') -file = open('ipc.csv', 'r') - -reader = csv.reader(file) - -header_parsed = False -for row in reader: - if not header_parsed: - header_parsed = True - continue - - record = { - 'section': row[0], - 'section_text': row[1], - 'minimum_punishment': row[2], - 'severity': row[3], - 'comment': row[4] - } - db.insert(record) - -file.close() |