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_FILE = "db.json"
ADMIN_PASSWORD = "shivermetimbers"
db = TinyDB(DB_FILE)
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'
}
key_from_name = { value: key for key, value in name_from_key.items() }
crime_query = Query()
# ---------- Gradio Logic ----------
def lookup_crime(section, act_key):
table = db.table(key_from_name[act_key])
results = table.search(crime_query.section == section)
if not results:
return f"No record found for section {section} under {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')}"
]
gradio_ui = gr.Blocks()
with gradio_ui:
gr.Markdown("## Heinous Crime Lookup Tool")
with gr.Row():
section_input = gr.Text(label="Enter Section Number")
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()
submit_btn.click(
fn=lookup_crime,
inputs=[section_input, act_dropdown],
outputs=[severity, section_text, punishment]
)
# ---------- FastAPI Logic ----------
app = FastAPI()
@app.get("/admin", response_class=HTMLResponse)
async def admin_form():
options_html = "".join(
f"" for key, label in name_from_key.items()
)
return f"""
Admin Upload
Admin Panel - Upload CSV
"""
@app.post("/admin")
async def handle_admin_upload(
password: str = Form(...),
act: str = Form(...),
file: UploadFile = Form(...)
):
if password != ADMIN_PASSWORD:
return HTMLResponse("Incorrect password.
", status_code=401)
if act not in name_from_key:
return HTMLResponse("Invalid Act selected.
", 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) < 4:
continue
table.insert({
"section": row[0],
"section_text": row[1],
"minimum_punishment": row[2],
"severity": row[3]
})
return RedirectResponse("/admin", status_code=303)
# Mount Gradio on "/"
gr.mount_gradio_app(app, gradio_ui, path="/")